diff --git a/program/function/user_data.py b/program/function/user_data.py new file mode 100644 index 0000000..fee3c37 --- /dev/null +++ b/program/function/user_data.py @@ -0,0 +1,33 @@ +import logging +import datetime +from driver.database import Database +from config import DB_URL, DB_NAME, LOG_CH + + +db = Database(DB_URL, DB_NAME) + + +async def user_status(bot, cmd): + chat_id = cmd.from_user.id + if not await db.user_exist(chat_id): + data = await bot.get_me() + BOT_USERNAME = data.username + await db.add_user(chat_id) + if LOG_CH: + await bot.send_message( + LOG_CH, + f"#NEWUSER: \n\nNew User [{cmd.from_user.first_name}](tg://user?id={cmd.from_user.id}) started @{BOT_USERNAME} !!", + ) + else: + logging.info(f"#NewUser :- Name : {cmd.from_user.first_name} ID : {cmd.from_user.id}") + + ban_status = await db.get_ban_status(chat_id) + if ban_status["is_banned"]: + if ( + datetime.date.today() - datetime.date.fromisoformat(ban_status["banned_on"]) + ).days > ban_status["ban_duration"]: + await db.remove_ban(chat_id) + else: + await cmd.reply_text("You are Banned to Use This Bot ", quote=True) + return + await cmd.continue_propagation()