mirror of
https://github.com/TeamPGM/PagerMaid-Pyro.git
synced 2024-11-16 19:06:21 +00:00
228 lines
9.3 KiB
Python
228 lines
9.3 KiB
Python
import contextlib
|
|
import sys
|
|
import secrets
|
|
from asyncio import sleep
|
|
from time import strftime, gmtime, time
|
|
from traceback import format_exc
|
|
|
|
from pyrogram import ContinuePropagation, StopPropagation, filters, Client
|
|
from pyrogram.errors.exceptions.bad_request_400 import (
|
|
MessageIdInvalid,
|
|
MessageNotModified,
|
|
MessageEmpty,
|
|
UserNotParticipant
|
|
)
|
|
from pyrogram.handlers import MessageHandler, EditedMessageHandler
|
|
|
|
from pagermaid import help_messages, logs, Config, bot, read_context, all_permissions
|
|
from pagermaid.group_manager import Permission
|
|
from pagermaid.single_utils import Message
|
|
from pagermaid.utils import lang, attach_report, sudo_filter, alias_command, get_permission_name, process_exit
|
|
from pagermaid.utils import client as httpx_client
|
|
|
|
secret_generator = secrets.SystemRandom()
|
|
|
|
|
|
def noop(*args, **kw):
|
|
pass
|
|
|
|
|
|
def listener(**args):
|
|
""" Register an event listener. """
|
|
command = args.get("command")
|
|
disallow_alias = args.get("disallow_alias", False)
|
|
need_admin = args.get("need_admin", False)
|
|
description = args.get("description")
|
|
parameters = args.get("parameters")
|
|
pattern = sudo_pattern = args.get("pattern")
|
|
diagnostics = args.get("diagnostics", True)
|
|
ignore_edited = args.get("ignore_edited", False)
|
|
is_plugin = args.get("is_plugin", True)
|
|
incoming = args.get("incoming", False)
|
|
outgoing = args.get("outgoing", True)
|
|
groups_only = args.get("groups_only", False)
|
|
privates_only = args.get("privates_only", False)
|
|
|
|
if command is not None:
|
|
if command in help_messages:
|
|
raise ValueError(f"{lang('error_prefix')} {lang('command')} \"{command}\" {lang('has_reg')}")
|
|
pattern = fr"^,{alias_command(command, disallow_alias)}(?: |$)([\s\S]*)"
|
|
sudo_pattern = fr"^/{alias_command(command, disallow_alias)}(?: |$)([\s\S]*)"
|
|
if pattern is not None and not pattern.startswith("(?i)"):
|
|
args["pattern"] = f"(?i){pattern}"
|
|
else:
|
|
args["pattern"] = pattern
|
|
if sudo_pattern is not None and not sudo_pattern.startswith("(?i)"):
|
|
sudo_pattern = f"(?i){sudo_pattern}"
|
|
if outgoing and not incoming:
|
|
base_filters = filters.me & ~filters.via_bot & ~filters.forwarded
|
|
elif incoming and not outgoing:
|
|
base_filters = filters.incoming & ~filters.me
|
|
else:
|
|
base_filters = filters.all
|
|
permission_name = get_permission_name(is_plugin, need_admin, command)
|
|
sudo_filters = (
|
|
sudo_filter(permission_name)
|
|
& ~filters.via_bot
|
|
& ~filters.forwarded
|
|
)
|
|
if args["pattern"]:
|
|
base_filters &= filters.regex(args["pattern"])
|
|
sudo_filters &= filters.regex(sudo_pattern)
|
|
if groups_only:
|
|
base_filters &= filters.group
|
|
sudo_filters &= filters.group
|
|
if privates_only:
|
|
base_filters &= filters.private
|
|
sudo_filters &= filters.private
|
|
if "ignore_edited" in args:
|
|
del args["ignore_edited"]
|
|
if "command" in args:
|
|
del args["command"]
|
|
if "diagnostics" in args:
|
|
del args["diagnostics"]
|
|
if "description" in args:
|
|
del args["description"]
|
|
if "parameters" in args:
|
|
del args["parameters"]
|
|
if "is_plugin" in args:
|
|
del args["is_plugin"]
|
|
if "owners_only" in args:
|
|
del args["owners_only"]
|
|
if "admins_only" in args:
|
|
del args["admins_only"]
|
|
if "groups_only" in args:
|
|
del args["groups_only"]
|
|
if "need_admin" in args:
|
|
del args["need_admin"]
|
|
|
|
def decorator(function):
|
|
|
|
async def handler(client: Client, message: Message):
|
|
message.bot = client
|
|
message.request = httpx_client
|
|
|
|
try:
|
|
try:
|
|
parameter = message.matches[0].group(1).split(" ")
|
|
if parameter == [""]:
|
|
parameter = []
|
|
message.parameter = parameter
|
|
message.arguments = message.matches[0].group(1)
|
|
except BaseException:
|
|
message.parameter = None
|
|
message.arguments = None
|
|
# solve same process
|
|
if not message.outgoing:
|
|
await sleep(secret_generator.randint(1, 100) / 1000)
|
|
if (message.chat.id, message.id) in read_context:
|
|
raise ContinuePropagation
|
|
read_context[(message.chat.id, message.id)] = True
|
|
|
|
if function.__code__.co_argcount == 1:
|
|
await function(message)
|
|
elif function.__code__.co_argcount == 2:
|
|
await function(client, message)
|
|
except StopPropagation as e:
|
|
raise StopPropagation from e
|
|
except KeyboardInterrupt as e:
|
|
raise KeyboardInterrupt from e
|
|
except MessageNotModified:
|
|
pass
|
|
except MessageIdInvalid:
|
|
logs.warning(
|
|
"Please Don't Delete Commands While it's Processing.."
|
|
)
|
|
except UserNotParticipant:
|
|
pass
|
|
except ContinuePropagation as e:
|
|
raise ContinuePropagation from e
|
|
except SystemExit:
|
|
await process_exit(start=False, _client=client, message=message)
|
|
sys.exit(0)
|
|
except BaseException:
|
|
exc_info = sys.exc_info()[1]
|
|
exc_format = format_exc()
|
|
with contextlib.suppress(BaseException):
|
|
await message.edit(lang("run_error"), no_reply=True) # noqa
|
|
if not diagnostics:
|
|
return
|
|
if Config.ERROR_REPORT:
|
|
report = f"""# Generated: {strftime('%H:%M %d/%m/%Y', gmtime())}. \n# ChatID: {message.chat.id}. \n# UserID: {message.from_user.id if message.from_user else message.sender_chat.id}. \n# Message: \n-----BEGIN TARGET MESSAGE-----\n{message.text or message.caption}\n-----END TARGET MESSAGE-----\n# Traceback: \n-----BEGIN TRACEBACK-----\n{str(exc_format)}\n-----END TRACEBACK-----\n# Error: "{str(exc_info)}". \n"""
|
|
|
|
await attach_report(report, f"exception.{time()}.pagermaid", None,
|
|
"Error report generated.")
|
|
if (message.chat.id, message.id) in read_context:
|
|
del read_context[(message.chat.id, message.id)]
|
|
message.continue_propagation()
|
|
|
|
bot.add_handler(MessageHandler(handler, filters=base_filters), group=0)
|
|
bot.add_handler(MessageHandler(handler, filters=sudo_filters), group=1)
|
|
if not ignore_edited:
|
|
bot.add_handler(EditedMessageHandler(handler, filters=base_filters), group=2)
|
|
bot.add_handler(EditedMessageHandler(handler, filters=sudo_filters), group=3)
|
|
|
|
return handler
|
|
|
|
if description is not None and command is not None:
|
|
if parameters is None:
|
|
parameters = ""
|
|
help_messages.update({
|
|
f"{command}": {"permission": permission_name,
|
|
"use": f"**{lang('use_method')}:** `,{command} {parameters}`\n"
|
|
f"**{lang('need_permission')}:** `{permission_name}`\n"
|
|
f"{description}"}
|
|
})
|
|
all_permissions.append(Permission(permission_name))
|
|
|
|
return decorator
|
|
|
|
|
|
def raw_listener(filter_s):
|
|
"""Simple Decorator To Handel Custom Filters"""
|
|
|
|
def decorator(function):
|
|
async def handler(client, message):
|
|
message.bot = client
|
|
message.request = httpx_client
|
|
|
|
try:
|
|
if function.__code__.co_argcount == 1:
|
|
await function(message)
|
|
elif function.__code__.co_argcount == 2:
|
|
await function(client, message)
|
|
except StopPropagation as e:
|
|
raise StopPropagation from e
|
|
except ContinuePropagation as e:
|
|
raise ContinuePropagation from e
|
|
except SystemExit:
|
|
await process_exit(start=False, _client=client, message=message)
|
|
sys.exit(0)
|
|
except UserNotParticipant:
|
|
pass
|
|
except MessageEmpty:
|
|
pass
|
|
except BaseException:
|
|
exc_info = sys.exc_info()[1]
|
|
exc_format = format_exc()
|
|
with contextlib.suppress(BaseException):
|
|
await message.edit(lang('run_error'), no_reply=True)
|
|
if Config.ERROR_REPORT:
|
|
report = f"# Generated: {strftime('%H:%M %d/%m/%Y', gmtime())}. \n" \
|
|
f"# ChatID: {message.chat.id}. \n" \
|
|
f"# UserID: {message.from_user.id if message.from_user else message.sender_chat.id}. \n" \
|
|
f"# Message: \n-----BEGIN TARGET MESSAGE-----\n" \
|
|
f"{message.text}\n-----END TARGET MESSAGE-----\n" \
|
|
f"# Traceback: \n-----BEGIN TRACEBACK-----\n" \
|
|
f"{str(exc_format)}\n-----END TRACEBACK-----\n" \
|
|
f"# Error: \"{str(exc_info)}\". \n"
|
|
await attach_report(report, f"exception.{time()}.pagermaid", None,
|
|
"Error report generated.")
|
|
message.continue_propagation()
|
|
|
|
bot.add_handler(MessageHandler(handler, filters=filter_s), group=2)
|
|
|
|
return handler
|
|
|
|
return decorator
|