Split up return statements in get_value functions

This commit is contained in:
Arthur Khachaturov 2024-10-22 21:22:27 +03:00
parent 8fc682b3f4
commit e5f1d5be60
No known key found for this signature in database
GPG key ID: CAC2B7EB6DF45D55
3 changed files with 6 additions and 3 deletions

View file

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

View file

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

View file

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