import hashlib
import os
'''
可以求文件或者字符串的md5
'''
def getMD5_SHA256(filename_or_string, md5_sha256="sha256"):
d = hashlib.md5() if md5_sha256=="md5" else hashlib.sha256()
if os.path.isfile(filename_or_string):
with open(filename_or_string, 'rb') as f:
while True:
buf = f.read(4096)
if not buf:
break
d.update(buf)
return "文件【%s】的【%s】为:%s" % (filename_or_string, md5_sha256, d.hexdigest())
else:
d.update(filename_or_string.encode('utf-8'))
return "字符串【%s】的【%s】为:%s:" % (filename_or_string, md5_sha256, d.hexdigest())
xunlei_path = '/Users/yangbingpeng/Documents/python全栈/模块/hashlib模块/hashlib_study.py'
print(getMD5_SHA256(xunlei_path, "md5"))
print(getMD5_SHA256(xunlei_path, "sha256"))
print(getMD5_SHA256("你好", "md5"))
print(getMD5_SHA256("你好", "sha256"))