aboutsummaryrefslogtreecommitdiff
path: root/uploader/files.py
blob: d37a53ea1d71914790b59a84800ccb5e14d5d6f7 (about) (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
"""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()