chore: replace diskcache with diskcache2 to improve typing

main.py

@@ -6,7 +6,7 @@
 from pydantic_core import Url
 
 from src.api import get
-from src.utils.db import TypedDB
+from src.utils.db import Cache
 
 app = FastAPI()
 
@@ -26,7 +26,7 @@ def get_url(url: Url):
     return str(url)
 
 
-db = TypedDB[Response]("v1")
+db = Cache[Response]("v1")
 
 
 @app.get("/proxy")

pyproject.toml

@@ -5,7 +5,7 @@ requires-python = ">=3.12"
 dependencies = [
     "brotli-asgi~=1.4.0",
     "curl-cffi~=0.7.3",
-    "diskcache~=5.6.3",
+    "diskcache2~=0.1.1",
     "fake-useragent~=1.5.1",
     "fastapi~=0.115.5",
     "httptools~=0.6.4",

src/utils/db.py

@@ -1,19 +1,12 @@
 from pathlib import Path
 
-cache_root = Path(__file__, "../../../cache").resolve()
+from diskcache import Cache as BaseCache
 
+cache_root = Path(__file__, "../../../cache").resolve()
 
-class TypedDB[T]:
-    def __init__(self, name: str):
-        from diskcache import Cache
 
-        path = cache_root / name
+class Cache[V](BaseCache[str, V]):
+    def __init__(self, directory: str):
+        path = cache_root / directory
         path.mkdir(parents=True, exist_ok=True)
-
-        self.cache = Cache(path)
-
-    def __getitem__(self, key: str) -> T:
-        return self.cache[key]  # type: ignore
-
-    def __setitem__(self, key: str, value: T):
-        self.cache[key] = value
+        super().__init__(path)