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

IF/ELSE statement simplified

Browse source
This commit is contained in:
Rishat Fayzullin 2024-03-11 09:50:49 +03:00
parent 9cbeb28f06
commit 4118ce6015
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
aiogram/fsm/storage/mongo.py
View file

@ -84,8 +84,7 @@ class MongoStorage(BaseStorage):
document = await self._states_collection.find_one({"_id": document_id})
if document is None or document["state"] is None:
return None
else:
return str(document["state"])
return str(document["state"])
async def set_data(self, key: StorageKey, data: Dict[str, Any]) -> None:
document_id = self._key_builder.build(key, "data")
@ -103,8 +102,7 @@ class MongoStorage(BaseStorage):
document = await self._data_collection.find_one({"_id": document_id}, {"_id": 0})
if not document:
return {}
else:
return cast(Dict[str, Any], document)
return cast(Dict[str, Any], document)
async def update_data(self, key: StorageKey, data: Dict[str, Any]) -> Dict[str, Any]:
document_id = self._key_builder.build(key, "data")