from os import path, sep from pathlib import Path from pyrogram import filters from pyrogram.client import Client from pyrogram.types import CallbackQuery from modules.api_client import upload_pic from modules.app import app from modules.submissions import subBlock, subUnblock from modules.utils import configGet, jsonLoad, jsonSave, locale from modules.database import col_submitted from bson import ObjectId @app.on_callback_query(filters.regex("sub_yes_[\s\S]*")) async def callback_query_yes(app: Client, clb: CallbackQuery): fullclb = clb.data.split("_") user_locale = clb.from_user.language_code # Check if submission is in DB and really exists # Upload the file to the API server # Modify submission in DB to state that it's accepted (or so called "done") # Change keyboard to a completed variant # Send replies to both user and admin about accepting the application db_entry = col_submitted.find_one({"_id": ObjectId(fullclb[2])}) if db_entry is None: await clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True) return else: if db_entry["tmp"]["uuid"] is not None: if not path.exists(path.join(configGet("data", "locations"), "submissions", db_entry["tmp"]["uuid"], db_entry["tmp"]["file"])): await clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True) return else: try: submission = await app.get_messages(db_entry["user"], db_entry["telegram"]["msg_id"]) except: await clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True) return try: if configGet("api_based", "mode") is True: media = await app.download_media(submission, file_name=configGet("tmp", "locations")+sep) upload = upload_pic(media) if upload[0] is False: await clb.answer(text=locale("sub_media_failed", "message", locale=user_locale), show_alert=True) elif len(upload[1]) > 0: await clb.answer(text=locale("sub_media_duplicates", "message", locale=user_locale)) await clb.message.reply_text(locale("sub_media_duplicates_list", "message", locale=user_locale).format("\n • ".join(upload[1]))) else: if clb.data.endswith("_caption"): index = jsonLoad(configGet("index", "locations")) index["captions"][Path(media).name] = submission.caption jsonSave(index, configGet("index", "locations")) else: media = await app.download_media(submission, file_name=configGet("queue", "locations")+sep) if clb.data.endswith("_caption"): index = jsonLoad(configGet("index", "locations")) index["captions"][Path(media).name] = submission.caption jsonSave(index, configGet("index", "locations")) except: await clb.answer(text=locale("sub_media_unavail", "message", locale=user_locale), show_alert=True) return await submission.reply_text(locale("sub_yes", "message", locale=submission.from_user.language_code), quote=True) await clb.answer(text=locale("sub_yes", "callback", locale=user_locale).format(fullclb[2]), show_alert=True) # edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]] # await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)) @app.on_callback_query(filters.regex("sub_no_[\s\S]*")) async def callback_query_no(app: Client, clb: CallbackQuery): fullclb = clb.data.split("_") user_locale = clb.from_user.language_code try: submission = await app.get_messages(int(fullclb[2]), int(fullclb[3])) except: await clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True) return await submission.reply_text(locale("sub_no", "message", locale=submission.from_user.language_code), quote=True) await clb.answer(text=locale("sub_no", "callback", locale=user_locale).format(fullclb[2]), show_alert=True) # edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]] # await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)) @app.on_callback_query(filters.regex("sub_block_[\s\S]*")) async def callback_query_block(app: Client, clb: CallbackQuery): fullclb = clb.data.split("_") user_locale = clb.from_user.language_code await app.send_message(int(fullclb[2]), locale("sub_msg_unavail", "message", locale=configGet("locale"))) subBlock(int(fullclb[2])) await clb.answer(text=locale("sub_block", "callback", locale=user_locale).format(fullclb[2]), show_alert=True) # edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]] # await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)) @app.on_callback_query(filters.regex("sub_unblock_[\s\S]*")) async def callback_query_unblock(app: Client, clb: CallbackQuery): fullclb = clb.data.split("_") user_locale = clb.from_user.language_code await app.send_message(int(fullclb[2]), locale("sub_msg_unavail", "message", locale=configGet("locale"))) subUnblock(int(fullclb[2])) await clb.answer(text=locale("sub_unblock", "callback", locale=user_locale).format(fullclb[2]), show_alert=True) # edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]] # await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))