python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: adapter.get_redis become public

Browse source
This commit is contained in:
Oleg A 2021-08-01 22:40:36 +03:00
parent e0ada08708
commit fa87b4dfd4
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aiogram/contrib/fsm_storage/redis.py
View file

@ -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: