49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import json
|
|
import pathlib
|
|
import typing
|
|
|
|
T = typing.TypeVar("T")
|
|
|
|
|
|
class Cache:
|
|
__slots__ = ["__data", "__filepath", "__no_load"]
|
|
|
|
def __init__(self, filepath: str, no_load: bool) -> None: # noqa: FBT001
|
|
self.__filepath = pathlib.Path(filepath)
|
|
self.__no_load = no_load
|
|
self.__data: dict[str, typing.Any] | None = None
|
|
|
|
def load(self) -> None:
|
|
if not self.__no_load and self.__filepath.exists():
|
|
self.__data = json.loads(self.__filepath.read_text())
|
|
else:
|
|
self.__data = {}
|
|
|
|
def write(self) -> None:
|
|
self.__filepath.write_text(json.dumps(self.__data))
|
|
|
|
def __enter__(self) -> typing.Self:
|
|
self.load()
|
|
return self
|
|
|
|
def __exit__(self, *_: object) -> None:
|
|
self.write()
|
|
|
|
def has(self, key: str) -> bool:
|
|
return self.__data is not None and key in self.__data
|
|
|
|
def get(self, key: str) -> typing.Any:
|
|
if self.__data is not None and key in self.__data:
|
|
return self.__data[key]
|
|
return None
|
|
|
|
def set(self, key: str, value: typing.Any) -> None:
|
|
if self.__data is not None:
|
|
self.__data[key] = value
|
|
self.write()
|
|
|
|
def get_or_set(self, key: str, callback: typing.Callable[[], T]) -> T:
|
|
if not self.has(key):
|
|
self.set(key, callback())
|
|
return self.get(key)
|