mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 17:44:50 +00:00
6e381c7403
* 'Refactored by Sourcery' * reformating: code Co-authored-by: Sourcery AI <> Co-authored-by: yasirarism <yasiramunandar@gmail.com>
18 lines
518 B
Python
18 lines
518 B
Python
from database import dbname
|
|
|
|
imbd_db = dbname.imdb
|
|
|
|
|
|
async def is_imdbset(user_id: int) -> bool:
|
|
user = await imbd_db.find_one({"user_id": user_id})
|
|
return (True, user["lang"]) if user else (False, {})
|
|
|
|
|
|
async def add_imdbset(user_id: int, lang):
|
|
await imbd_db.update_one({"user_id": user_id}, {"$set": {"lang": lang}}, upsert=True)
|
|
|
|
|
|
async def remove_imdbset(user_id: int):
|
|
user = await imbd_db.find_one({"user_id": user_id})
|
|
if user:
|
|
return await imbd_db.delete_one({"user_id": user_id})
|