python学习，计算文件MD5值

# -*- coding: utf-8 -*-

import hashlib
import os

BUFSIZE = 8192

def getFileMD5(strFilePath):
    
    f = open(strFilePath, 'rb')
    m = hashlib.md5()
    try:
        while True:
            data = f.read(BUFSIZE)
            if not data:
                break
            m.update(data)

    finally:
        f.close()

    return m.hexdigest()

def getMD5Info(strPath):

    dictMD5 = {}
    
    if os.path.isfile(strPath):
        dictMD5[strPath] = getFileMD5(strPath)
    else:
        if os.path.isdir(strPath):
            for strDirPath, lstDirName, lstFileName in os.walk(strPath):
                for strFileName in lstFileName:
                    strTempPath = os.path.join(strDirPath, strFileName)
                    dictMD5[strTempPath] = getFileMD5(strTempPath)


    return dictMD5


if __name__=="__main__":

    strFilePath = "C:\\Python27\\ss.py"

    print getFileMD5(strFilePath)

    strDirPath  = "C:\\pkg"
    print getMD5Info(strDirPath)