hibiapi/utils/temp.py

from pathlib import Path
from tempfile import mkdtemp, mkstemp
from threading import Lock
from urllib.parse import ParseResult

from fastapi import Request


class TempFile:
    path = Path(mkdtemp())
    path_depth = 3
    name_length = 16

    _lock = Lock()

    @classmethod
    def create(cls, ext: str = ".tmp"):
        descriptor, str_path = mkstemp(suffix=ext, dir=str(cls.path))
        return descriptor, Path(str_path)

    @classmethod
    def to_url(cls, request: Request, path: Path) -> str:
        assert cls.path
        return ParseResult(
            scheme=request.url.scheme,
            netloc=request.url.netloc,
            path=f"/temp/{path.relative_to(cls.path)}",
            params="",
            query="",
            fragment="",
        ).geturl()