from datetime import date, datetime from traceback import format_exc from pyrogram import filters from pyrogram.enums import ChatMemberStatus, ChatType from pyrogram.types import CallbackQuery from pyrogram.types import InlineKeyboardButton as IKB from pyrogram.types import InlineKeyboardMarkup as IKM from pyrogram.types import Message from Powers import BDB_URI, LOGGER from Powers.bot_class import Gojo from Powers.database.chats_db import Chats if BDB_URI: from Powers.plugins import bday_cinfo, bday_info from Powers.utils.custom_filters import command def give_date(date, form="%d/%m/%Y"): return datetime.strptime(date, form).date() @Gojo.on_message(command("remember")) async def remember_me(c: Gojo, m: Message): if not BDB_URI: await m.reply_text("BDB_URI is not configured") return splited = m.text.split() if len(splited) == 1: await m.reply_text( "**USAGE**:\n/remember [username or user id or reply to user] [DOB]\nDOB should be in format of dd/mm/yyyy\nYear is optional it is not necessary to pass it\nIf not replied to user then register the birthday of the one who have given the command") return if len(splited) != 2 and m.reply_to_message: await m.reply_text( "**USAGE**:\n/remember [username or user id or reply to user] [DOB]\nDOB should be in format of dd/mm/yyyy\nYear is optional it is not necessary to pass it\nIf not replied to user then register the birthday of the one who have given the command") return DOB = splited[1] if len(splited) == 2 else splited[2] if len(splited) == 2 and m.reply_to_message: user = m.reply_to_message.from_user.id else: user = m.from_user.id DOB = DOB.split("/") if len(DOB) not in [3, 2]: await m.reply_text("DOB should be in format of dd/mm/yyyy\nYear is optional it is not necessary to pass it") return is_correct = (len(DOB[2]) == 4) if len(DOB) == 3 else False if len(DOB[0]) != 2 and len(DOB[1]) != 2 and not is_correct: await m.reply_text("DOB should be in format of dd/mm/yyyy\nYear is optional it is not necessary to pass it") return try: date = int(DOB[0]) month = int(DOB[1]) if is_correct: year = int(DOB[2]) is_year = 1 else: year = "1900" is_year = 0 DOB = f"{date}/{month}/{str(year)}" except ValueError: await m.reply_text("DOB should be numbers only") return data = {"user_id": user, "dob": DOB, "is_year": is_year} try: if bday_info.find_one({"user_id": user}): await m.reply_text("User is already in my database") return except Exception as e: await m.reply_text(f"Got an error\n{e}") LOGGER.error(e) LOGGER.error(format_exc()) return try: bday_info.insert_one(data) await m.reply_text("Your birthday is now registered in my database") except Exception as e: await m.reply_text(f"Got an error\n{e}") LOGGER.error(e) LOGGER.error(format_exc()) return @Gojo.on_message(command(["removebday", "rmbday"])) async def who_are_you_again(c: Gojo, m: Message): if not BDB_URI: await m.reply_text("BDB_URI is not configured") return user = m.from_user.id try: if bday_info.find_one({"user_id": user}): bday_info.delete_one({"user_id": user}) await m.reply_text("Removed your birthday") else: await m.reply_text("User is not in my database") return except Exception as e: await m.reply_text(f"Got an error\n{e}") return @Gojo.on_message(command(["nextbdays", "nbdays", "birthdays", "bdays"])) async def who_is_next(c: Gojo, m: Message): if not BDB_URI: await m.reply_text("BDB_URI is not configured") return blist = list(bday_info.find()) if m.chat.type == ChatType.PRIVATE: await m.reply_text("Use it in group") return curr = datetime.now().date() xx = await m.reply_text("📆") users = [] if blist: for i in blist: if Chats(m.chat.id).user_is_in_chat(i["user_id"]): dob = give_date(i["dob"]) if dob.month >= curr.month: users.append(i) if len(users) == 10: break if not users: await xx.delete() await m.reply_text( "There are no upcoming birthdays of any user in this chat:/\nEither all the birthdays are passed or no user from this chat have registered their birthday") return txt = "🎊 Upcomming Birthdays Are 🎊\n" for i in users: try: user = await c.get_users(i["user_id"]) if user.is_deleted: bday_info.delete_one({"user_id": i["user_id"]}) continue name = user.full_name except: name = i["user_id"] DOB = give_date(i["dob"]) dete = date(curr.year, DOB.month, DOB.day) leff = (dete - curr).days txt += f"{name} : {leff} days left\n" txt += "\n\nYou can use /info [user id] to get info about the user" await xx.delete() await m.reply_text(txt) return @Gojo.on_message(command(["getbday", "gbday", "mybirthday", "mybday"])) async def cant_recall_it(c: Gojo, m: Message): if not BDB_URI: await m.reply_text("BDB_URI is not configured") return user = m.from_user.id men = m.from_user.mention if m.reply_to_message: user = m.reply_to_message.from_user.id men = m.reply_to_message.from_user.mention try: result = bday_info.find_one({"user_id": user}) if not result: if not m.reply_to_message: await m.reply_text("You are not registered in my database\nUse `/remember` to register your birth day so I can wish you") await m.reply_text("User is not in my database") return except Exception as e: await m.reply_text(f"Got an error\n{e}") return curr = datetime.now().date() u_dob = give_date(result["dob"]) formatted = str(u_dob.strftime('%d' + '%B %Y'))[2:-5] day = int(result["dob"].split('/')[0]) suffix = {1: 'st', 2: 'nd', 3: 'rd'}.get(day % 10, 'th') bday_on = f"{day}{suffix} {formatted}" if (u_dob.day, u_dob.month) < (curr.day, curr.month): next_b = date(curr.year + 1, u_dob.month, u_dob.day) days_left = (next_b - curr).days txt = f"{men} 's birthday is passed 🫤\nDays left until next one {abs(days_left)}" txt += f"\nBirthday on: {bday_on}" txt += f"\n\nDate of birth: {result['dob']}" elif (u_dob.day, u_dob.month) == (curr.day, curr.month): txt = f"Today is {men}'s birthday." else: u_dobm = date(curr.year, u_dob.month, u_dob.day) days_left = (u_dobm - curr).days txt = f"User's birthday is coming🥳\nDays left: {abs(days_left)}" txt += f"\nBirthday on: {bday_on}" txt += f"\n\nDate of birth: {result['dob']}" txt += "\n\n**NOTE**:\nDOB may be wrong if user haven't entered his/her birth year" await m.reply_text(txt) return @Gojo.on_message(command(["settingbday", "sbday"])) async def chat_birthday_settings(c: Gojo, m: Message): if not BDB_URI: await m.reply_text("BDB_URI is not configured") return if m.chat.type == ChatType.PRIVATE: await m.reply_text("Use in groups") return chats = m.chat.id c_in = bday_cinfo.find_one({"chat_id": chats}) kb = IKM( [ [ IKB(f"{'No' if c_in else 'Yes'}", f"switchh_{'no' if c_in else 'yes'}"), IKB("Close", "f_close") ] ] ) await m.reply_text("Do you want to wish members for their birthday in the group?", reply_markup=kb) return @Gojo.on_callback_query(filters.regex(r"^switchh_(yes|no)$")) async def switch_on_off(c: Gojo, q: CallbackQuery): user = (await q.message.chat.get_member(q.from_user.id)).status await q.message.chat.get_member(q.from_user.id) if user not in [ChatMemberStatus.ADMINISTRATOR, ChatMemberStatus.OWNER]: await q.answer("...") return data = q.data.split("_")[1] chats = q.message.chat.id query = {"chat_id": chats} if data == "yes": bday_cinfo.delete_one(query) elif data == "no": bday_cinfo.insert_one(query) await q.edit_message_text(f"Done! I will {'wish' if data == 'yes' else 'not wish'}", reply_markup=IKM([[IKB("Close", "f_close")]])) return __PLUGIN__ = "birthday" __HELP__ = """ • /remember [reply to user] [DOB] : To registers user date of birth in my database. If not replied to user then the DOB givien will be treated as yours • /nextbdays (/nbdays,/brithdays,/bdays) : Return upcoming birthdays of 10 users • /removebday (/rmbday) : To remove birthday from database (One can only remove their data from database not of others) • /settingbday (/sbday) : To configure the settings for wishing and all for the chat • /getbday (/gbday,/mybirthday,/mybday) [reply to user] : If replied to user get the replied user's birthday else returns your birthday DOB should be in format of dd/mm/yyyy Year is optional it is not necessary to pass it """