liqian 3 tahun lalu
induk
melakukan
d2df5d28fa
2 mengubah file dengan 16 tambahan dan 8 penghapusan
  1. 2 2
      config.py
  2. 14 6
      db_helper.py

+ 2 - 2
config.py

@@ -179,6 +179,6 @@ class ProductionConfig(BaseConfig):
 
 def set_config():
     # return DevelopmentConfig()
-    # return TestConfig()
+    return TestConfig()
     # return PreProductionConfig()
-    return ProductionConfig()
+    # return ProductionConfig()

+ 14 - 6
db_helper.py

@@ -170,8 +170,15 @@ class RedisHelper(object):
         if not conn.exists(key_name):
             # key不存在
             return None
-        data = conn.sscan(key_name)
-        return data[1]
+        data = []
+        cursor = 0
+        while True:
+            cur, temp = conn.sscan(key_name, cursor=cursor, count=2000)
+            data.extend(temp)
+            if cur == 0:
+                break
+            cursor = cur
+        return list(set(data))
 
     def add_data_with_set(self, key_name, values, expire_time=30*60):
         """
@@ -291,7 +298,8 @@ if __name__ == '__main__':
     # print(len(data))
     # key_name = 'com.weiqu.video.hot.recommend.previewed.4.weixin_openid_otjoB5VG780SB4aVjYqBBNLb - X6M'
     # values = (6134455, 9772930, 9912678, 9901969, 9926876, 9904203, 2384831, 9932272, 9737653, 9925240)
-    key_name = 'com.weiqu.video.hot.recommend.previewed.4.weixin_openid_otjoB5Q2b - PuUKPKVDLXMa62P_zU'
-    values = (9902612, 9905573, 9928264, 9932148, 9809440, 9919900, 6093379, 9917093, 9793537, 9814345)
-    redis_helper.add_data_with_set(key_name=key_name, values=values, expire_time=30 * 60)
-
+    key_name = 'com.weiqu.video.hot.recommend.previewed.4.abcd1'
+    # values = (9902612, 9905573, 9928264, 9932148, 9809440, 9919900, 6093379, 9917093, 9793537, 9814345)
+    # redis_helper.add_data_with_set(key_name=key_name, values=values, expire_time=30 * 60)
+    res = redis_helper.get_data_from_set(key_name=key_name)
+    print(res)