82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
"""
|
|
Seismograph Dashboard API Router
|
|
Provides endpoints for the seismograph-specific dashboard
|
|
"""
|
|
|
|
from fastapi import APIRouter, Request, Depends, Query
|
|
from fastapi.responses import HTMLResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from sqlalchemy.orm import Session
|
|
from backend.database import get_db
|
|
from backend.models import RosterUnit
|
|
|
|
router = APIRouter(prefix="/api/seismo-dashboard", tags=["seismo-dashboard"])
|
|
templates = Jinja2Templates(directory="templates")
|
|
|
|
|
|
@router.get("/stats", response_class=HTMLResponse)
|
|
async def get_seismo_stats(request: Request, db: Session = Depends(get_db)):
|
|
"""
|
|
Returns HTML partial with seismograph statistics summary
|
|
"""
|
|
# Get all seismograph units
|
|
seismos = db.query(RosterUnit).filter_by(
|
|
device_type="seismograph",
|
|
retired=False
|
|
).all()
|
|
|
|
total = len(seismos)
|
|
deployed = sum(1 for s in seismos if s.deployed)
|
|
benched = sum(1 for s in seismos if not s.deployed)
|
|
|
|
# Count modems assigned to deployed seismographs
|
|
with_modem = sum(1 for s in seismos if s.deployed and s.deployed_with_modem_id)
|
|
without_modem = deployed - with_modem
|
|
|
|
return templates.TemplateResponse(
|
|
"partials/seismo_stats.html",
|
|
{
|
|
"request": request,
|
|
"total": total,
|
|
"deployed": deployed,
|
|
"benched": benched,
|
|
"with_modem": with_modem,
|
|
"without_modem": without_modem
|
|
}
|
|
)
|
|
|
|
|
|
@router.get("/units", response_class=HTMLResponse)
|
|
async def get_seismo_units(
|
|
request: Request,
|
|
db: Session = Depends(get_db),
|
|
search: str = Query(None)
|
|
):
|
|
"""
|
|
Returns HTML partial with filterable seismograph unit list
|
|
"""
|
|
query = db.query(RosterUnit).filter_by(
|
|
device_type="seismograph",
|
|
retired=False
|
|
)
|
|
|
|
# Apply search filter
|
|
if search:
|
|
search_lower = search.lower()
|
|
query = query.filter(
|
|
(RosterUnit.id.ilike(f"%{search}%")) |
|
|
(RosterUnit.note.ilike(f"%{search}%")) |
|
|
(RosterUnit.address.ilike(f"%{search}%"))
|
|
)
|
|
|
|
seismos = query.order_by(RosterUnit.id).all()
|
|
|
|
return templates.TemplateResponse(
|
|
"partials/seismo_unit_list.html",
|
|
{
|
|
"request": request,
|
|
"units": seismos,
|
|
"search": search or ""
|
|
}
|
|
)
|