Add cache

curl_cffi/requests/cache.py

@@ -0,0 +1,62 @@
+# Inspired by requests-cache, but much simpler
+
+import sqlite3
+from abc import abstractmethod
+from dataclasses import dataclass
+from typing import Literal
+
+CacheBackend = Literal["memory", "file", "sqlite", "redis"]
+
+
+@dataclass
+class CacheSettings:
+    backend: CacheBackend
+
+
+class BaseBackend:
+    def __init__(self):
+        self.type = type
+
+    @abstractmethod
+    def __getitem__(self, key):
+        raise NotImplementedError()
+
+    @abstractmethod
+    def __setitem__(self, key, value):
+        raise NotImplementedError()
+
+    @abstractmethod
+    def __delitem__(self, key, value):
+        raise NotImplementedError()
+
+    @abstractmethod
+    def __iter__(self, key, value):
+        raise NotImplementedError()
+
+    @abstractmethod
+    def __len__(self, key, value):
+        raise NotImplementedError()
+
+    @abstractmethod
+    def clear(self, key, value):
+        raise NotImplementedError()
+
+
+class MemeoryBackend(BaseBackend):
+    pass
+
+
+class FileBackend(BaseBackend):
+    pass
+
+
+class SqliteBackend(BaseBackend):
+    def __init__(self, filename: str, **kwargs):
+        self.filename = filename
+        self._db = sqlite3.connect(**kwargs)
+
+
+class RedisBackend(BaseBackend):
+    def ___init__(self, host: str, port: int):
+        self.host = host
+        self.port = port

curl_cffi/requests/session.py

@@ -24,10 +24,15 @@
 )
 from urllib.parse import ParseResult, parse_qsl, quote, unquote, urlencode, urljoin, urlparse
 
-from typing_extensions import Unpack
+# Unpack introduced in 3.11
+try:
+    from typing import Unpack
+except ImportError:
+    from typing_extensions import Unpack
 
 from .. import AsyncCurl, Curl, CurlError, CurlHttpVersion, CurlInfo, CurlOpt, CurlSslVersion
 from ..curl import CURL_WRITEFUNC_ERROR, CurlMime
+from .cache import CacheStrategy
 from .cookies import Cookies, CookieTypes, CurlMorsel
 from .exceptions import ImpersonateError, RequestException, SessionClosed, code2error
 from .headers import Headers, HeaderTypes
@@ -221,6 +226,7 @@ def __init__(
         debug: bool = False,
         interface: Optional[str] = None,
         cert: Optional[Union[str, Tuple[str, str]]] = None,
+        cache: Optional[CacheStrategy] = None,
     ):
         self.headers = Headers(headers)
         self.cookies = Cookies(cookies)
@@ -244,6 +250,7 @@ def __init__(
         self.debug = debug
         self.interface = interface
         self.cert = cert
+        self.cache = cache
 
         if proxy and proxies:
             raise TypeError("Cannot specify both 'proxy' and 'proxies'")

pyproject.toml

@@ -7,6 +7,7 @@ license = { file = "LICENSE" }
 dependencies = [
     "cffi>=1.12.0",
     "certifi>=2024.2.2",
+    "typing_extensions>=4.7.0",
 ]
 readme = "README.md"
 requires-python = ">=3.8"