快捷導(dǎo)航

Python的hashlib庫(kù)獲取超大文件的md5值實(shí)例探究

更新時(shí)間：2024年01月23日 11:15:33 作者：weibin python學(xué)習(xí)與大數(shù)據(jù)分析

這篇文章主要為大家介紹了Python的hashlib庫(kù)獲取超大文件的md5值實(shí)現(xiàn)示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪

Python hashlib獲取文件md5值

不知道大家在工作中有沒(méi)有涉及到超大文件的讀取，在處理超大文件時(shí)，如果直接將整個(gè)文件讀到內(nèi)存，然后再計(jì)算MD5值可能會(huì)占用大量?jī)?nèi)存，嚴(yán)重的情況下可能會(huì)導(dǎo)致系統(tǒng)故障。

為了減輕系統(tǒng)內(nèi)存壓力，我們通常采取分塊讀取文件內(nèi)容的方式來(lái)計(jì)算MD5。

Python的hashlib庫(kù)，它支持將大文件分塊讀取并計(jì)算MD5值，從而避免一次性加載整個(gè)文件到內(nèi)存中。

import hashlib
import os
def get_file_md5(file_path, block_size=2 ** 20):  # 默認(rèn)塊大小為1MB
    md5_hash = hashlib.md5()
    with open(file_path, "rb") as f:
        while True:
            data = f.read(block_size)
            if not data:
                break
            md5_hash.update(data)
    return md5_hash.hexdigest()
# 使用方法
file_path_large_file = r"E:\xxxxxxx\Win10_64位專業(yè)版本.ISO"
md5_value = get_file_md5(file_path_large_file)
print(f"大文件的大小是：{os.path.getsize(file_path_large_file)/1024/1024/1024:.2f}GB")
print(f"大文件的MD5值是：{md5_value}")

測(cè)試5.42G操作系統(tǒng)文件

我自己找了一個(gè)5.42G操作系統(tǒng)文件來(lái)測(cè)試了下，運(yùn)行過(guò)程計(jì)算機(jī)內(nèi)存的使用率并沒(méi)有明顯增加，表現(xiàn)很平穩(wěn)。

內(nèi)存利用率變化

可能大家看到這里不能明顯感受到分塊讀取文件和一次性將大文件讀入內(nèi)存的差異，我下面的函數(shù)去掉了分塊讀取的設(shè)置，我們?cè)賮?lái)看下內(nèi)存利用率的變化

def get_file_md52(file_path):  # 這里去掉了塊大小的設(shè)置
    md5_hash = hashlib.md5()
    with open(file_path, "rb") as f:
        while True:
            data = f.read()  # 這里是一次性將文件讀入內(nèi)存
            if not data:
                break
            md5_hash.update(data)
    return md5_hash.hexdigest()
# 使用方法
file_path_large_file = r"E:\xxxxxx\Win10_64位專業(yè)版本.ISO"
md5_value = get_file_md52(file_path_large_file)
print(f"大文件的大小是：{os.path.getsize(file_path_large_file)/1024/1024/1024:.2f}GB")
print(f"大文件的MD5值是：{md5_value}")