78 lines
4.1 KiB
Python
78 lines
4.1 KiB
Python
from pyrogram import filters
|
|
from pyrogram.client import Client
|
|
from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, Message
|
|
from modules.utils import jsonLoad, configGet, locale
|
|
from modules.logging import logWrite
|
|
from modules.app import app
|
|
from modules.submissions import subLimited, subLimit
|
|
|
|
@app.on_message(~ filters.scheduled & filters.photo | filters.video | filters.animation | filters.document)
|
|
def get_submission(_: Client, msg: Message):
|
|
try:
|
|
if msg.from_user.id not in jsonLoad(configGet("blocked", "locations")):
|
|
user_locale = msg.from_user.language_code
|
|
if not subLimited(msg.from_user):
|
|
|
|
if msg.document != None:
|
|
if msg.document.mime_type not in configGet("mime_types", "submission"):
|
|
msg.reply_text(locale("mime_not_allowed", "message", locale=user_locale), quote=True)
|
|
return
|
|
if msg.document.file_size > configGet("file_size", "submission"):
|
|
msg.reply_text(locale("document_too_large", "message", locale=user_locale).format(str(configGet("file_size", "submission")/1024/1024)), quote=True)
|
|
return
|
|
|
|
if msg.video != None:
|
|
if msg.video.file_size > configGet("file_size", "submission"):
|
|
msg.reply_text(locale("document_too_large", "message", locale=user_locale).format(str(configGet("file_size", "submission")/1024/1024)), quote=True)
|
|
return
|
|
|
|
buttons = [
|
|
[
|
|
InlineKeyboardButton(text=locale("sub_yes", "button", locale=configGet("locale")), callback_data=f"sub_yes_{msg.from_user.id}_{msg.id}")
|
|
]
|
|
]
|
|
|
|
if msg.caption != None:
|
|
caption = str(msg.caption)
|
|
buttons[0].append(
|
|
InlineKeyboardButton(text=locale("sub_yes_caption", "button", locale=configGet("locale")), callback_data=f"sub_yes_{msg.from_user.id}_{msg.id}_caption")
|
|
)
|
|
buttons[0].append(
|
|
InlineKeyboardButton(text=locale("sub_no", "button", locale=configGet("locale")), callback_data=f"sub_no_{msg.from_user.id}_{msg.id}")
|
|
)
|
|
else:
|
|
caption = ""
|
|
buttons[0].append(
|
|
InlineKeyboardButton(text=locale("sub_no", "button", locale=configGet("locale")), callback_data=f"sub_no_{msg.from_user.id}_{msg.id}")
|
|
)
|
|
|
|
caption += locale("sub_by", "message", locale=locale(configGet("locale")))
|
|
|
|
if msg.from_user.first_name != None:
|
|
caption += f" {msg.from_user.first_name}"
|
|
if msg.from_user.last_name != None:
|
|
caption += f" {msg.from_user.last_name}"
|
|
if msg.from_user.username != None:
|
|
caption += f" (@{msg.from_user.username})"
|
|
if msg.from_user.phone_number != None:
|
|
caption += f" ({msg.from_user.phone_number})"
|
|
|
|
msg.copy(configGet("admin"), caption=caption, reply_markup=InlineKeyboardMarkup(buttons))
|
|
|
|
if msg.from_user.id != configGet("admin"):
|
|
buttons += [
|
|
[
|
|
InlineKeyboardButton(text=locale("sub_block", "button", locale=configGet("locale")), callback_data=f"sub_block_{msg.from_user.id}")
|
|
],
|
|
[
|
|
InlineKeyboardButton(text=locale("sub_unblock", "button", locale=configGet("locale")), callback_data=f"sub_unblock_{msg.from_user.id}")
|
|
]
|
|
]
|
|
|
|
msg.reply_text(locale("sub_sent", "message", locale=user_locale), quote=True)
|
|
subLimit(msg.from_user)
|
|
|
|
else:
|
|
msg.reply_text(locale("sub_cooldown", "message", locale=user_locale).format(str(configGet("timeout", "submission"))))
|
|
except AttributeError:
|
|
logWrite(f"from_user in function get_submission does not seem to contain id") |