nessi
f12dd46c21
This commit introduces a detailed database overview endpoint and service, providing key metrics such as replication status, database sizes, and performance indicators. On the frontend, a new UI section displays these metrics along with improved forms and troubleshooting tips. Enhancements improve user experience by adding informative tooltips and formatting for byte and time values.
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
import asyncio
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from sqlalchemy import select
|
|
from app.api.router import api_router
|
|
from app.core.config import get_settings
|
|
from app.core.db import SessionLocal
|
|
from app.core.logging import configure_logging
|
|
from app.core.security import hash_password
|
|
from app.models.models import User
|
|
from app.services.collector import collector_loop
|
|
|
|
settings = get_settings()
|
|
configure_logging(settings.log_level)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
collector_task: asyncio.Task | None = None
|
|
collector_stop_event = asyncio.Event()
|
|
|
|
|
|
async def ensure_admin_user() -> None:
|
|
async with SessionLocal() as db:
|
|
admin = await db.scalar(select(User).where(User.email == settings.init_admin_email))
|
|
if admin:
|
|
return
|
|
user = User(
|
|
email=settings.init_admin_email,
|
|
password_hash=hash_password(settings.init_admin_password),
|
|
role="admin",
|
|
)
|
|
db.add(user)
|
|
await db.commit()
|
|
logger.info("created initial admin user")
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(_: FastAPI):
|
|
global collector_task
|
|
await ensure_admin_user()
|
|
collector_task = asyncio.create_task(collector_loop(collector_stop_event))
|
|
try:
|
|
yield
|
|
finally:
|
|
collector_stop_event.set()
|
|
if collector_task:
|
|
await collector_task
|
|
|
|
|
|
app = FastAPI(title=settings.app_name, lifespan=lifespan)
|
|
use_wildcard_cors = settings.cors_origins.strip() == "*"
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"] if use_wildcard_cors else settings.cors_origins_list,
|
|
allow_credentials=False if use_wildcard_cors else True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
app.include_router(api_router, prefix=settings.api_v1_prefix)
|