File size: 1,085 Bytes
80287e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from typing import Dict, Union
from motor.motor_asyncio import AsyncIOMotorClient as MongoCli
from config import MONGO_DB_URI
mongo = MongoCli(MONGO_DB_URI)
db = mongo.Devine
coupledb = db.couple
afkdb = db.afk
nightmodedb = db.nightmode
notesdb = db.notes
filtersdb = db.filters
async def _get_lovers(cid: int):
lovers = await coupledb.find_one({"chat_id": cid})
if lovers:
lovers = lovers["couple"]
else:
lovers = {}
return lovers
async def _get_image(cid: int):
lovers = await coupledb.find_one({"chat_id": cid})
if lovers:
lovers = lovers["img"]
else:
lovers = {}
return lovers
async def get_couple(cid: int, date: str):
lovers = await _get_lovers(cid)
if date in lovers:
return lovers[date]
else:
return False
async def save_couple(cid: int, date: str, couple: dict, img: str):
lovers = await _get_lovers(cid)
lovers[date] = couple
await coupledb.update_one(
{"chat_id": cid},
{"$set": {"couple": lovers, "img": img}},
upsert=True,
)
|