From 395472a319bdf51867ccff6a4365035ee12a1591 Mon Sep 17 00:00:00 2001 From: Yasir Aris M Date: Sun, 30 Jun 2024 21:37:21 +0700 Subject: [PATCH] Prepare for feature disable welcome --- database/greetings_db.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 database/greetings_db.py diff --git a/database/greetings_db.py b/database/greetings_db.py new file mode 100644 index 00000000..2bf1f362 --- /dev/null +++ b/database/greetings_db.py @@ -0,0 +1,19 @@ +from database import dbname + +greetingdb = dbname["greetings"] + +async def is_welcome(chat_id: int) -> bool: + greets = await greetingdb.find_one({"chat_id": chat_id}) + return bool(greets) + +async def welcome_off(chat_id: int): + wlc = await greetingdb.find_one({"chat_id": chat_id}) + if wlc: + return await cleandb.delete_one({"chat_id": chat_id}) + +async def welcome_on(chat_id: int): + wlc = await greetingdb.find_one({"chat_id": chat_id}) + if not wlc: + return await greetingdb.insert_one({"chat_id": chat_id}) + +# todo other features for custom welcome here