Merge pull request #10 from AyumuKasuga/patch-1

Add compatibility with celery 4 #9
dealertrack · Mar 15, 2017 · 4999de1 · 4999de1
2 parents b755ff5 + 1696912
commit 4999de1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/celery_redis_sentinel/register.py b/celery_redis_sentinel/register.py
@@ -1,7 +1,12 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, print_function, unicode_literals
+import celery
+
+if celery.VERSION.major < 4:
+    from celery.backends import BACKEND_ALIASES
+else:
+    from celery.app.backends import BACKEND_ALIASES
 
-from celery.backends import BACKEND_ALIASES
 from kombu.transport import TRANSPORT_ALIASES
 
 from .backend import RedisSentinelBackend