From 959fb4bb4ec50af4bc28a4de1455d25ab82b10e8 Mon Sep 17 00:00:00 2001 From: Frederick Muriuki Muriithi Date: Thu, 19 Dec 2024 15:08:10 -0600 Subject: Make files into a module of its own. --- uploader/files.py | 49 ------------------------------------------------- 1 file changed, 49 deletions(-) delete mode 100644 uploader/files.py (limited to 'uploader/files.py') diff --git a/uploader/files.py b/uploader/files.py deleted file mode 100644 index d37a53e..0000000 --- a/uploader/files.py +++ /dev/null @@ -1,49 +0,0 @@ -"""Utilities to deal with uploaded files.""" -import hashlib -from pathlib import Path -from typing import Iterator -from datetime import datetime - -from flask import current_app - -from werkzeug.utils import secure_filename -from werkzeug.datastructures import FileStorage - -def save_file(fileobj: FileStorage, upload_dir: Path) -> Path: - """Save the uploaded file and return the path.""" - assert bool(fileobj), "Invalid file object!" - hashed_name = hashlib.sha512( - f"{fileobj.filename}::{datetime.now().isoformat()}".encode("utf8") - ).hexdigest() - filename = Path(secure_filename(hashed_name)) # type: ignore[arg-type] - if not upload_dir.exists(): - upload_dir.mkdir() - - filepath = Path(upload_dir, filename) - fileobj.save(filepath) - return filepath - - -def fullpath(filename: str): - """Get a file's full path. This makes use of `flask.current_app`.""" - return Path(current_app.config["UPLOAD_FOLDER"], filename).absolute() - - -def chunked_binary_read(filepath: Path, chunksize: int = 2048) -> Iterator: - """Read a file in binary mode in chunks.""" - with open(filepath, "rb") as inputfile: - while True: - data = inputfile.read(chunksize) - if data != b"": - yield data - continue - break - - -def sha256_digest_over_file(filepath: Path) -> str: - """Compute the sha256 digest over a file's contents.""" - filehash = hashlib.sha256() - for chunk in chunked_binary_read(filepath): - filehash.update(chunk) - - return filehash.hexdigest() -- cgit 1.4.1