from datetime import datetime from app import app from pyrogram.types import ( InlineKeyboardMarkup, InlineKeyboardButton, ForceReply, CallbackQuery, ) from pyrogram.client import Client from pyrogram import filters from classes.errors.holo_user import LabelSettingError from classes.holo_user import HoloUser from modules.utils import configGet, locale, logWrite, should_quote from modules.database import col_tmp, col_sponsorships @app.on_callback_query(filters.regex("sponsor_apply_[\s\S]*")) async def callback_query_sponsor_apply(app: Client, clb: CallbackQuery): fullclb = clb.data.split("_") holo_user = HoloUser(int(fullclb[2])) if holo_user.application_state()[0] == "fill": await clb.message.reply_text( locale("finish_application", "message"), quote=should_quote(clb.message) ) return logWrite(f"User {holo_user.id} applied for sponsorship") holo_user.sponsorship_restart() edited_markup = [ [ InlineKeyboardButton( text=str(locale("sponsor_started", "button")), callback_data="nothing" ) ] ] await clb.message.edit( text=locale("sponsorship_applying", "message", locale=holo_user), reply_markup=InlineKeyboardMarkup(edited_markup), ) await app.send_message( holo_user.id, locale(f"sponsor1", "message", locale=holo_user), reply_markup=ForceReply( placeholder=str(locale(f"sponsor1", "force_reply", locale=holo_user.locale)) ), ) await clb.answer( text=locale("sponsor_started", "callback", locale=holo_user).format( holo_user.id ), show_alert=False, ) @app.on_callback_query(filters.regex("sponsor_yes_[\s\S]*")) async def callback_query_sponsor_yes(app: Client, clb: CallbackQuery): fullclb = clb.data.split("_") holo_user = HoloUser(int(fullclb[2])) await app.send_message( configGet("admin", "groups"), locale("sponsor_approved_by", "message").format( clb.from_user.first_name, holo_user.id ), disable_notification=True, ) await app.send_message( holo_user.id, locale("sponsor_approved", "message", locale=holo_user) ) logWrite(f"User {holo_user.id} got sponsorship approved by {clb.from_user.id}") if col_sponsorships.find_one({"user": holo_user.id}) is not None: col_sponsorships.update_one( {"user": holo_user.id}, { "$set": { "date": datetime.now(), "admin": clb.from_user.id, "sponsorship": col_tmp.find_one( {"user": holo_user.id, "type": "sponsorship"} )["sponsorship"], } }, ) else: col_sponsorships.insert_one( { "user": holo_user.id, "date": datetime.now(), "admin": clb.from_user.id, "sponsorship": col_tmp.find_one( {"user": holo_user.id, "type": "sponsorship"} )["sponsorship"], } ) col_tmp.update_one( {"user": holo_user.id, "type": "sponsorship"}, {"$set": {"state": "approved", "sent": False}}, ) try: await holo_user.label_set( configGet("users", "groups"), col_tmp.find_one( {"user": {"$eq": holo_user.id}, "type": {"$eq": "sponsorship"}} )["sponsorship"]["label"], ) except LabelSettingError as exp: await app.send_message( configGet("owner"), exp.__str__(), disable_notification=True ) await clb.message.reply_text( locale("label_set_exception", "message", locale=clb.from_user), disable_notification=True, ) edited_markup = [ [ InlineKeyboardButton( text=str(locale("accepted", "button")), callback_data="nothing" ) ] ] await app.edit_message_caption( clb.message.chat.id, clb.message.id, caption=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup), ) await clb.answer( text=locale("sponsor_accepted", "callback").format(fullclb[2]), show_alert=False ) @app.on_callback_query(filters.regex("sponsor_no_[\s\S]*")) async def callback_query_sponsor_no(app: Client, clb: CallbackQuery): fullclb = clb.data.split("_") holo_user = HoloUser(int(fullclb[2])) await app.send_message( configGet("admin", "groups"), locale("sponsor_rejected_by", "message").format( clb.from_user.first_name, holo_user.id ), disable_notification=True, ) await app.send_message( holo_user.id, locale("sponsor_rejected", "message", locale=holo_user) ) logWrite(f"User {holo_user.id} got sponsorship rejected by {clb.from_user.id}") col_tmp.update_one( {"user": holo_user.id, "type": "sponsorship"}, {"$set": {"state": "rejected", "sent": False}}, ) edited_markup = [ [ InlineKeyboardButton( text=str(locale("declined", "button")), callback_data="nothing" ) ] ] await app.edit_message_caption( clb.message.chat.id, clb.message.id, caption=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup), ) await clb.answer( text=locale("sponsor_rejected", "callback").format(fullclb[2]), show_alert=False )