TelegramPoster/plugins/callbacks/submission.py

from os import path
from shutil import rmtree
from pyrogram import filters
from pyrogram.types import CallbackQuery, InlineKeyboardMarkup, InlineKeyboardButton
from classes.exceptions import SubmissionDuplicatesError, SubmissionUnavailableError
from classes.poster_client import PosterClient
from classes.user import PosterUser

from modules.app import app
from modules.logger import logWrite
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)
            ),
            quote=True,
        )
        logWrite(
            locale(
                "submission_duplicate",
                "console",
                locale=configGet("locale_log").format(
                    fullclb[2],
                    str(exp.duplicates),
                ),
            ),
            debug=True,
        )
        return

    if submission[0] is not None:
        await submission[0].reply_text(
            locale("sub_yes", "message", locale=submission[0].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", locale=user_locale)),
                    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", locale=user_locale)),
                    callback_data="nothing",
                )
            ]
        ]
    )

    if configGet("send_uploaded_id", "submission"):
        await clb.message.edit_caption(
            clb.message.caption + f"\n\nID: `{submission[1]}`"
        )

    await clb.message.edit_reply_markup(
        reply_markup=InlineKeyboardMarkup(edited_markup)
    )

    logWrite(
        locale(
            "submission_accepted",
            "console",
            locale=configGet("locale_log").format(fullclb[2], submission[1]),
        ),
        debug=True,
    )

    # 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


@app.on_callback_query(filters.regex("sub_no_[\s\S]*"))
async def callback_query_no(app: PosterClient, clb: CallbackQuery):
    fullclb = str(clb.data).split("_")
    user_locale = clb.from_user.language_code

    db_entry = col_submitted.find_one_and_delete({"_id": ObjectId(fullclb[2])})

    if db_entry["temp"]["uuid"] is not None:
        if path.exists(
            path.join(
                configGet("data", "locations"), "submissions", db_entry["temp"]["uuid"]
            )
        ):
            rmtree(
                path.join(
                    configGet("data", "locations"),
                    "submissions",
                    db_entry["temp"]["uuid"],
                ),
                ignore_errors=True,
            )

    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

    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("declined", "button", locale=user_locale)),
                    callback_data="nothing",
                )
            ],
            clb.message.reply_markup.inline_keyboard[1],
        ]
        if len(clb.message.reply_markup.inline_keyboard) > 1
        else [
            [
                InlineKeyboardButton(
                    text=str(locale("declined", "button", locale=user_locale)),
                    callback_data="nothing",
                )
            ]
        ]
    )
    await clb.message.edit_reply_markup(
        reply_markup=InlineKeyboardMarkup(edited_markup)
    )
    logWrite(
        locale(
            "submission_rejected",
            "console",
            locale=configGet("locale_log").format(fullclb[2]),
        ),
        debug=True,
    )


@app.on_callback_query(filters.regex("sub_block_[\s\S]*"))
async def callback_query_block(app: PosterClient, clb: CallbackQuery):
    fullclb = str(clb.data).split("_")
    user_locale = clb.from_user.language_code
    await app.send_message(
        int(fullclb[2]), locale("sub_blocked", "message", locale=configGet("locale"))
    )
    PosterUser(int(fullclb[2])).block()
    await clb.answer(
        text=locale("sub_block", "callback", locale=user_locale).format(fullclb[2]),
        show_alert=True,
    )

    edited_markup = [
        clb.message.reply_markup.inline_keyboard[0],
        [
            InlineKeyboardButton(
                text=str(locale("sub_unblock", "button", locale=user_locale)),
                callback_data=f"sub_unblock_{fullclb[2]}",
            )
        ],
    ]
    await clb.message.edit_reply_markup(
        reply_markup=InlineKeyboardMarkup(edited_markup)
    )
    logWrite(
        locale(
            "user_blocked",
            "console",
            locale=configGet("locale_log").format(fullclb[2]),
        ),
        debug=True,
    )


@app.on_callback_query(filters.regex("sub_unblock_[\s\S]*"))
async def callback_query_unblock(app: PosterClient, clb: CallbackQuery):
    fullclb = str(clb.data).split("_")
    user_locale = clb.from_user.language_code
    await app.send_message(
        int(fullclb[2]), locale("sub_unblocked", "message", locale=configGet("locale"))
    )
    PosterUser(int(fullclb[2])).unblock()
    await clb.answer(
        text=locale("sub_unblock", "callback", locale=user_locale).format(fullclb[2]),
        show_alert=True,
    )

    edited_markup = [
        clb.message.reply_markup.inline_keyboard[0],
        [
            InlineKeyboardButton(
                text=str(locale("sub_block", "button", locale=user_locale)),
                callback_data=f"sub_block_{fullclb[2]}",
            )
        ],
    ]
    await clb.message.edit_reply_markup(
        reply_markup=InlineKeyboardMarkup(edited_markup)
    )
    logWrite(
        locale(
            "user_unblocked",
            "console",
            locale=configGet("locale_log").format(fullclb[2]),
        ),
        debug=True,
    )