This repository has been archived on 2024-08-21. You can view files and clone it, but cannot push or open issues or pull requests.
TelegramPoster/plugins/callbacks/submission.py

116 lines
5.7 KiB
Python
Raw Normal View History

2023-02-15 15:06:06 +02:00
from os import path, sep
2023-01-10 13:52:44 +02:00
from pathlib import Path
2023-02-14 12:38:54 +02:00
2023-01-10 13:52:44 +02:00
from pyrogram import filters
from pyrogram.client import Client
2023-02-15 15:06:06 +02:00
from pyrogram.types import CallbackQuery
2023-02-14 12:38:54 +02:00
2023-01-17 15:11:23 +02:00
from modules.api_client import upload_pic
2023-01-10 13:52:44 +02:00
from modules.app import app
2023-02-14 12:38:54 +02:00
from modules.submissions import subBlock, subUnblock
from modules.utils import configGet, jsonLoad, jsonSave, locale
2023-02-15 15:06:06 +02:00
from modules.database import col_submitted
from bson import ObjectId
2023-01-10 13:52:44 +02:00
2023-01-10 14:06:24 +02:00
2023-02-15 15:06:06 +02:00
@app.on_callback_query(filters.regex("sub_yes_[\s\S]*"))
2023-01-10 14:06:24 +02:00
async def callback_query_yes(app: Client, clb: CallbackQuery):
2023-02-15 15:06:06 +02:00
2023-01-10 13:52:44 +02:00
fullclb = clb.data.split("_")
user_locale = clb.from_user.language_code
2023-02-15 15:06:06 +02:00
# 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:
2023-01-10 14:06:24 +02:00
await clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True)
2023-01-10 13:52:44 +02:00
return
2023-02-15 15:06:06 +02:00
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
2023-01-10 13:52:44 +02:00
try:
2023-01-17 15:11:23 +02:00
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"))
2023-01-10 13:52:44 +02:00
except:
2023-01-10 14:06:24 +02:00
await clb.answer(text=locale("sub_media_unavail", "message", locale=user_locale), show_alert=True)
2023-01-10 13:52:44 +02:00
return
2023-01-10 14:06:24 +02:00
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)
2023-01-17 15:38:21 +02:00
# edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]]
# await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
2023-01-10 13:52:44 +02:00
2023-02-15 15:06:06 +02:00
@app.on_callback_query(filters.regex("sub_no_[\s\S]*"))
2023-01-10 14:06:24 +02:00
async def callback_query_no(app: Client, clb: CallbackQuery):
2023-01-10 13:52:44 +02:00
fullclb = clb.data.split("_")
user_locale = clb.from_user.language_code
try:
2023-01-10 14:06:24 +02:00
submission = await app.get_messages(int(fullclb[2]), int(fullclb[3]))
2023-01-10 13:52:44 +02:00
except:
2023-01-10 14:06:24 +02:00
await clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True)
2023-01-10 13:52:44 +02:00
return
2023-01-10 14:06:24 +02:00
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)
2023-01-17 15:38:21 +02:00
# edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]]
# await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
2023-01-10 13:52:44 +02:00
@app.on_callback_query(filters.regex("sub_block_[\s\S]*"))
2023-01-10 14:06:24 +02:00
async def callback_query_block(app: Client, clb: CallbackQuery):
2023-01-10 13:52:44 +02:00
fullclb = clb.data.split("_")
user_locale = clb.from_user.language_code
2023-01-10 14:06:24 +02:00
await app.send_message(int(fullclb[2]), locale("sub_msg_unavail", "message", locale=configGet("locale")))
2023-01-10 13:52:44 +02:00
subBlock(int(fullclb[2]))
2023-01-10 14:06:24 +02:00
await clb.answer(text=locale("sub_block", "callback", locale=user_locale).format(fullclb[2]), show_alert=True)
2023-01-17 15:38:21 +02:00
# edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]]
# await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
2023-01-10 13:52:44 +02:00
@app.on_callback_query(filters.regex("sub_unblock_[\s\S]*"))
2023-01-10 14:06:24 +02:00
async def callback_query_unblock(app: Client, clb: CallbackQuery):
2023-01-10 13:52:44 +02:00
fullclb = clb.data.split("_")
user_locale = clb.from_user.language_code
2023-01-10 14:06:24 +02:00
await app.send_message(int(fullclb[2]), locale("sub_msg_unavail", "message", locale=configGet("locale")))
2023-01-10 13:52:44 +02:00
subUnblock(int(fullclb[2]))
2023-01-17 15:38:21 +02:00
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))