35 lines
779 B
Python
35 lines
779 B
Python
import hashlib
|
|
from blake3 import blake3
|
|
from typing import Callable
|
|
|
|
|
|
BLOCKSIZE = 2**10 * 128 # kb
|
|
|
|
|
|
def hashfile(path: str, hash_algo: str, *, hit: Callable[[str], None]):
|
|
if hash_algo == "md5":
|
|
h = hashlib.md5()
|
|
elif hash_algo == "sha512":
|
|
h = hashlib.sha512()
|
|
elif hash_algo == "blake3":
|
|
h = blake3()
|
|
else:
|
|
raise Exception(f"algo '{hash_algo}' is unknown.")
|
|
|
|
with open(path, "rb") as f:
|
|
while True:
|
|
buf = f.read(BLOCKSIZE)
|
|
l = len(buf)
|
|
if l <= 0:
|
|
break
|
|
h.update(buf)
|
|
if hit:
|
|
hit(l)
|
|
return h.hexdigest()
|
|
|
|
|
|
def hashtext(text: str):
|
|
md5 = hashlib.md5()
|
|
md5.update(text.encode("utf-8"))
|
|
return md5.hexdigest()
|