From e5f1d5be60ebd6b5f40ad72835146040b034ae35 Mon Sep 17 00:00:00 2001 From: Arthur Khachaturov Date: Tue, 22 Oct 2024 21:22:27 +0300 Subject: [PATCH] Split up return statements in `get_value` functions --- aiogram/fsm/storage/memory.py | 3 ++- aiogram/fsm/storage/mongo.py | 3 ++- aiogram/fsm/storage/redis.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/aiogram/fsm/storage/memory.py b/aiogram/fsm/storage/memory.py index 348f2a3b..f26d15c0 100644 --- a/aiogram/fsm/storage/memory.py +++ b/aiogram/fsm/storage/memory.py @@ -59,7 +59,8 @@ class MemoryStorage(BaseStorage): async def get_value( self, storage_key: StorageKey, dict_key: str, default: Optional[Any] = None ) -> Optional[Any]: - return copy(self.storage[storage_key].data.get(dict_key, default)) + data = self.storage[storage_key].data + return copy(data.get(dict_key, default)) class DisabledEventIsolation(BaseEventIsolation): diff --git a/aiogram/fsm/storage/mongo.py b/aiogram/fsm/storage/mongo.py index 5e291be4..e6fd4761 100644 --- a/aiogram/fsm/storage/mongo.py +++ b/aiogram/fsm/storage/mongo.py @@ -124,7 +124,8 @@ class MongoStorage(BaseStorage): async def get_value( self, storage_key: StorageKey, dict_key: str, default: Optional[Any] = None ) -> Optional[Any]: - return (await self.get_data(storage_key)).get(dict_key, default) + data = await self.get_data(storage_key) + return data.get(dict_key, default) async def update_data(self, key: StorageKey, data: Dict[str, Any]) -> Dict[str, Any]: document_id = self._key_builder.build(key) diff --git a/aiogram/fsm/storage/redis.py b/aiogram/fsm/storage/redis.py index 02eaa421..decd7506 100644 --- a/aiogram/fsm/storage/redis.py +++ b/aiogram/fsm/storage/redis.py @@ -141,7 +141,8 @@ class RedisStorage(BaseStorage): async def get_value( self, storage_key: StorageKey, dict_key: str, default: Optional[Any] = None ) -> Optional[Any]: - return (await self.get_data(storage_key)).get(dict_key, default) + data = await self.get_data(storage_key) + return data.get(dict_key, default) class RedisEventIsolation(BaseEventIsolation):