生成文件的md5和sha1
#coding=utf-8
# xinghe studio file to md5 and sha1
# 星河创作室(xinghestudio.com)
# create by stream.wang 2012-04-10
# modify by stream.wang 2012-04-10
def fil_changefileext(filename,extname=''):
# fil changefileext 改变文件扩展名
# filename 输入的文件路径名
# extname='' 要更改分扩展名，如 .txt
import os
try:
if extname[0]<>'.':
extname='.' extname
except:
extname=''
if os.path.splitext(filename)[1]=='':
uouttxt=filename '.' extname
elif os.path.splitext(filename)[1]=='.':
uouttxt=filename extname
else:
uouttxt=filename[:0-len(os.path.splitext(filename)[1])] extname
return uouttxt
def fil_str2file(fstr,filename):
# fil str2file 把字符串保存到文件
try:
outfile=open(filename,'w')
outfile.writelines(fstr)
outfile.close()
return true
except:
return false
#系统主程序开始
#-------------------------------------------------------
if __name__ == '__main__':
import os,sys,math
# if len(sys.argv)==1:sys.argv.append(r'c:\freenas-8.0.4-release-x64.iso')
if len(sys.argv)>1:
filepathname=sys.argv[1]
if os.path.isfile(filepathname) :
fmaxs = os.path.getsize(filepathname) 0.00
fpos = 0.00
ppos=0
pmax=50
pjy=0
fmd5=fil_changefileext(filepathname,'.md5')
fsha1=fil_changefileext(filepathname,'.sha')
ffm=os.path.basename(filepathname)
#开始处理
print 'program runing "' filepathname '" ...'
import hashlib
try:
umd5str = hashlib.md5()
usha1str = hashlib.sha1()
xfilepathname=unicode(filepathname,'utf8')
#需要使用二进制格式读取文件内容
ufile = file(xfilepathname,'rb')
while true:
datas = ufile.read(1024*512)
if not datas: break
umd5str.update( datas )
usha1str.update( datas )
fpos=fpos 1024*512
ppos=int(math.floor(fpos / fmaxs * 50))
pposbfh=int(math.floor(fpos / fmaxs * 100))
#打印进度条
if pjy<>ppos: print '[' '*'*ppos '='*(pmax-ppos) '] ' str(pposbfh) '%'
#打印进度问题未解决，不知道print函数怎么退回到行首覆盖输出，不换行我倒是知道，加个“,”
pjy=ppos
uoutstrmd5=str(umd5str.hexdigest())
uoutstrsha1=str(usha1str.hexdigest())
ufile.close()
print ''
fil_str2file(uoutstrmd5 ' *' ffm '\n',fmd5)
fil_str2file(uoutstrsha1 ' *' ffm '\n',fsha1)
print 'task is completed !' r' [ xinghe studio file to md5 and sha1 ]'
except:
print 'error !' r' [ xinghe studio file to md5 and sha1 ]'
else:
print 'file "' filepathname '" not exist!' r' [ xinghe studio file to md5 and sha1 ]'
else:
print r' [ xinghe studio file to md5 and sha1 ]'
葛力姆乔org
