diff options
Diffstat (limited to 'scripts/internal/print_hashes.py')
-rwxr-xr-x | scripts/internal/print_hashes.py | 26 |
1 files changed, 16 insertions, 10 deletions
diff --git a/scripts/internal/print_hashes.py b/scripts/internal/print_hashes.py index 434c43d5..69bc9edb 100755 --- a/scripts/internal/print_hashes.py +++ b/scripts/internal/print_hashes.py @@ -5,13 +5,13 @@ # found in the LICENSE file. """ -Prints files hashes. -See: https://pip.pypa.io/en/stable/reference/pip_install/#hash-checking-mode +Prints files hashes, see: +https://pip.pypa.io/en/stable/reference/pip_install/#hash-checking-mode """ +import argparse import hashlib import os -import sys def csum(file, kind): @@ -22,13 +22,19 @@ def csum(file, kind): def main(): - dir = sys.argv[1] - for name in sorted(os.listdir(dir)): - file = os.path.join(dir, name) - md5 = csum(file, "md5") - sha256 = csum(file, "sha256") - print("%s\nmd5: %s\nsha256: %s\n" % ( - os.path.basename(file), md5, sha256)) + parser = argparse.ArgumentParser(description=__doc__) + parser.add_argument('dir', type=str, + help='directory containing tar.gz or wheel files') + args = parser.parse_args() + for name in sorted(os.listdir(args.dir)): + file = os.path.join(args.dir, name) + if os.path.isfile(file): + md5 = csum(file, "md5") + sha256 = csum(file, "sha256") + print("%s\nmd5: %s\nsha256: %s\n" % ( + os.path.basename(file), md5, sha256)) + else: + print("skipping %r (not a file)" % file) if __name__ == "__main__": |