Monkey-patch django cache redis to support locking

django_celery_extensions/beat.py

@@ -1,4 +1,5 @@
 from django.core.cache import caches
+from django.core.cache.backends.redis import RedisCacheClient
 from django.core.exceptions import ImproperlyConfigured
 
 try:
@@ -8,17 +9,24 @@
 except ImportError:
     raise ImproperlyConfigured('Missing celery library, please install it')
 
-try:
-    from redis_cache import RedisCache
-except ImportError:
-    raise ImproperlyConfigured('Missing redis_cache library, please install it')
+from redis.lock import Lock
 
 from .config import settings
 
 
 logger = get_logger(__name__)
 
 
+def monkey_patch_django_redis():
+    def lock(self, key, timeout, sleep):
+        return Lock(self.get_client(), key, timeout, sleep)
+
+    RedisCacheClient.lock = lock
+
+
+monkey_patch_django_redis()
+
+
 class LockedPersistentScheduler(PersistentScheduler):
 
     lock = None
@@ -44,14 +52,10 @@ def close(self):
 @beat_init.connect
 def acquire_distributed_beat_lock(sender=None, **kwargs):
     scheduler = sender.scheduler
-
     cache = caches[settings.CACHE_NAME]
 
-    if not isinstance(cache, RedisCache):
-        raise ImproperlyConfigured('Only redis cache is allowed to use LockedPersistentScheduler')
-
     logger.debug('beat: Acquiring lock...')
-    lock = caches[settings.CACHE_NAME].lock(
+    lock = cache.lock(
         scheduler.lock_key,
         timeout=scheduler.lock_timeout,
         sleep=scheduler.lock_sleep