From bf3edf9e9c69035d2723eac9ccb27ddbb45729f5 Mon Sep 17 00:00:00 2001 From: asimaranov Date: Fri, 17 Dec 2021 21:54:50 +0300 Subject: [PATCH] Fixed set_data --- aiogram/dispatcher/fsm/storage/mongo.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/aiogram/dispatcher/fsm/storage/mongo.py b/aiogram/dispatcher/fsm/storage/mongo.py index d68a3c7c..7e590754 100644 --- a/aiogram/dispatcher/fsm/storage/mongo.py +++ b/aiogram/dispatcher/fsm/storage/mongo.py @@ -125,9 +125,11 @@ class MongoStorage(BaseStorage): key: StorageKey, data: Dict[str, Any], ) -> None: - data_to_insert = self._get_db_filter(key) - data_to_insert['data'] = data - await self._db[DATA].insert_one(data_to_insert) + await self._db[DATA].update_one( + filter=self._get_db_filter(key), + update={'$set': {'data': data}}, + upsert=True, + ) async def get_data( self,