refactor(backend): remove BaseBackend

Author: Rizhiy

class_cache/__init__.py

@@ -1,7 +1,7 @@
 """Caching for class based generators"""
 
-from .backends import BaseBackend
 from .core import Cache, CacheWithDefault
+from .types import CacheInterface
 from .wrappers import BaseWrapper
 
 __version__ = "0.7.0"

class_cache/backends.py

@@ -18,11 +18,7 @@
 LOGGER = logging.getLogger(__name__)
 
 
-class BaseBackend(CacheInterface[KeyType, ValueType]):
-    pass
-
-
-class PickleBackend(BaseBackend[KeyType, ValueType]):
+class PickleBackend(CacheInterface[KeyType, ValueType]):
     ROOT_DIR = get_class_cache_dir() / "PickleBackend"
     BLOCK_SUFFIX = ".block.pkl"
     META_TYPE = dict[str, Any]
@@ -163,7 +159,7 @@ def clear(self) -> None:
             self._write_clean_meta()
 
 
-class SQLiteBackend(BaseBackend[KeyType, ValueType]):
+class SQLiteBackend(CacheInterface[KeyType, ValueType]):
     ROOT_DIR = get_class_cache_dir() / "SQLiteBackend"
     ROOT_DIR.mkdir(parents=True, exist_ok=True)
     DATA_TABLE_NAME = "data"

class_cache/core.py

@@ -3,14 +3,14 @@
 
 from replete.consistent_hash import consistent_hash
 
-from class_cache.backends import BaseBackend, PickleBackend
+from class_cache.backends import PickleBackend
 from class_cache.types import CacheInterface, KeyType, ValueType
 
 DEFAULT_BACKEND_TYPE = PickleBackend
 
 
 class Cache(CacheInterface[KeyType, ValueType]):
-    def __init__(self, id_: str | int | None = None, backend_type: type[BaseBackend] = DEFAULT_BACKEND_TYPE) -> None:
+    def __init__(self, id_: str | int | None = None, backend_type: type[CacheInterface] = DEFAULT_BACKEND_TYPE) -> None:
         super().__init__(id_)
         self._backend = backend_type(id_)
         # TODO: Implement max_size logic
@@ -19,7 +19,7 @@ def __init__(self, id_: str | int | None = None, backend_type: type[BaseBackend]
         self._to_delete = set()
 
     @property
-    def backend(self) -> BaseBackend:
+    def backend(self) -> CacheInterface:
         return self._backend
 
     def __contains__(self, key: KeyType) -> bool:
@@ -71,7 +71,7 @@ class CacheWithDefault(Cache[KeyType, ValueType]):
         {"_backend", "_backend_set", "_data", "_to_write", "_to_delete"},
     )
 
-    def __init__(self, backend: type[BaseBackend] = DEFAULT_BACKEND_TYPE):
+    def __init__(self, backend: type[CacheInterface] = DEFAULT_BACKEND_TYPE):
         super().__init__(self.id_for_backend, backend)
         self._backend_set = True
 

tests/test_backends.py

@@ -3,13 +3,14 @@
 
 import pytest
 
-from class_cache.backends import BaseBackend, PickleBackend, SQLiteBackend
+from class_cache.backends import PickleBackend, SQLiteBackend
+from class_cache.types import CacheInterface
 
 MAX_WORKERS = 16
 INCREASE_AMOUNT = 32
 
 
-def _increase_cache(backend_type: type[BaseBackend], offset: int):
+def _increase_cache(backend_type: type[CacheInterface], offset: int):
     backend = backend_type()
     for idx in range(INCREASE_AMOUNT):
         num = offset + idx
@@ -18,7 +19,7 @@ def _increase_cache(backend_type: type[BaseBackend], offset: int):
 
 @pytest.mark.parametrize(("backend_type"), [PickleBackend, SQLiteBackend])
 class TestCore:
-    def test_basic(self, test_id, test_key, test_value, backend_type: type[BaseBackend]):
+    def test_basic(self, test_id, test_key, test_value, backend_type: type[CacheInterface]):
         backend = backend_type(test_id)
         backend.clear()
         assert test_key not in backend
@@ -30,7 +31,7 @@ def test_basic(self, test_id, test_key, test_value, backend_type: type[BaseBacke
         del backend[test_key]
         assert test_key not in backend
 
-    def test_write_read(self, test_id, test_key, test_value, backend_type: type[BaseBackend]):
+    def test_write_read(self, test_id, test_key, test_value, backend_type: type[CacheInterface]):
         write_backend = backend_type(test_id)
         write_backend.clear()
         assert test_key not in write_backend