Selaa lähdekoodia

Update redis.py

崔庆才丨静觅 6 vuotta sitten
vanhempi
sitoutus
4bca2d6de0
1 muutettua tiedostoa jossa 3 lisäystä ja 3 poistoa
  1. 3 3
      proxypool/storages/redis.py

+ 3 - 3
proxypool/storages/redis.py

@@ -39,8 +39,8 @@ class RedisClient(object):
             return
         if not self.exists(proxy):
             if IS_REDIS_VERSION_2:
-                return self.db.zadd(REDIS_KEY, score, proxy)
-            return self.db.zadd(REDIS_KEY, {proxy: score})
+                return self.db.zadd(REDIS_KEY, score, proxy.string())
+            return self.db.zadd(REDIS_KEY, {proxy.string(): score})
     
     def random(self) -> Proxy:
         """
@@ -77,7 +77,7 @@ class RedisClient(object):
         # otherwise delete proxy
         else:
             logger.info(f'{proxy.string()} current score {score}, remove')
-            return self.db.zrem(REDIS_KEY, proxy)
+            return self.db.zrem(REDIS_KEY, proxy.string())
     
     def exists(self, proxy: Proxy) -> bool:
         """