home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / SUM5.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  2.2 KB  |  98 lines
  1. #! /usr/bin/env python
  2.  
  3. # print md5 checksum for files
  4.  
  5. bufsize = 8096
  6. fnfilter = None
  7. rmode = 'r'
  8.  
  9. usage = """
  10. usage: sum5 [-b] [-t] [-l] [-s bufsize] [file ...]
  11. -b        : read files in binary mode
  12. -t        : read files in text mode (default)
  13. -l        : print last pathname component only
  14. -s bufsize: read buffer size (default %d)
  15. file ...  : files to sum; '-' or no files means stdin
  16. """ % bufsize
  17.  
  18. import sys
  19. import string
  20. import os
  21. import md5
  22. import regsub
  23.  
  24. StringType = type('')
  25. FileType = type(sys.stdin)
  26.  
  27. def sum(*files):
  28.     sts = 0
  29.     if files and type(files[-1]) == FileType:
  30.         out, files = files[-1], files[:-1]
  31.     else:
  32.         out = sys.stdout
  33.     if len(files) == 1 and type(files[0]) != StringType:
  34.         files = files[0]
  35.     for f in files:
  36.         if type(f) == StringType:
  37.             if f == '-':
  38.                 sts = printsumfp(sys.stdin, '<stdin>', out) or sts
  39.             else:
  40.                 sts = printsum(f, out) or sts
  41.         else:
  42.             sts = sum(f, out) or sts
  43.     return sts
  44.  
  45. def printsum(file, out = sys.stdout):
  46.     try:
  47.         fp = open(file, rmode)
  48.     except IOError, msg:
  49.         sys.stderr.write('%s: Can\'t open: %s\n' % (file, msg))
  50.         return 1
  51.     if fnfilter:
  52.         file = fnfilter(file)
  53.     sts = printsumfp(fp, file, out)
  54.     fp.close()
  55.     return sts
  56.  
  57. def printsumfp(fp, file, out = sys.stdout):
  58.     m = md5.md5()
  59.     try:
  60.         while 1:
  61.             data = fp.read(bufsize)
  62.             if not data: break
  63.             m.update(data)
  64.     except IOError, msg:
  65.         sys.stderr.write('%s: I/O error: %s\n' % (file, msg))
  66.         return 1
  67.     out.write('%s %s\n' % (hexify(m.digest()), file))
  68.     return 0
  69.  
  70. def hexify(s):
  71.     res = ''
  72.     for c in s:
  73.         res = res + '%02x' % ord(c)
  74.     return res
  75.  
  76. def main(args = sys.argv[1:], out = sys.stdout):
  77.     global fnfilter, rmode, bufsize
  78.     import getopt
  79.     try:
  80.         opts, args = getopt.getopt(args, 'blts:')
  81.     except getopt.error, msg:
  82.         sys.stderr.write('%s: %s\n%s' % (sys.argv[0], msg, usage))
  83.         return 2
  84.     for o, a in opts:
  85.         if o == '-l':
  86.             fnfilter = os.path.basename
  87.         if o == '-b':
  88.             rmode = 'rb'
  89.         if o == '-t':
  90.             rmode = 'r'
  91.         if o == '-s':
  92.             bufsize = string.atoi(a)
  93.     if not args: args = ['-']
  94.     return sum(args, out)
  95.  
  96. if __name__ == '__main__' or __name__ == sys.argv[0]:
  97.     sys.exit(main(sys.argv[1:], sys.stdout))
  98.