arno/gns3-server
1
0
Fork 0
mirror of https://github.com/GNS3/gns3-server synced 2025-01-14 10:00:58 +00:00
Code Issues Releases Wiki Activity

Catch SQLAlchemyError exception.

Browse Source
This commit is contained in:
grossmj 2021-05-16 18:37:17 +09:30
parent d759564e20
commit fffbb08a8e
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
gns3server/api/server.py
View File

@ -25,6 +25,7 @@ from fastapi import FastAPI, Request
from starlette.exceptions import HTTPException as StarletteHTTPException from starlette.exceptions import HTTPException as StarletteHTTPException
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from sqlalchemy.exc import SQLAlchemyError
from uvicorn.main import Server as UvicornServer from uvicorn.main import Server as UvicornServer
from gns3server.controller.controller_error import ( from gns3server.controller.controller_error import (
@ -160,6 +161,15 @@ async def http_exception_handler(request: Request, exc: StarletteHTTPException):
) )
@app.exception_handler(SQLAlchemyError)
async def sqlalchemry_error_handler(request: Request, exc: SQLAlchemyError):
log.error(f"Controller database error: {exc}")
return JSONResponse(
status_code=500,
content={"message": "Database error detected, please check logs to find details"},
)
@app.middleware("http") @app.middleware("http")
async def add_extra_headers(request: Request, call_next): async def add_extra_headers(request: Request, call_next):
start_time = time.time() start_time = time.time()