Spaces:
Runtime error
Runtime error
File size: 2,313 Bytes
21e6506 |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
from __future__ import annotations
import base64
import contextlib
import os
import tempfile
import time
import colorama
__all__ = (
"timed",
"file_to_data_uri",
)
@contextlib.contextmanager
def timed(msg):
start = time.perf_counter()
print(colorama.Style.DIM + f"╔ {msg}" + colorama.Style.RESET_ALL)
yield
elapsed = time.perf_counter() - start
print(
colorama.Style.DIM + f"╚ Finished in {elapsed:.3f}s" + colorama.Style.RESET_ALL
)
def file_to_data_uri(file_path, mime_type):
with open(file_path, "rb") as file:
encoded_string = base64.b64encode(file.read()).decode("utf-8")
return f"data:{mime_type};base64,{encoded_string}"
class NamedTemporaryFile(contextlib.AbstractContextManager):
"""
tempfile.NamedTemporaryFile with an additional `delete_on_close` parameter.
The `delete_on_close` parameter was only added in Python 3.12, but we badly
need it on Windows: because file access on Windows is exclusive, we can't
write to and then read from a file without closing it in between. But
without `delete_on_close`, the file is deleted on close.
This class is a thin shim around tempfile.NamedTemporaryFile that adds the
parameter for older Python versions.
"""
def __init__(
self,
mode: str = "w+b",
buffering: int = -1,
encoding: str | None = None,
newline: str | None = None,
suffix: str = "",
prefix: str = "tmp",
dir: str | None = None,
delete: bool = True,
*,
errors: str | None = None,
delete_on_close: bool = True,
):
self._needs_manual_delete = delete and not delete_on_close
self._file = tempfile.NamedTemporaryFile(
mode=mode,
buffering=buffering,
encoding=encoding,
newline=newline,
suffix=suffix,
prefix=prefix,
dir=dir,
delete=delete and delete_on_close,
errors=errors,
)
def __enter__(self) -> tempfile.NamedTemporaryFile:
return self._file.__enter__()
def __exit__(self, exc_type, exc_val, exc_tb):
self._file.__exit__(exc_type, exc_val, exc_tb)
if self._needs_manual_delete:
os.unlink(self._file.name)
|