From fed154c7f1bba444aaf650328380badce70955fa Mon Sep 17 00:00:00 2001 From: Untone Date: Fri, 13 Oct 2023 13:48:17 +0300 Subject: [PATCH] fix-redis --- services/redis.py | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/services/redis.py b/services/redis.py index 2afb24f0..9f3f56f2 100644 --- a/services/redis.py +++ b/services/redis.py @@ -1,56 +1,56 @@ -import redis.asyncio as redis +import redis.asyncio as aredis from settings import REDIS_URL class RedisCache: def __init__(self, uri=REDIS_URL): self._uri: str = uri self.pubsub_channels = [] - self._redis = None + self._client = None async def connect(self): - self._redis = redis.Redis.from_url(self._uri, decode_responses=True) + self._client = aredis.Redis.from_url(self._uri, decode_responses=True) async def disconnect(self): - await self._redis.aclose() + await self._client.aclose() async def execute(self, command, *args, **kwargs): - if not self._redis: + if not self._client: await self.connect() try: print("[redis] " + command + " " + " ".join(args)) - return await self._redis.execute_command(command, *args, **kwargs) + return await self._client.execute_command(command, *args, **kwargs) except Exception as e: print(f"[redis] error: {e}") return None async def subscribe(self, *channels): - if not self._redis: + if not self._client: await self.connect() - async with self._redis.pubsub() as pubsub: + async with self._client.pubsub() as pubsub: for channel in channels: await pubsub.subscribe(channel) self.pubsub_channels.append(channel) async def unsubscribe(self, *channels): - if not self._redis: + if not self._client: return - async with self._redis.pubsub() as pubsub: + async with self._client.pubsub() as pubsub: for channel in channels: await pubsub.unsubscribe(channel) self.pubsub_channels.remove(channel) async def publish(self, channel, data): - if not self._redis: + if not self._client: return - await self._redis.publish(channel, data) + await self._client.publish(channel, data) async def lrange(self, key, start, stop): print(f"[redis] LRANGE {key} {start} {stop}") - return await self._redis.lrange(key, start, stop) + return await self._client.lrange(key, start, stop) async def mget(self, key, *keys): print(f"[redis] MGET {key} {keys}") - return await self._redis.mget(key, *keys) + return await self._client.mget(key, *keys) redis = RedisCache()