Files
MoneyPrinterTurbo/app/models/exception.py
2024-03-11 16:37:49 +08:00

27 lines
752 B
Python

import traceback
from typing import Any
from loguru import logger
class HttpException(Exception):
def __init__(self, task_id: str, status_code: int, message: str = '', data: Any = None):
self.message = message
self.status_code = status_code
self.data = data
# 获取异常堆栈信息
tb_str = traceback.format_exc().strip()
if not tb_str or tb_str == "NoneType: None":
msg = f'HttpException: {status_code}, {task_id}, {message}'
else:
msg = f'HttpException: {status_code}, {task_id}, {message}\n{tb_str}'
if status_code == 400:
logger.warning(msg)
else:
logger.error(msg)
class FileNotFoundException(Exception):
pass