Browse Source

style: 使用ruff格式化exceptions模块,优化导入

master
insistence 7 months ago
parent
commit
13cfc60782
  1. 39
      ruoyi-fastapi-backend/exceptions/handle.py

39
ruoyi-fastapi-backend/exceptions/handle.py

@ -1,15 +1,23 @@
from fastapi import FastAPI, Request from fastapi import FastAPI, Request
from fastapi.exceptions import HTTPException from fastapi.exceptions import HTTPException
from pydantic_validation_decorator import FieldValidationError from pydantic_validation_decorator import FieldValidationError
from exceptions.exception import AuthException, LoginException, PermissionException, ServiceException, ServiceWarning, ModelValidatorException from exceptions.exception import (
AuthException,
LoginException,
ModelValidatorException,
PermissionException,
ServiceException,
ServiceWarning,
)
from utils.log_util import logger from utils.log_util import logger
from utils.response_util import ResponseUtil, JSONResponse, jsonable_encoder from utils.response_util import jsonable_encoder, JSONResponse, ResponseUtil
def handle_exception(app: FastAPI): def handle_exception(app: FastAPI):
""" """
全局异常处理 全局异常处理
""" """
# 自定义token检验异常 # 自定义token检验异常
@app.exception_handler(AuthException) @app.exception_handler(AuthException)
async def auth_exception_handler(request: Request, exc: AuthException): async def auth_exception_handler(request: Request, exc: AuthException):
@ -20,6 +28,18 @@ def handle_exception(app: FastAPI):
async def login_exception_handler(request: Request, exc: LoginException): async def login_exception_handler(request: Request, exc: LoginException):
return ResponseUtil.failure(data=exc.data, msg=exc.message) return ResponseUtil.failure(data=exc.data, msg=exc.message)
# 自定义模型检验异常
@app.exception_handler(ModelValidatorException)
async def model_validator_exception_handler(request: Request, exc: ModelValidatorException):
logger.warning(exc.message)
return ResponseUtil.failure(data=exc.data, msg=exc.message)
# 自定义字段检验异常
@app.exception_handler(FieldValidationError)
async def field_validation_error_handler(request: Request, exc: FieldValidationError):
logger.warning(exc.message)
return ResponseUtil.failure(msg=exc.message)
# 自定义权限检验异常 # 自定义权限检验异常
@app.exception_handler(PermissionException) @app.exception_handler(PermissionException)
async def permission_exception_handler(request: Request, exc: PermissionException): async def permission_exception_handler(request: Request, exc: PermissionException):
@ -37,24 +57,11 @@ def handle_exception(app: FastAPI):
logger.warning(exc.message) logger.warning(exc.message)
return ResponseUtil.failure(data=exc.data, msg=exc.message) return ResponseUtil.failure(data=exc.data, msg=exc.message)
# 自定义模型检验异常
@app.exception_handler(ModelValidatorException)
async def model_validator_exception_handler(request: Request, exc: ModelValidatorException):
logger.warning(exc.message)
return ResponseUtil.failure(data=exc.data, msg=exc.message)
# 自定义字段检验异常
@app.exception_handler(FieldValidationError)
async def field_validation_error_handler(request: Request, exc: FieldValidationError):
logger.warning(exc.message)
return ResponseUtil.failure(msg=exc.message)
# 处理其他http请求异常 # 处理其他http请求异常
@app.exception_handler(HTTPException) @app.exception_handler(HTTPException)
async def http_exception_handler(request: Request, exc: HTTPException): async def http_exception_handler(request: Request, exc: HTTPException):
return JSONResponse( return JSONResponse(
content=jsonable_encoder({"code": exc.status_code, "msg": exc.detail}), content=jsonable_encoder({'code': exc.status_code, 'msg': exc.detail}), status_code=exc.status_code
status_code=exc.status_code
) )
# 处理其他异常 # 处理其他异常

Loading…
Cancel
Save