Python生成文件的MD5和sha1

生成文件的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 ]'