kysoft-github
diff --git a/‎feapder/db/redisdb.py‎
Lines changed: 7 additions & 3 deletions b/‎feapder/db/redisdb.py‎
Lines changed: 7 additions & 3 deletions
diff --git a/‎setup.py‎
Lines changed: 1 addition & 1 deletion b/‎setup.py‎
Lines changed: 1 addition & 1 deletion
@@ -6,15 +6,14 @@
 ---------
 @author: Boris
 """
-
+import os
 import time
 
 import redis
 from redis.connection import Encoder as _Encoder
 from redis.exceptions import ConnectionError, TimeoutError
 from redis.exceptions import DataError
 from redis.sentinel import Sentinel
-from rediscluster import RedisCluster
 
 import feapder.setting as setting
 from feapder.utils.log import log
@@ -157,6 +156,12 @@ def get_connect(self):
                         )
 
                     else:
+                        try:
+                            from rediscluster import RedisCluster
+                        except ModuleNotFoundError as e:
+                            log.error('请安装 pip install "feapder[all]"')
+                            os._exit(0)
+
                         # log.debug("使用redis集群模式")
                         self._redis = RedisCluster(
                             startup_nodes=startup_nodes,
@@ -584,7 +589,6 @@ def zexists(self, table, values):
         return is_exists
 
     def lpush(self, table, values):
-
         if isinstance(values, list):
             pipe = self._redis.pipeline()
 
 
@@ -42,7 +42,6 @@
     "requests>=2.22.0",
     "bs4>=0.0.1",
     "ipython>=7.14.0",
-    "redis-py-cluster>=2.1.0",
     "cryptography>=3.3.2",
     "urllib3>=1.25.8",
     "loguru>=0.5.3",
@@ -61,6 +60,7 @@
     "bitarray>=1.5.3",
     "PyExecJS>=1.5.1",
     "pymongo>=3.10.1",
+    "redis-py-cluster>=2.1.0",
 ] + render_requires
 
 setuptools.setup(