Lib/uu.py

   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 # This file implements the UUencode and UUdecode functions.
  28
  29 # encode(in_file, out_file [,name, mode])
  30 # decode(in_file [, out_file, mode])
  31
  32 import binascii
  33 import os
  34 import string
  35 import sys
  36
  37 Error = 'uu.Error'
  38
  39 def encode(in_file, out_file, name=None, mode=None):
  40     """Uuencode file"""
  41     #
  42     # If in_file is a pathname open it and change defaults
  43     #
  44     if in_file == '-':
  45         in_file = sys.stdin
  46     elif type(in_file) == type(''):
  47         if name == None:
  48             name = os.path.basename(in_file)
  49         if mode == None:
  50             try:
  51                 mode = os.stat(in_file)[0]
  52             except AttributeError:
  53                 pass
  54         in_file = open(in_file, 'rb')
  55     #
  56     # Open out_file if it is a pathname
  57     #
  58     if out_file == '-':
  59         out_file = sys.stdout
  60     elif type(out_file) == type(''):
  61         out_file = open(out_file, 'w')
  62     #
  63     # Set defaults for name and mode
  64     #
  65     if name == None:
  66         name = '-'
  67     if mode == None:
  68         mode = 0666
  69     #
  70     # Write the data
  71     #
  72     out_file.write('begin %o %s\n' % ((mode&0777),name))
  73     str = in_file.read(45)
  74     while len(str) > 0:
  75         out_file.write(binascii.b2a_uu(str))
  76         str = in_file.read(45)
  77     out_file.write(' \nend\n')
  78
  79
  80 def decode(in_file, out_file=None, mode=None):
  81     """Decode uuencoded file"""
  82     #
  83     # Open the input file, if needed.
  84     #
  85     if in_file == '-':
  86         in_file = sys.stdin
  87     elif type(in_file) == type(''):
  88         in_file = open(in_file)
  89     #
  90     # Read until a begin is encountered or we've exhausted the file
  91     #
  92     while 1:
  93         hdr = in_file.readline()
  94         if not hdr:
  95             raise Error, 'No valid begin line found in input file'
  96         if hdr[:5] != 'begin':
  97             continue
  98         hdrfields = string.split(hdr)
  99         if len(hdrfields) == 3 and hdrfields[0] == 'begin':
 100             try:
 101                 string.atoi(hdrfields[1], 8)
 102                 break
 103             except ValueError:
 104                 pass
 105     if out_file == None:
 106         out_file = hdrfields[2]
 107     if mode == None:
 108         mode = string.atoi(hdrfields[1], 8)
 109     #
 110     # Open the output file
 111     #
 112     if out_file == '-':
 113         out_file = sys.stdout
 114     elif type(out_file) == type(''):
 115         fp = open(out_file, 'wb')
 116         try:
 117             os.path.chmod(out_file, mode)
 118         except AttributeError:
 119             pass
 120         out_file = fp
 121     #
 122     # Main decoding loop
 123     #
 124     s = in_file.readline()
 125     while s and s != 'end\n':
 126         try:
 127             data = binascii.a2b_uu(s)
 128         except binascii.Error, v:
 129             # Workaround for broken uuencoders by /Fredrik Lundh
 130             nbytes = (((ord(s[0])-32) & 63) * 4 + 5) / 3
 131             data = binascii.a2b_uu(s[:nbytes])
 132             sys.stderr.write("Warning: %s\n" % str(v))
 133         out_file.write(data)
 134         s = in_file.readline()
 135     if not str:
 136         raise Error, 'Truncated input file'
 137
 138 def test():
 139     """uuencode/uudecode main program"""
 140     import getopt
 141
 142     dopt = 0
 143     topt = 0
 144     input = sys.stdin
 145     output = sys.stdout
 146     ok = 1
 147     try:
 148         optlist, args = getopt.getopt(sys.argv[1:], 'dt')
 149     except getopt.error:
 150         ok = 0
 151     if not ok or len(args) > 2:
 152         print 'Usage:', sys.argv[0], '[-d] [-t] [input [output]]'
 153         print ' -d: Decode (in stead of encode)'
 154         print ' -t: data is text, encoded format unix-compatible text'
 155         sys.exit(1)
 156
 157     for o, a in optlist:
 158         if o == '-d': dopt = 1
 159         if o == '-t': topt = 1
 160
 161     if len(args) > 0:
 162         input = args[0]
 163     if len(args) > 1:
 164         output = args[1]
 165
 166     if dopt:
 167         if topt:
 168             if type(output) == type(''):
 169                 output = open(output, 'w')
 170             else:
 171                 print sys.argv[0], ': cannot do -t to stdout'
 172                 sys.exit(1)
 173         decode(input, output)
 174     else:
 175         if topt:
 176             if type(input) == type(''):
 177                 input = open(input, 'r')
 178             else:
 179                 print sys.argv[0], ': cannot do -t from stdin'
 180                 sys.exit(1)
 181         encode(input, output)
 182
 183 if __name__ == '__main__':
 184     test()