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

Revert "fixed PyMongoStorage update_data method implementation"

Browse source 
This reverts commit 86170e1cb9.
This commit is contained in:
kievzenit 2025-08-11 18:34:10 +03:00
parent dbf6b535f8
commit bc7672c1da
1 changed files with 8 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

25
aiogram/fsm/storage/pymongo.py
View file

@ -123,23 +123,14 @@ class PyMongoStorage(BaseStorage):
async def update_data(self, key: StorageKey, data: Mapping[str, Any]) -> Dict[str, Any]:
document_id = self._key_builder.build(key)
update_result = {}
if not data:
update_result = await self._collection.find_one_and_update(
filter={"_id": document_id},
update={"$unset": {"data": 1}},
upsert=True,
return_document=True,
projection={"_id": 0},
)
else:
update_result = await self._collection.find_one_and_update(
filter={"_id": document_id},
update={"$set": {"data": data}},
upsert=True,
return_document=True,
projection={"_id": 0},
)
update_with = {f"data.{key}": value for key, value in data.items()}
update_result = await self._collection.find_one_and_update(
filter={"_id": document_id},
update={"$set": update_with},
upsert=True,
return_document=True,
projection={"_id": 0},
)
if not update_result:
await self._collection.delete_one({"_id": document_id})
return update_result.get("data", {})