Răsfoiți Sursa

Update redis.py

崔庆才丨静觅 6 ani în urmă
părinte
comite
4bca2d6de0
1 a modificat fișierele cu 3 adăugiri și 3 ștergeri
  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:
         """