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

Refactoring while review

Browse source
This commit is contained in:
Rishat Fayzullin 2024-03-14 18:49:10 +03:00
parent 13bd7f2b98
commit 90b55e39dd
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -86,9 +86,9 @@ class MongoStorage(BaseStorage):
async def get_state(self, key: StorageKey) -> Optional[str]:
document_id = self._key_builder.build(key)
document = await self._collection.find_one({"_id": document_id})
if document is None or document.get("state") is None:
if document is None:
return None
return str(document["state"])
return document.get("state")
async def set_data(self, key: StorageKey, data: Dict[str, Any]) -> None:
document_id = self._key_builder.build(key)