Python怎么实现文件MD5码批量存储-编程学习网

这篇文章将为大家详细讲解有关Python怎么实现文件MD5码批量存储，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

Python实现文件MD5码批量存储

需求概要

批量计算一组文件的MD5码，并将其存储在数据库或其他永久存储中。

方法

1. 使用hashlib模块计算MD5码

import hashlib

def calculate_md5(path):
    with open(path, "rb") as f:
        md5_hash = hashlib.md5()
        while True:
            data = f.read(1024)
            if not data:
                break
            md5_hash.update(data)
    return md5_hash.hexdigest()

2. 使用多线程或多进程提高计算效率

from concurrent.futures import ThreadPoolExecutor, ProcessPoolExecutor

def calculate_md5_concurrently(paths, max_workers=10):
    if max_workers < 1:
        raise ValueError("Max workers must be greater than 0")

    with ThreadPoolExecutor(max_workers) if max_workers < len(paths) else ProcessPoolExecutor() as executor:
        md5_hashes = list(executor.map(calculate_md5, paths))
    return md5_hashes

3. 使用数据库或其他存储机制存储MD5码

import sqlite3

def store_md5_codes(md5_hashes, database_path):
    conn = sqlite3.connect(database_path)
    cursor = conn.cursor()
    for path, md5_hash in md5_hashes:
        cursor.execute("INSERT INTO md5_codes (path, md5_hash) VALUES (?, ?)", (path, md5_hash))
    conn.commit()
    conn.close()

4. 完整示例

import hashlib
from concurrent.futures import ThreadPoolExecutor
import sqlite3

def calculate_and_store_md5_codes(paths, database_path, max_workers=10):
    md5_hashes = calculate_md5_concurrently(paths, max_workers)
    store_md5_codes(md5_hashes, database_path)

paths = ["path/to/file1.txt", "path/to/file2.txt", ...]
database_path = "path/to/md5_codes.db"
calculate_and_store_md5_codes(paths, database_path)

优点