← Back to branch summary

~malept/ubuntu/lucid/python2.6/dev-dependency-fix

« back to all changes in this revision

Viewing changes to Tools/scripts/md5sum.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-02-13 12:51:00 UTC
  • Revision ID: james.westby@ubuntu.com-20090213125100-uufgcb9yeqzujpqw
Tags: upstream-2.6.1
Import upstream version 2.6.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
Tools/scripts/md5sum.py
 
1
#! /usr/bin/env python
 
2
 
 
3
"""Python utility to print MD5 checksums of argument files.
 
4
"""
 
5
 
 
6
 
 
7
bufsize = 8096
 
8
fnfilter = None
 
9
rmode = 'rb'
 
10
 
 
11
usage = """
 
12
usage: sum5 [-b] [-t] [-l] [-s bufsize] [file ...]
 
13
-b        : read files in binary mode (default)
 
14
-t        : read files in text mode (you almost certainly don't want this!)
 
15
-l        : print last pathname component only
 
16
-s bufsize: read buffer size (default %d)
 
17
file ...  : files to sum; '-' or no files means stdin
 
18
""" % bufsize
 
19
 
 
20
import sys
 
21
import os
 
22
import getopt
 
23
import md5
 
24
 
 
25
def sum(*files):
 
26
    sts = 0
 
27
    if files and isinstance(files[-1], file):
 
28
        out, files = files[-1], files[:-1]
 
29
    else:
 
30
        out = sys.stdout
 
31
    if len(files) == 1 and not isinstance(files[0], str):
 
32
        files = files[0]
 
33
    for f in files:
 
34
        if isinstance(f, str):
 
35
            if f == '-':
 
36
                sts = printsumfp(sys.stdin, '<stdin>', out) or sts
 
37
            else:
 
38
                sts = printsum(f, out) or sts
 
39
        else:
 
40
            sts = sum(f, out) or sts
 
41
    return sts
 
42
 
 
43
def printsum(filename, out=sys.stdout):
 
44
    try:
 
45
        fp = open(filename, rmode)
 
46
    except IOError, msg:
 
47
        sys.stderr.write('%s: Can\'t open: %s\n' % (filename, msg))
 
48
        return 1
 
49
    if fnfilter:
 
50
        filename = fnfilter(filename)
 
51
    sts = printsumfp(fp, filename, out)
 
52
    fp.close()
 
53
    return sts
 
54
 
 
55
def printsumfp(fp, filename, out=sys.stdout):
 
56
    m = md5.new()
 
57
    try:
 
58
        while 1:
 
59
            data = fp.read(bufsize)
 
60
            if not data:
 
61
                break
 
62
            m.update(data)
 
63
    except IOError, msg:
 
64
        sys.stderr.write('%s: I/O error: %s\n' % (filename, msg))
 
65
        return 1
 
66
    out.write('%s %s\n' % (m.hexdigest(), filename))
 
67
    return 0
 
68
 
 
69
def main(args = sys.argv[1:], out=sys.stdout):
 
70
    global fnfilter, rmode, bufsize
 
71
    try:
 
72
        opts, args = getopt.getopt(args, 'blts:')
 
73
    except getopt.error, msg:
 
74
        sys.stderr.write('%s: %s\n%s' % (sys.argv[0], msg, usage))
 
75
        return 2
 
76
    for o, a in opts:
 
77
        if o == '-l':
 
78
            fnfilter = os.path.basename
 
79
        elif o == '-b':
 
80
            rmode = 'rb'
 
81
        elif o == '-t':
 
82
            rmode = 'r'
 
83
        elif o == '-s':
 
84
            bufsize = int(a)
 
85
    if not args:
 
86
        args = ['-']
 
87
    return sum(args, out)
 
88
 
 
89
if __name__ == '__main__' or __name__ == sys.argv[0]:
 
90
    sys.exit(main(sys.argv[1:], sys.stdout))