|
|
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() |
|
|
|
