为什么我的校验和计算器停止打印？

Question

我正在尝试编写一个小程序来获取某些文件的校验和。使用提到的代码over here，我做了这个：

import hashlib
import sys
from functools import partial

def md5sum(filename):
    with open(filename, mode='rb') as f:
        d = hashlib.md5()
        for buf in iter(partial(f.read, 128), b''):
            d.update(buf)
    return d.hexdigest()

print(md5sum(sys.argv[0]))

该代码运行良好。但是，当我尝试像这样添加 SHA1 支持时：

import hashlib
import sys
from functools import partial

def md5sum(filename):
    with open(filename, mode='rb') as f:
        d = hashlib.md5()
        for buf in iter(partial(f.read, 128), b''):
            d.update(buf)
    return d.hexdigest()

def sha1sum(filename):
    with open(filename, mode='rb') as f:
        d = hashlib.sha1()
        for buf in iter(partial(f.read, 128), b''):
            d.update(buf)
    return d.hexdigest()

if sys.argv[0] == 'md5':
    print(md5sum(sys.argv[1]))
elif sys.argv[0] == 'sha1':
    print(sha1sum(sys.argv[1]))

它停止打印输出（对于 MD5 和 SHA1）。我在这里做错了什么？

Answer 1

sys.argv[0] 是脚本的名称；命令行参数存储在sys.argv[1:] 中。您可能想改为测试 sys.argv[1]：

if sys.argv[1] == 'md5':
    print(md5sum(sys.argv[2]))
elif sys.argv[1] == 'sha1':
    print(sha1sum(sys.argv[2]))

请注意，您原来的“工作”版本只计算了脚本本身的 MD5 哈希值。