Python写自动化之获取文件的MD5值

使用Python 获取文件的MD5 值是一件很简单的事情,Python 提供了md5 和 hashlib 两个模块,都可以获取到文件的md5值。

代码如下:

#获取文件的MD5值,适用于小文件
    def getFileMD5(self,filepath):
        if self.isFile(filepath):
            f = open(filepath,'rb')
            md5obj = hashlib.md5()
            md5obj.update(f.read())
            hash = md5obj.hexdigest()
            f.close()
            return str(hash).upper()
        return None

上面的代码已经适用与获取绝大多数文件的md5值。


然而,仔细看代码,就会发现,在获取md5值时,实际上有一个读文件的过程。这样就会有一个问题,倘若文件非常大,大于机器内存的时候,上面的代码就会有问题,解决方法是可以分多次读取,代码如下:

#获取文件的MD5值,适用于较大的文件
    def getBigFileMD5(self,filepath):
        if self.isFile(filepath):
            md5obj = hashlib.md5()
            maxbuf = 8192
            f = open(filepath,'rb')
            while True:
                buf = f.read(maxbuf)
                if not buf:
                    break
                md5obj.update(buf)
            f.close()
            hash = md5obj.hexdigest()
            return str(hash).upper()
        return None

另外,需要注意的是:在python 3.2 版本中已经取消了 md5 模块,因此在获取md5 时,建议采用 hashlib 模块。


转载请注明出处:http://blog.csdn.net/sogouauto

你可能感兴趣的:(测试脚本)