mirror of
https://github.com/python/cpython.git
synced 2024-11-24 02:15:30 +08:00
54ac832a24
This patch changes a few of the scripts to have __name__=='__main__' clauses so that they are importable without running. Also fixes the syntax errors revealed by the tests.
91 lines
2.3 KiB
Python
Executable File
91 lines
2.3 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
"""Python utility to print MD5 checksums of argument files.
|
|
"""
|
|
|
|
|
|
bufsize = 8096
|
|
fnfilter = None
|
|
rmode = 'rb'
|
|
|
|
usage = """
|
|
usage: sum5 [-b] [-t] [-l] [-s bufsize] [file ...]
|
|
-b : read files in binary mode (default)
|
|
-t : read files in text mode (you almost certainly don't want this!)
|
|
-l : print last pathname component only
|
|
-s bufsize: read buffer size (default %d)
|
|
file ... : files to sum; '-' or no files means stdin
|
|
""" % bufsize
|
|
|
|
import sys
|
|
import os
|
|
import getopt
|
|
from hashlib import md5
|
|
|
|
def sum(*files):
|
|
sts = 0
|
|
if files and isinstance(files[-1], file):
|
|
out, files = files[-1], files[:-1]
|
|
else:
|
|
out = sys.stdout
|
|
if len(files) == 1 and not isinstance(files[0], str):
|
|
files = files[0]
|
|
for f in files:
|
|
if isinstance(f, str):
|
|
if f == '-':
|
|
sts = printsumfp(sys.stdin, '<stdin>', out) or sts
|
|
else:
|
|
sts = printsum(f, out) or sts
|
|
else:
|
|
sts = sum(f, out) or sts
|
|
return sts
|
|
|
|
def printsum(filename, out=sys.stdout):
|
|
try:
|
|
fp = open(filename, rmode)
|
|
except IOError as msg:
|
|
sys.stderr.write('%s: Can\'t open: %s\n' % (filename, msg))
|
|
return 1
|
|
if fnfilter:
|
|
filename = fnfilter(filename)
|
|
sts = printsumfp(fp, filename, out)
|
|
fp.close()
|
|
return sts
|
|
|
|
def printsumfp(fp, filename, out=sys.stdout):
|
|
m = md5.new()
|
|
try:
|
|
while 1:
|
|
data = fp.read(bufsize)
|
|
if not data:
|
|
break
|
|
m.update(data)
|
|
except IOError as msg:
|
|
sys.stderr.write('%s: I/O error: %s\n' % (filename, msg))
|
|
return 1
|
|
out.write('%s %s\n' % (m.hexdigest(), filename))
|
|
return 0
|
|
|
|
def main(args = sys.argv[1:], out=sys.stdout):
|
|
global fnfilter, rmode, bufsize
|
|
try:
|
|
opts, args = getopt.getopt(args, 'blts:')
|
|
except getopt.error as msg:
|
|
sys.stderr.write('%s: %s\n%s' % (sys.argv[0], msg, usage))
|
|
return 2
|
|
for o, a in opts:
|
|
if o == '-l':
|
|
fnfilter = os.path.basename
|
|
elif o == '-b':
|
|
rmode = 'rb'
|
|
elif o == '-t':
|
|
rmode = 'r'
|
|
elif o == '-s':
|
|
bufsize = int(a)
|
|
if not args:
|
|
args = ['-']
|
|
return sum(args, out)
|
|
|
|
if __name__ == '__main__' or __name__ == sys.argv[0]:
|
|
sys.exit(main(sys.argv[1:], sys.stdout))
|