【學習】大文件MD5校驗

將大文件分塊讀取到內存中，每次讀取的內容通過update()更新校驗值，最終獲得完整文件的md5校驗值

#!python3
#coding:utf-8

'''
This module is coded by Yuning Ju

Copyright 2018 Yuning Ju

實現：大文件的MD5校驗

'''
import hashlib,time


def md5_check(fh):
    '''
    Calculate the md5 of files
    將文件分塊讀入內存，防止內存溢出
    
    '''
    
    fp.seek(0)
    fr = fp.read(8096)
    while fr:
        yield fr
        fr = fp.read(8096)
    else:
        fp.seek(0)


if __name__ == '__main__':
    start = time.clock()
    filename = input('請輸入文件路徑：')
    m = hashlib.md5()
    with open(filename,'rb') as fp:
        for fr in md5_check(fp):
            m.update(fr)    #逐次更新校驗值
    end = time.clock()
    print(m.hexdigest(),'\n用時: %.3f s' %(end-start))