代码语言：javascript复制
#coding: GBK
#md5SHA1文件校验值计算
#使用方法： python 文件.py 验证文件
import hashlib
import sys
import os
defmd5_sum(filename): #校验值方法
fd=open(filename,"rb") #打开文件
fd.seek(0) #将文件打操作标记移到offset的位置
line=fd.readline() #读取文件第一行进入line
#md5校验值计算
md5=hashlib.md5()
md5.update(line)
#SHA1校验值计算
sha1=hashlib.sha1()
sha1.update(line)
while line: #循环读取文件
line=fd.readline()
md5.update(line)
sha1.update(line)
fmd5=md5.hexdigest() #生成文件MD5校验值
fsha1=sha1.hexdigest()#生成文件SHA1校验值
fsum=[fmd5,fsha1]
fd.close()
return fsum
defm_help(): #程序帮助方法
st_help='''
****************************************
** MD5 SHA1 校验使用帮助
** 输入格式 python md5.py file
** Example: python md5.py d:/1/1.txt
****************************************'''
print st_help
if __name__ =="__main__":
try:
filename=sys.argv[1] #获取输入值
if filename=='--help' orfilename=='': #进入帮助菜单
m_help()
elif os.path.isfile(filename): #进入文件校验菜单
fmd5=md5_sum(filename)
print 'File: {}'.format(filename)
print 'MD5 : {}'.format(fmd5[0])
print 'SHA1: {}'.format(fmd5[1])
else: #进入错误菜单
er='''
==========输入错误====================='''
print er
m_help()
except Exception:
m_help()
