mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-11-25 18:04:00 +00:00
79 lines
3.7 KiB
Python
79 lines
3.7 KiB
Python
import html
|
||
import json
|
||
import traceback
|
||
|
||
from telegram import Update, ReplyKeyboardRemove
|
||
from telegram.constants import ParseMode
|
||
from telegram.error import BadRequest, Forbidden
|
||
from telegram.ext import CallbackContext
|
||
|
||
from core.bot import bot
|
||
from core.plugin import error_handler, Plugin
|
||
from utils.log import logger
|
||
|
||
notice_chat_id = bot.config.error_notification_chat_id
|
||
|
||
|
||
class ErrorHandler(Plugin):
|
||
|
||
@error_handler(block=False) # pylint: disable=E1123, E1120
|
||
async def error_handler(self, update: object, context: CallbackContext) -> None:
|
||
"""记录错误并发送消息通知开发人员。 logger the error and send a telegram message to notify the developer."""
|
||
|
||
logger.error("处理函数时发生异常")
|
||
logger.exception(context.error, exc_info=(type(context.error), context.error, context.error.__traceback__))
|
||
|
||
if notice_chat_id is None:
|
||
return
|
||
|
||
tb_list = traceback.format_exception(None, context.error, context.error.__traceback__)
|
||
tb_string = ''.join(tb_list)
|
||
|
||
update_str = update.to_dict() if isinstance(update, Update) else str(update)
|
||
text_1 = (
|
||
f'<b>处理函数时发生异常</b> \n'
|
||
f'Exception while handling an update \n'
|
||
f'<pre>update = {html.escape(json.dumps(update_str, indent=2, ensure_ascii=False))}'
|
||
'</pre>\n\n'
|
||
f'<pre>context.chat_data = {html.escape(str(context.chat_data))}</pre>\n\n'
|
||
f'<pre>context.user_data = {html.escape(str(context.user_data))}</pre>\n\n'
|
||
)
|
||
text_2 = (
|
||
f'<pre>{html.escape(tb_string)}</pre>'
|
||
)
|
||
try:
|
||
if 'make sure that only one bot instance is running' in tb_string:
|
||
logger.error("其他机器人在运行，请停止！")
|
||
return
|
||
await context.bot.send_message(notice_chat_id, text_1, parse_mode=ParseMode.HTML)
|
||
await context.bot.send_message(notice_chat_id, text_2, parse_mode=ParseMode.HTML)
|
||
except BadRequest as exc:
|
||
if 'too long' in str(exc):
|
||
text = (
|
||
f'<b>处理函数时发生异常，traceback太长导致无法发送，但已写入日志</b> \n'
|
||
f'<code>{html.escape(str(context.error))}</code>'
|
||
)
|
||
try:
|
||
await context.bot.send_message(notice_chat_id, text, parse_mode=ParseMode.HTML)
|
||
except BadRequest:
|
||
text = (
|
||
'<b>处理函数时发生异常，traceback太长导致无法发送，但已写入日志</b> \n')
|
||
try:
|
||
await context.bot.send_message(notice_chat_id, text, parse_mode=ParseMode.HTML)
|
||
except BadRequest as exc_1:
|
||
logger.exception(exc_1)
|
||
effective_user = update.effective_user
|
||
effective_message = update.effective_message
|
||
try:
|
||
if effective_message is not None:
|
||
chat = effective_message.chat
|
||
logger.info(f"尝试通知用户 {effective_user.full_name}[{effective_user.id}] "
|
||
f"在 {chat.full_name}[{chat.id}]"
|
||
f"的 update_id[{update.update_id}] 错误信息")
|
||
text = "出错了呜呜呜 ~ 派蒙这边发生了点问题无法处理！"
|
||
await context.bot.send_message(effective_message.chat_id, text, reply_markup=ReplyKeyboardRemove(),
|
||
parse_mode=ParseMode.HTML)
|
||
except (BadRequest, Forbidden) as exc:
|
||
logger.error(f"发送 update_id[{update.update_id}] 错误信息失败 错误信息为")
|
||
logger.exception(exc)
|