From 36ddaa6d2b76b555587b78ee7f4860394a62387c Mon Sep 17 00:00:00 2001 From: Yasir Aris M Date: Sun, 30 Jun 2024 21:49:46 +0700 Subject: [PATCH] Penyederhanaan by gpt 4 --- database/greetings_db.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/database/greetings_db.py b/database/greetings_db.py index 2bf1f362..38554926 100644 --- a/database/greetings_db.py +++ b/database/greetings_db.py @@ -3,17 +3,12 @@ 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) + return bool(await greetingdb.find_one({"chat_id": chat_id})) -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}) +async def toggle_welcome(chat_id: int): + if await is_welcome(chat_id): + await greetingdb.delete_one({"chat_id": chat_id}) + else: + await greetingdb.insert_one({"chat_id": chat_id}) # todo other features for custom welcome here