MissKatyPyro/database/locale_db.py
2023-08-03 11:01:15 +07:00

22 lines
587 B
Python

from typing import Iterable
from pyrogram.enums import ChatType
from database import dbname
localesdb = dbname["locale"] # DB for localization
group_types: Iterable[ChatType] = (ChatType.GROUP, ChatType.SUPERGROUP)
async def set_db_lang(chat_id: int, chat_type: str, lang_code: str):
await localesdb.update_one(
{"chat_id": chat_id},
{"$set": {"lang": lang_code, "chat_type": chat_type.value}},
upsert=True,
)
async def get_db_lang(chat_id: int) -> str:
ul = await localesdb.find_one({"chat_id": chat_id})
return ul["lang"] if ul else {}