+14
-4
@@ -1,5 +1,5 @@
|
||||
from contextlib import asynccontextmanager
|
||||
from urllib.parse import urlparse, parse_qs
|
||||
from urllib.parse import parse_qs, urlparse
|
||||
|
||||
from fastapi import FastAPI, HTTPException, Request
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
@@ -31,6 +31,10 @@ from app.modules.zootech.wesp_compat_admin import router as wesp_compat_admin_ro
|
||||
from app.modules.zootech.wesp_compat_misc import router as wesp_compat_misc_router
|
||||
from app.modules.zootech.wesp_compat_reports import router as wesp_compat_reports_router
|
||||
from app.modules.zootech.wesp_compat_analytics import router as wesp_compat_analytics_router
|
||||
from app.modules.zootech.wesp_compat_sklad import router as wesp_compat_sklad_router
|
||||
from app.modules.zootech.wesp_compat_feed_accounting import router as wesp_compat_feed_accounting_router
|
||||
from app.modules.zootech.wesp_compat_lab import router as wesp_compat_lab_router
|
||||
from app.modules.zootech.wesp_compat_daily_plan import router as wesp_compat_daily_plan_router
|
||||
from app.modules.enterprise.router import router as enterprise_router
|
||||
|
||||
|
||||
@@ -83,13 +87,15 @@ def create_app() -> FastAPI:
|
||||
CORSMiddleware,
|
||||
allow_origins=settings.cors_origins,
|
||||
allow_credentials=True,
|
||||
allow_methods=["GET", "POST", "PATCH", "DELETE", "OPTIONS"],
|
||||
allow_headers=["Authorization", "Content-Type", "X-Request-ID"],
|
||||
allow_methods=["GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"],
|
||||
allow_headers=["Authorization", "Content-Type", "X-Request-ID", "X-Enterprise-Id"],
|
||||
)
|
||||
|
||||
@app.exception_handler(HTTPException)
|
||||
async def wesp_compat_http_exception_handler(request: Request, exc: HTTPException):
|
||||
if isinstance(exc.detail, dict) and exc.detail.get("status") == "error":
|
||||
if isinstance(exc.detail, dict) and (
|
||||
exc.detail.get("status") == "error" or exc.detail.get("error") is True
|
||||
):
|
||||
return JSONResponse(status_code=exc.status_code, content=exc.detail)
|
||||
return await http_exception_handler(request, exc)
|
||||
|
||||
@@ -113,6 +119,10 @@ def create_app() -> FastAPI:
|
||||
app.include_router(wesp_compat_misc_router, prefix="/api", tags=["zootech-wesp-compat"])
|
||||
app.include_router(wesp_compat_reports_router, prefix="/api", tags=["zootech-wesp-compat"])
|
||||
app.include_router(wesp_compat_analytics_router, prefix="/api", tags=["zootech-wesp-compat"])
|
||||
app.include_router(wesp_compat_sklad_router, prefix="/api", tags=["zootech-wesp-compat"])
|
||||
app.include_router(wesp_compat_feed_accounting_router, prefix="/api", tags=["zootech-wesp-compat"])
|
||||
app.include_router(wesp_compat_lab_router, prefix="/api", tags=["zootech-wesp-compat"])
|
||||
app.include_router(wesp_compat_daily_plan_router, prefix="/api", tags=["zootech-wesp-compat"])
|
||||
if settings.enable_test_routes:
|
||||
app.include_router(test_router, prefix="/api/v1/test", tags=["test"])
|
||||
return app
|
||||
|
||||
Reference in New Issue
Block a user