From 4118ce6015951db6f0bbc5598cf04f22e3caaa7f Mon Sep 17 00:00:00 2001 From: Rishat Fayzullin Date: Mon, 11 Mar 2024 09:50:49 +0300 Subject: [PATCH] IF/ELSE statement simplified --- aiogram/fsm/storage/mongo.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/aiogram/fsm/storage/mongo.py b/aiogram/fsm/storage/mongo.py index ceb2dfb3..ae10c82a 100644 --- a/aiogram/fsm/storage/mongo.py +++ b/aiogram/fsm/storage/mongo.py @@ -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")