caff-previewer-wrapper/caff_previewer_wrapper/utils.py

38 lines
932 B
Python
Raw Normal View History

2020-11-27 05:44:32 +01:00
from flask import request, current_app
import werkzeug.exceptions
import hashlib
def write_file_to_fd_while_calculating_md5(fd: int) -> str:
chunksize = current_app.config['RECIEVE_CHUNKSIZE']
m = hashlib.md5()
total_recieved = 0
# Begin recieving the file
with open(fd, "bw") as f:
while True: # This is where uploading happens
chunk = request.stream.read(chunksize)
if len(chunk) == 0:
break
total_recieved += len(chunk)
if total_recieved > current_app.config['MAX_RECIEVE_SIZE']:
raise werkzeug.exceptions.RequestEntityTooLarge()
m.update(chunk)
f.write(chunk)
return m.hexdigest()
def create_md5_sum_for_file(fname):
m = hashlib.md5()
with open(fname, "rb") as f:
for chunk in iter(lambda: f.read(4096), b""):
m.update(chunk)
return m.hexdigest()