From fa87b4dfd4a8355d860a6e8e23c37d1cc810b924 Mon Sep 17 00:00:00 2001 From: Oleg A Date: Sun, 1 Aug 2021 22:40:36 +0300 Subject: [PATCH] chore: adapter.get_redis become public --- aiogram/contrib/fsm_storage/redis.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/aiogram/contrib/fsm_storage/redis.py b/aiogram/contrib/fsm_storage/redis.py index ee86b4cc..bd77e8d5 100644 --- a/aiogram/contrib/fsm_storage/redis.py +++ b/aiogram/contrib/fsm_storage/redis.py @@ -240,13 +240,13 @@ class AioRedisAdapterBase(ABC): self._connection_lock = asyncio.Lock(loop=self._loop) @abstractmethod - async def _get_redis(self) -> aioredis.Redis: + async def get_redis(self) -> aioredis.Redis: """Get Redis connection.""" pass async def apply_proxy_methods(self): """Proxy aioredis.Redis methods.""" - redis = await self._get_redis() + redis = await self.get_redis() for name, value in inspect.getmembers(redis, callable): if not hasattr(self, name): setattr(self, name, value) @@ -271,7 +271,7 @@ class AioRedisAdapterBase(ABC): class AioRedisAdapterV1(AioRedisAdapterBase): """Redis adapter for aioredis v1.""" - async def _get_redis(self) -> aioredis.Redis: + async def get_redis(self) -> aioredis.Redis: """Get Redis connection.""" async with self._connection_lock: # to prevent race if self._redis is None or self._redis.closed: @@ -308,7 +308,7 @@ class AioRedisAdapterV1(AioRedisAdapterBase): class AioRedisAdapterV2(AioRedisAdapterBase): """Redis adapter for aioredis v2.""" - async def _get_redis(self) -> aioredis.Redis: + async def get_redis(self) -> aioredis.Redis: """Get Redis connection.""" async with self._connection_lock: # to prevent race if self._redis is None: