Pārlūkot izejas kodu

Merge pull request #43 from DemSephiroth/patch-1

Update for redis3.0  db.py
崔庆才丨静觅 6 gadi atpakaļ
vecāks
revīzija
329dff07c1
1 mainītis faili ar 9 papildinājumiem un 3 dzēšanām
  1. 9 3
      proxypool/db.py

+ 9 - 3
proxypool/db.py

@@ -27,7 +27,9 @@ class RedisClient(object):
             print('代理不符合规范', proxy, '丢弃')
             return
         if not self.db.zscore(REDIS_KEY, proxy):
-            return self.db.zadd(REDIS_KEY, score, proxy)
+            #return self.db.zadd(REDIS_KEY, score, proxy)
+            #更新为redis3.0+版本,解决redis3.0更新后的报错,如用旧版本还原上方代码
+            return self.db.zadd(REDIS_KEY, {proxy:score})
     
     def random(self):
         """
@@ -53,7 +55,9 @@ class RedisClient(object):
         score = self.db.zscore(REDIS_KEY, proxy)
         if score and score > MIN_SCORE:
             print('代理', proxy, '当前分数', score, '减1')
-            return self.db.zincrby(REDIS_KEY, proxy, -1)
+            #return self.db.zincrby(REDIS_KEY, proxy, -1)
+            #更新为redis3.0+版本,解决redis3.0更新后的报错,如用旧版本还原上方代码
+            return self.db.zincrby(REDIS_KEY, -1, proxy)
         else:
             print('代理', proxy, '当前分数', score, '移除')
             return self.db.zrem(REDIS_KEY, proxy)
@@ -73,7 +77,9 @@ class RedisClient(object):
         :return: 设置结果
         """
         print('代理', proxy, '可用,设置为', MAX_SCORE)
-        return self.db.zadd(REDIS_KEY, MAX_SCORE, proxy)
+        #return self.db.zadd(REDIS_KEY, MAX_SCORE, proxy)
+        #更新为redis3.0+版本,解决redis3.0更新后的报错,如用旧版本还原上方代码
+        return self.db.zadd(REDIS_KEY, {proxy:MAX_SCORE})
     
     def count(self):
         """