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