Fixed set_data

This commit is contained in:
asimaranov 2021-12-17 21:54:50 +03:00
parent dbfb87ddb2
commit bf3edf9e9c

View file

@ -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,