home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / pytho152.zip / emx / lib / python1.5 / uu.py < prev    next >
Text File  |  2000-08-10  |  6KB  |  185 lines
  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()
  185.