904e917f1f
- constants.py: APP_VERSION, LLM_DEFAULT_TEMPERATURE, LLM_RESPONSE_FORMAT - main.py: версия из APP_VERSION (вместо хардкод '0.1.0'×2) - llm.py: temperature и response_format из constants Всего: 85 тестов, ruff clean
95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
"""GuardDog Nexus — FastAPI application entry point."""
|
|
|
|
import os
|
|
from contextlib import asynccontextmanager
|
|
|
|
import uvicorn
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.staticfiles import StaticFiles
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
|
from guarddog_nexus.config import config
|
|
from guarddog_nexus.constants import (
|
|
APP_DESCRIPTION,
|
|
APP_NAME,
|
|
APP_PACKAGE,
|
|
APP_VERSION,
|
|
STATIC_MOUNT_PATH,
|
|
)
|
|
from guarddog_nexus.db.engine import init_db
|
|
from guarddog_nexus.i18n import DEFAULT_LANG, LANGUAGES
|
|
from guarddog_nexus.logging_setup import log
|
|
from guarddog_nexus.routes import api_findings, api_packages, api_scans
|
|
from guarddog_nexus.routes.metrics import router as metrics_router
|
|
from guarddog_nexus.routes.web import router as web_router
|
|
from guarddog_nexus.routes.webhooks import router as webhook_router
|
|
|
|
STATIC_DIR = os.path.join(os.path.dirname(__file__), "web", "static")
|
|
|
|
|
|
class LangMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next):
|
|
# Cookie takes priority, then query param, then default
|
|
cookie_lang = request.cookies.get("lang")
|
|
query_lang = request.query_params.get("lang")
|
|
|
|
if query_lang and query_lang in LANGUAGES and query_lang != cookie_lang:
|
|
lang = query_lang
|
|
elif cookie_lang and cookie_lang in LANGUAGES:
|
|
lang = cookie_lang
|
|
else:
|
|
lang = DEFAULT_LANG
|
|
|
|
request.state.lang = lang
|
|
response = await call_next(request)
|
|
|
|
if query_lang and query_lang in LANGUAGES:
|
|
response.set_cookie("lang", query_lang, max_age=365 * 24 * 3600, httponly=True)
|
|
return response
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
await init_db()
|
|
log.info("%s started on %s:%s", APP_NAME, config.host, config.port)
|
|
yield
|
|
log.info("%s shutting down", APP_NAME)
|
|
|
|
|
|
app = FastAPI(
|
|
title=APP_NAME,
|
|
version=APP_VERSION,
|
|
description=APP_DESCRIPTION,
|
|
lifespan=lifespan,
|
|
)
|
|
app.add_middleware(LangMiddleware)
|
|
|
|
app.include_router(webhook_router)
|
|
app.include_router(metrics_router)
|
|
app.include_router(api_scans.router)
|
|
app.include_router(api_packages.router)
|
|
app.include_router(api_findings.router)
|
|
app.include_router(web_router)
|
|
|
|
if os.path.isdir(STATIC_DIR):
|
|
app.mount(STATIC_MOUNT_PATH, StaticFiles(directory=STATIC_DIR), name="static")
|
|
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok", "version": APP_VERSION}
|
|
|
|
|
|
def main():
|
|
uvicorn.run(
|
|
f"{APP_PACKAGE}.main:app",
|
|
host=config.host,
|
|
port=config.port,
|
|
log_level=config.log_level.lower(),
|
|
reload=False,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|