98 lines
3.9 KiB
Python
98 lines
3.9 KiB
Python
""" PagerMaid event listener. """
|
|
|
|
from os.path import exists
|
|
from telethon import events
|
|
from telethon.errors import MessageTooLongError
|
|
from distutils2.util import strtobool
|
|
from traceback import format_exc
|
|
from time import gmtime, strftime, time
|
|
from sys import exc_info
|
|
from telethon.events import StopPropagation
|
|
from pagermaid import bot, config, help_messages
|
|
from pagermaid.utils import attach_log
|
|
|
|
|
|
def listener(**args):
|
|
""" Register an event listener. """
|
|
command = args.get('command', None)
|
|
description = args.get('description', None)
|
|
parameters = args.get('parameters', None)
|
|
pattern = args.get('pattern', None)
|
|
diagnostics = args.get('diagnostics', True)
|
|
ignore_edited = args.get('ignore_edited', False)
|
|
if command is not None:
|
|
if command in help_messages:
|
|
raise ValueError(f"出错了呜呜呜 ~ 命令 \"{command}\" 已经被注册。")
|
|
pattern = fr"^-{command}(?: |$)([\s\S]*)"
|
|
if pattern is not None and not pattern.startswith('(?i)'):
|
|
args['pattern'] = f"(?i){pattern}"
|
|
else:
|
|
args['pattern'] = pattern
|
|
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']
|
|
|
|
def decorator(function):
|
|
|
|
async def handler(context):
|
|
try:
|
|
try:
|
|
parameter = context.pattern_match.group(1).split(' ')
|
|
if parameter == ['']:
|
|
parameter = []
|
|
context.parameter = parameter
|
|
context.arguments = context.pattern_match.group(1)
|
|
except BaseException:
|
|
context.parameter = None
|
|
context.arguments = None
|
|
await function(context)
|
|
except StopPropagation:
|
|
raise StopPropagation
|
|
except MessageTooLongError:
|
|
await context.edit("出错了呜呜呜 ~ 生成的输出太长，无法显示。")
|
|
except BaseException:
|
|
try:
|
|
await context.edit("出错了呜呜呜 ~ 执行此命令时发生错误。")
|
|
except BaseException:
|
|
pass
|
|
if not diagnostics:
|
|
return
|
|
if strtobool(config['error_report']):
|
|
report = f"# Generated: {strftime('%H:%M %d/%m/%Y', gmtime())}. \n" \
|
|
f"# ChatID: {str(context.chat_id)}. \n" \
|
|
f"# UserID: {str(context.sender_id)}. \n" \
|
|
f"# Message: \n-----BEGIN TARGET MESSAGE-----\n" \
|
|
f"{context.text}\n-----END TARGET MESSAGE-----\n" \
|
|
f"# Traceback: \n-----BEGIN TRACEBACK-----\n" \
|
|
f"{str(format_exc())}\n-----END TRACEBACK-----\n" \
|
|
f"# Error: \"{str(exc_info()[1])}\". \n"
|
|
if exists('/var/lib/pagermaid/public.lock'):
|
|
invite_chatid = -1001213866307
|
|
else:
|
|
invite_chatid = -1001441461877
|
|
await attach_log(report, invite_chatid, f"exception.{time()}.pagermaid", None,
|
|
"Error report generated.")
|
|
|
|
if not ignore_edited:
|
|
bot.add_event_handler(handler, events.MessageEdited(**args))
|
|
bot.add_event_handler(handler, events.NewMessage(**args))
|
|
|
|
return handler
|
|
|
|
if description is not None and command is not None:
|
|
if parameters is None:
|
|
parameters = ""
|
|
help_messages.update({
|
|
f"{command}": f"**使用方法:** `-{command} {parameters}`\
|
|
\n{description}"
|
|
})
|
|
|
|
return decorator
|