mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
110 lines
4.5 KiB
Python
110 lines
4.5 KiB
Python
import json
|
|
import os
|
|
import time
|
|
import traceback
|
|
|
|
import aiofiles
|
|
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 modules.error.pb import PbClient
|
|
from modules.error.sentry import Sentry
|
|
from utils.log import logger
|
|
|
|
notice_chat_id = bot.config.error_notification_chat_id
|
|
current_dir = os.getcwd()
|
|
logs_dir = os.path.join(current_dir, "logs")
|
|
if not os.path.exists(logs_dir):
|
|
os.mkdir(logs_dir)
|
|
report_dir = os.path.join(current_dir, "report")
|
|
if not os.path.exists(report_dir):
|
|
os.mkdir(report_dir)
|
|
pb_client = PbClient()
|
|
sentry = Sentry()
|
|
|
|
|
|
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 not notice_chat_id:
|
|
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)
|
|
|
|
error_text = (
|
|
f"-----Exception while handling an update-----\n"
|
|
f"update = {json.dumps(update_str, indent=2, ensure_ascii=False)}\n"
|
|
f"context.chat_data = {str(context.chat_data)}\n"
|
|
f"context.user_data = {str(context.user_data)}\n"
|
|
"\n"
|
|
"-----Traceback info-----\n"
|
|
f"{tb_string}"
|
|
)
|
|
file_name = f"error_{update.update_id if isinstance(update, Update) else int(time.time())}.txt"
|
|
log_file = os.path.join(report_dir, file_name)
|
|
try:
|
|
async with aiofiles.open(log_file, mode="w+", encoding="utf-8") as f:
|
|
await f.write(error_text)
|
|
except Exception as exc: # pylint: disable=W0703
|
|
logger.error("保存日记失败")
|
|
logger.exception(exc)
|
|
try:
|
|
if "make sure that only one bot instance is running" in tb_string:
|
|
logger.error("其他机器人在运行，请停止！")
|
|
return
|
|
await context.bot.send_document(
|
|
chat_id=notice_chat_id,
|
|
document=open(log_file, "rb"),
|
|
caption=f'Error: "{context.error.__class__.__name__}"',
|
|
)
|
|
except (BadRequest, Forbidden) as exc:
|
|
logger.error("发送日记失败")
|
|
logger.exception(exc)
|
|
except FileNotFoundError:
|
|
logger.error("发送日记失败 文件不存在")
|
|
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)
|
|
try:
|
|
pb_url = await pb_client.create_pb(error_text)
|
|
if pb_url:
|
|
await context.bot.send_message(
|
|
chat_id=notice_chat_id,
|
|
text=f"错误信息已上传至 <a href='{pb_url}'>fars</a> 请查看",
|
|
parse_mode=ParseMode.HTML,
|
|
)
|
|
except Exception as exc: # pylint: disable=W0703
|
|
logger.error("上传错误信息至 fars 失败")
|
|
logger.exception(exc)
|
|
try:
|
|
sentry.report_error(update, (type(context.error), context.error, context.error.__traceback__))
|
|
except Exception as exc: # pylint: disable=W0703
|
|
logger.error("上传错误信息至 sentry 失败")
|
|
logger.exception(exc)
|