mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-05-10 04:00:24 +00:00
712f20a8fa
Some checks are pending
Pre-commit / pre-commit (push) Waiting to run
CodeQL Advanced / Analyze (actions) (push) Waiting to run
CodeQL Advanced / Analyze (javascript-typescript) (push) Waiting to run
CodeQL Advanced / Analyze (python) (push) Waiting to run
Test / Run Python Tests (push) Waiting to run
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
|
|
"""v1 exception handler registration."""
|
|
|
|
from fastapi import Request, FastAPI
|
|
from fastapi.encoders import jsonable_encoder
|
|
from fastapi.exceptions import RequestValidationError
|
|
from fastapi.responses import JSONResponse
|
|
from sqlalchemy.exc import NoResultFound
|
|
|
|
from app.core import code
|
|
from app.core.pydantic.i18n import trans, get_language
|
|
from app.shared.exception import (
|
|
NoPermissionException,
|
|
TokenException,
|
|
UserException,
|
|
)
|
|
|
|
|
|
def register_exception_handlers(app: FastAPI) -> None:
|
|
"""Register all v1 exception handlers on the given FastAPI app."""
|
|
|
|
@app.exception_handler(RequestValidationError)
|
|
async def request_exception(request: Request, e: RequestValidationError):
|
|
lang = get_language(request.headers.get("Accept-Language")) or "en_US"
|
|
return JSONResponse(
|
|
content={
|
|
"code": code.form_error,
|
|
"error": jsonable_encoder(trans.translate(list(e.errors()), locale=lang)),
|
|
}
|
|
)
|
|
|
|
@app.exception_handler(TokenException)
|
|
async def token_exception(request: Request, e: TokenException):
|
|
return JSONResponse(content={"code": e.code, "text": e.text})
|
|
|
|
@app.exception_handler(UserException)
|
|
async def user_exception(request: Request, e: UserException):
|
|
return JSONResponse(content={"code": e.code, "text": e.description})
|
|
|
|
@app.exception_handler(NoPermissionException)
|
|
async def no_permission(request: Request, exception: NoPermissionException):
|
|
return JSONResponse(
|
|
status_code=200,
|
|
content={"code": code.no_permission_error, "text": exception.text},
|
|
)
|
|
|
|
@app.exception_handler(NoResultFound)
|
|
async def no_results(request: Request, exception: NoResultFound):
|
|
return JSONResponse(
|
|
status_code=200,
|
|
content={"code": code.not_found, "text": exception._message()},
|
|
)
|