reuse Options

Author: timgraham

django_mongodb_backend/cache.py

@@ -2,6 +2,7 @@
 from datetime import datetime, timezone
 
 from django.core.cache.backends.base import DEFAULT_TIMEOUT, BaseCache
+from django.core.cache.backends.db import Options
 from django.db import connections, router
 from django.utils.functional import cached_property
 from pymongo.errors import DuplicateKeyError
@@ -25,26 +26,11 @@ def loads(self, data):
             return pickle.loads(data)  # noqa: S301
 
 
-class Options:
-    """A class that will quack like a Django model _meta class.
-
-    This allows cache operations to be controlled by the router
-    """
-
-    def __init__(self, collection_name):
-        self.collection_name = collection_name
-        self.app_label = "django_cache"
-        self.model_name = "cacheentry"
-        self.verbose_name = "cache entry"
-        self.verbose_name_plural = "cache entries"
-        self.object_name = "CacheEntry"
-        self.abstract = False
-        self.managed = True
-        self.proxy = False
-        self.swapped = False
+class MongoDBCache(BaseCache):
+    # This class uses collection provided by the database connection.
 
+    pickle_protocol = pickle.HIGHEST_PROTOCOL
 
-class BaseDatabaseCache(BaseCache):
     def __init__(self, collection_name, params):
         super().__init__(params)
         self._collection_name = collection_name
@@ -54,12 +40,6 @@ class CacheEntry:
 
         self.cache_model_class = CacheEntry
 
-
-class MongoDBCache(BaseDatabaseCache):
-    # This class uses collection provided by the database connection.
-
-    pickle_protocol = pickle.HIGHEST_PROTOCOL
-
     def create_indexes(self):
         self.collection.create_index("expires_at", expireAfterSeconds=0)
         self.collection.create_index("key", unique=True)

django_mongodb_backend/management/commands/createcachecollection.py

@@ -7,7 +7,7 @@
     router,
 )
 
-from django_mongodb_backend.cache import BaseDatabaseCache, MongoDBCache
+from django_mongodb_backend.cache import MongoDBCache
 
 
 class Command(BaseCommand):
@@ -42,7 +42,7 @@ def handle(self, *collection_names, **options):
         else:
             for cache_alias in settings.CACHES:
                 cache = caches[cache_alias]
-                if isinstance(cache, BaseDatabaseCache):
+                if isinstance(cache, MongoDBCache):
                     self.check_collection(db, cache._collection_name)
 
     def check_collection(self, database, collection_name):