Files
plain-ub-overfork/app/utils/db_utils.py
2023-10-09 15:35:39 +05:30

17 lines
553 B
Python

from motor.core import AgnosticCollection
async def add_data(collection: AgnosticCollection, id: int | str, data: dict) -> None:
found = await collection.find_one({"_id": id})
if not found:
await collection.insert_one({"_id": id, **data})
else:
await collection.update_one({"_id": id}, {"$set": data})
async def delete_data(collection: AgnosticCollection, id: int | str) -> bool | None:
found = await collection.find_one({"_id": id})
if found:
await collection.delete_one({"_id": id})
return True