109 lines
6.1 KiB
Python
109 lines
6.1 KiB
Python
|
|
from pyrogram import filters
|
|
from pyrogram.types import CallbackQuery, InlineKeyboardMarkup, InlineKeyboardButton
|
|
from classes.exceptions import SubmissionDuplicatesError, SubmissionUnavailableError
|
|
from classes.poster_client import PosterClient
|
|
|
|
from modules.app import app
|
|
from modules.submissions import subBlock, subUnblock
|
|
from modules.utils import configGet, 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: PosterClient, clb: CallbackQuery):
|
|
|
|
fullclb = str(clb.data).split("_")
|
|
user_locale = clb.from_user.language_code
|
|
|
|
db_entry = col_submitted.find_one({"_id": ObjectId(fullclb[2])})
|
|
|
|
try:
|
|
submission = await app.submit_photo(fullclb[2])
|
|
except SubmissionUnavailableError:
|
|
await clb.answer(text=locale("sub_msg_unavail", "callback", locale=user_locale), show_alert=True)
|
|
return
|
|
except SubmissionDuplicatesError as exp:
|
|
await clb.answer(text=locale("sub_duplicates_found", "callback", locale=user_locale), show_alert=True)
|
|
await clb.message.reply_text(locale("sub_media_duplicates_list", "message", locale=user_locale).format("\n • ".join(exp.duplicates)))
|
|
return
|
|
|
|
if submission is not None:
|
|
await submission.reply_text(locale("sub_yes", "message", locale=submission.from_user.language_code), quote=True)
|
|
elif db_entry is not None:
|
|
await app.send_message(db_entry["user"], locale("sub_yes", "message"))
|
|
|
|
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")], clb.message.reply_markup.inline_keyboard[1]] if len(clb.message.reply_markup.inline_keyboard) > 1 else [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]]
|
|
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
|
|
|
# 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: PosterClient, 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: PosterClient, 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: PosterClient, 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)) |