66 lines
1.5 KiB
Python
66 lines
1.5 KiB
Python
from fastapi import FastAPI, Request
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from contextlib import asynccontextmanager
|
|
|
|
from database import create_tables, run_migrations, SessionLocal
|
|
from auth import create_default_admin
|
|
from routes import auth_router, api_router, web_router
|
|
from settings_store import get_setting
|
|
from timezone_utils import set_current_timezone
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
create_tables()
|
|
run_migrations()
|
|
db = SessionLocal()
|
|
try:
|
|
create_default_admin(db)
|
|
finally:
|
|
db.close()
|
|
yield
|
|
|
|
|
|
app = FastAPI(
|
|
title="InfoUser Monitor",
|
|
description="PC monitoring dashboard with FastAPI and agents",
|
|
version="1.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.mount("/static", StaticFiles(directory="static"), name="static")
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
|
|
@app.middleware("http")
|
|
async def timezone_middleware(request: Request, call_next):
|
|
"""Set current request timezone from DB settings for web rendering."""
|
|
db = SessionLocal()
|
|
try:
|
|
tz = get_setting(db, "TIMEZONE")
|
|
if tz:
|
|
set_current_timezone(tz)
|
|
else:
|
|
set_current_timezone("UTC")
|
|
finally:
|
|
db.close()
|
|
return await call_next(request)
|
|
|
|
|
|
app.include_router(auth_router)
|
|
app.include_router(api_router)
|
|
app.include_router(web_router)
|
|
|
|
|
|
@app.get("/health")
|
|
def health():
|
|
return {"status": "ok"}
|