1 #! /usr/bin/env python

     2 

     3 # Copyright 1994 by Lance Ellinghouse

     4 # Cathedral City, California Republic, United States of America.

     5 #                        All Rights Reserved

     6 # Permission to use, copy, modify, and distribute this software and its

     7 # documentation for any purpose and without fee is hereby granted,

     8 # provided that the above copyright notice appear in all copies and that

     9 # both that copyright notice and this permission notice appear in

    10 # supporting documentation, and that the name of Lance Ellinghouse

    11 # not be used in advertising or publicity pertaining to distribution

    12 # of the software without specific, written prior permission.

    13 # LANCE ELLINGHOUSE DISCLAIMS ALL WARRANTIES WITH REGARD TO

    14 # THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND

    15 # FITNESS, IN NO EVENT SHALL LANCE ELLINGHOUSE CENTRUM BE LIABLE

    16 # FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES

    17 # WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN

    18 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT

    19 # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

    20 #

    21 # Modified by Jack Jansen, CWI, July 1995:

    22 # - Use binascii module to do the actual line-by-line conversion

    23 #   between ascii and binary. This results in a 1000-fold speedup. The C

    24 #   version is still 5 times faster, though.

    25 # - Arguments more compliant with python standard

    26 

    27 """Implementation of the UUencode and UUdecode functions.

    28 

    29 encode(in_file, out_file [,name, mode])

    30 decode(in_file [, out_file, mode])

    31 """

    32 

    33 import binascii

    34 import os

    35 import sys

    36 

    37 __all__ = ["Error", "encode", "decode"]

    38 

    39 class Error(Exception):

    40     pass

    41 

    42 def encode(in_file, out_file, name=None, mode=None):

    43     """Uuencode file"""

    44     #

    45     # If in_file is a pathname open it and change defaults

    46     #

    47     if in_file == '-':

    48         in_file = sys.stdin

    49     elif isinstance(in_file, basestring):

    50         if name is None:

    51             name = os.path.basename(in_file)

    52         if mode is None:

    53             try:

    54                 mode = os.stat(in_file).st_mode

    55             except AttributeError:

    56                 pass

    57         in_file = open(in_file, 'rb')

    58     #

    59     # Open out_file if it is a pathname

    60     #

    61     if out_file == '-':

    62         out_file = sys.stdout

    63     elif isinstance(out_file, basestring):

    64         out_file = open(out_file, 'w')

    65     #

    66     # Set defaults for name and mode

    67     #

    68     if name is None:

    69         name = '-'

    70     if mode is None:

    71         mode = 0666

    72     #

    73     # Write the data

    74     #

    75     out_file.write('begin %o %s\n' % ((mode&0777),name))

    76     data = in_file.read(45)

    77     while len(data) > 0:

    78         out_file.write(binascii.b2a_uu(data))

    79         data = in_file.read(45)

    80     out_file.write(' \nend\n')

    81 

    82 

    83 def decode(in_file, out_file=None, mode=None, quiet=0):

    84     """Decode uuencoded file"""

    85     #

    86     # Open the input file, if needed.

    87     #

    88     if in_file == '-':

    89         in_file = sys.stdin

    90     elif isinstance(in_file, basestring):

    91         in_file = open(in_file)

    92     #

    93     # Read until a begin is encountered or we've exhausted the file

    94     #

    95     while True:

    96         hdr = in_file.readline()

    97         if not hdr:

    98             raise Error('No valid begin line found in input file')

    99         if not hdr.startswith('begin'):

   100             continue

   101         hdrfields = hdr.split(' ', 2)

   102         if len(hdrfields) == 3 and hdrfields[0] == 'begin':

   103             try:

   104                 int(hdrfields[1], 8)

   105                 break

   106             except ValueError:

   107                 pass

   108     if out_file is None:

   109         out_file = hdrfields[2].rstrip()

   110         if os.path.exists(out_file):

   111             raise Error('Cannot overwrite existing file: %s' % out_file)

   112     if mode is None:

   113         mode = int(hdrfields[1], 8)

   114     #

   115     # Open the output file

   116     #

   117     opened = False

   118     if out_file == '-':

   119         out_file = sys.stdout

   120     elif isinstance(out_file, basestring):

   121         fp = open(out_file, 'wb')

   122         try:

   123             os.path.chmod(out_file, mode)

   124         except AttributeError:

   125             pass

   126         out_file = fp

   127         opened = True

   128     #

   129     # Main decoding loop

   130     #

   131     s = in_file.readline()

   132     while s and s.strip() != 'end':

   133         try:

   134             data = binascii.a2b_uu(s)

   135         except binascii.Error, v:

   136             # Workaround for broken uuencoders by /Fredrik Lundh

   137             nbytes = (((ord(s[0])-32) & 63) * 4 + 5) // 3

   138             data = binascii.a2b_uu(s[:nbytes])

   139             if not quiet:

   140                 sys.stderr.write("Warning: %s\n" % v)

   141         out_file.write(data)

   142         s = in_file.readline()

   143     if not s:

   144         raise Error('Truncated input file')

   145     if opened:

   146         out_file.close()

   147 

   148 def test():

   149     """uuencode/uudecode main program"""

   150 

   151     import optparse

   152     parser = optparse.OptionParser(usage='usage: %prog [-d] [-t] [input [output]]')

   153     parser.add_option('-d', '--decode', dest='decode', help='Decode (instead of encode)?', default=False, action='store_true')

   154     parser.add_option('-t', '--text', dest='text', help='data is text, encoded format unix-compatible text?', default=False, action='store_true')

   155 

   156     (options, args) = parser.parse_args()

   157     if len(args) > 2:

   158         parser.error('incorrect number of arguments')

   159         sys.exit(1)

   160 

   161     input = sys.stdin

   162     output = sys.stdout

   163     if len(args) > 0:

   164         input = args[0]

   165     if len(args) > 1:

   166         output = args[1]

   167 

   168     if options.decode:

   169         if options.text:

   170             if isinstance(output, basestring):

   171                 output = open(output, 'w')

   172             else:

   173                 print sys.argv[0], ': cannot do -t to stdout'

   174                 sys.exit(1)

   175         decode(input, output)

   176     else:

   177         if options.text:

   178             if isinstance(input, basestring):

   179                 input = open(input, 'r')

   180             else:

   181                 print sys.argv[0], ': cannot do -t from stdin'

   182                 sys.exit(1)

   183         encode(input, output)

   184 

   185 if __name__ == '__main__':

   186     test()