Question

我试图编写一个脚本，为指定的根目录中的所有文件名和目录名生成哈希值。到目前为止，这是我的脚本：

import hashlib
import os
import sys

class Hasher:
    def __init__(self):
        self.hash_func = hashlib.md5()

    def hash_file(self, file_path):
        with open(file_path, "rb") as file:
            self.hash_func.update(file.read())
        return self.hash_func.digest()

    def hash_dir(self, dir_path):
        for dirpath, dirnames, filenames in os.walk(dir_path):
            self.hash_func.update(dirpath.encode("utf-8"))
            for file_path in filenames:
                self.hash_func.update(file_path.encode("utf-8"))
        return self.hash_func.digest()

hasher = Hasher()
root_dir = "D:/folder/"
hash_1 = str(hasher.hash_dir(root_dir))
hash_2 = str(hasher.hash_dir(root_dir))
print(hash_1)
print(hash_2)

由于某种原因，它为同一目录生成两个不同的哈希值，而无需对目录进行任何更改。如果目录保持不变，我怎样才能生成相同的哈希值？

Answer 1

问题是hashlib.md5对象每次都被重用，因此您不仅返回最后/预期的数据，还返回累积数据的哈希值。

您可以通过每次创建一个新的Hasher对象来解决此问题（在这种情况下，请两次调用Hasher().hash_dir(root_dir)）。但是，由于您的Hasher类除包含md5对象和两个可能是静态的方法外，不包含任何其他数据，因此建议您将这两个类方法都设为静态，并创建hashlib.md5对象在方法本身中：

import hashlib
import os


class Hasher:

    @staticmethod  # make it a static method
    def hash_file(file_path):  # no 'self' as first argument
        hash_func = hashlib.md5()  # create the hashlib.md5 object here
        with open(file_path, "rb") as file:
            hash_func.update(file.read())
        return hash_func.digest()

    @staticmethod  # make it a static method
    def hash_dir(dir_path):  # no 'self' as first argument
        hash_func = hashlib.md5()  # create the hashlib.md5 object here
        for dirpath, _, filenames in os.walk(dir_path):
            hash_func.update(dirpath.encode("utf-8"))
            for file_path in filenames:
                hash_func.update(file_path.encode("utf-8"))
        return hash_func.digest()


def main():
    root_dir = "D:/folder/"
    hash_1 = str(Hasher.hash_dir(root_dir))
    hash_2 = str(Hasher.hash_dir(root_dir))
    print(hash_1)
    print(hash_2)


if __name__ == "__main__":
    main()

Hashlib的md5为相同的输入产生不同的输出

1 个答案: