From 90b55e39dd9924042a1e7e51bcba065e306bcf70 Mon Sep 17 00:00:00 2001 From: Rishat Fayzullin Date: Thu, 14 Mar 2024 18:49:10 +0300 Subject: [PATCH] Refactoring while review --- aiogram/fsm/storage/mongo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aiogram/fsm/storage/mongo.py b/aiogram/fsm/storage/mongo.py index 44ea0430..b4b1eeaa 100644 --- a/aiogram/fsm/storage/mongo.py +++ b/aiogram/fsm/storage/mongo.py @@ -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)