fixes bad hash initialization

pull/189/head
msramalho 2025-02-10 16:45:28 +00:00
rodzic 15abf686b1
commit ab6cf52533
Nie znaleziono w bazie danych klucza dla tego podpisu
2 zmienionych plików z 8 dodań i 7 usunięć

Wyświetl plik

@ -30,10 +30,10 @@ class HashEnricher(Enricher):
to_enrich.media[i].set("hash", f"{self.algorithm}:{hd}")
def calculate_hash(self, filename) -> str:
hash = None
hash_algo = None
if self.algorithm == "SHA-256":
hash = hashlib.sha256()
hash_algo = hashlib.sha256
elif self.algorithm == "SHA3-512":
hash = hashlib.sha3_512()
hash_algo = hashlib.sha3_512
else: return ""
return calculate_file_hash(filename, hash, self.chunksize)
return calculate_file_hash(filename, hash_algo, self.chunksize)

Wyświetl plik

@ -65,10 +65,11 @@ def json_loader(cli_val):
return json.loads(cli_val)
def calculate_file_hash(filename: str, hash_algo = hashlib.sha256(), chunksize: int = 16000000) -> str:
def calculate_file_hash(filename: str, hash_algo = hashlib.sha256, chunksize: int = 16000000) -> str:
hash = hash_algo()
with open(filename, "rb") as f:
while True:
buf = f.read(chunksize)
if not buf: break
hash_algo.update(buf)
return hash_algo.hexdigest()
hash.update(buf)
return hash.hexdigest()