border_Inspection/server/routers/dashboard.py

from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy.orm import Session
from typing import List, Optional, Dict, Any
from datetime import datetime, timedelta
from core.database import get_db
from models.device import Device
from models.algorithm import Algorithm
from models.event import Event

router = APIRouter()

@router.get("/kpi", summary="获取主要KPI指标")
async def get_dashboard_kpi(db: Session = Depends(get_db)):
    """获取仪表板主要KPI指标"""
    try:
        # 设备统计
        total_devices = db.query(Device).count()
        online_devices = db.query(Device).filter(Device.status == "online").count()
        
        # 算法统计
        total_algorithms = db.query(Algorithm).count()
        active_algorithms = db.query(Algorithm).filter(Algorithm.is_enabled == True).count()
        
        # 事件统计
        total_events = db.query(Event).count()
        today = datetime.now().date()
        today_events = db.query(Event).filter(
            Event.created_at >= today
        ).count()
        
        # 告警事件统计
        alert_events = db.query(Event).filter(Event.is_alert == True).count()
        resolved_events = db.query(Event).filter(Event.status == "resolved").count()
        
        return {
            "total_devices": total_devices,
            "online_devices": online_devices,
            "total_algorithms": total_algorithms,
            "active_algorithms": active_algorithms,
            "total_events": total_events,
            "today_events": today_events,
            "alert_events": alert_events,
            "resolved_events": resolved_events
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取KPI数据失败: {str(e)}")

@router.get("/alarm-trend", summary="获取告警趋势统计")
async def get_alarm_trend(
    days: int = Query(7, ge=1, le=30, description="统计天数"),
    db: Session = Depends(get_db)
):
    """获取告警趋势统计数据"""
    try:
        # TODO: 实现真实的告警趋势统计
        # 当前返回模拟数据
        end_date = datetime.now().date()
        start_date = end_date - timedelta(days=days-1)
        
        dates = []
        alarms = []
        resolved = []
        
        for i in range(days):
            current_date = start_date + timedelta(days=i)
            dates.append(current_date.strftime("%Y-%m-%d"))
            # 模拟数据
            alarms.append(10 + (i * 2) % 20)
            resolved.append(8 + (i * 2) % 15)
        
        return {
            "dates": dates,
            "alarms": alarms,
            "resolved": resolved
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取告警趋势数据失败: {str(e)}")

@router.get("/camera-stats", summary="获取摄像头统计")
async def get_camera_stats(db: Session = Depends(get_db)):
    """获取摄像头统计数据"""
    try:
        # 设备统计
        total_cameras = db.query(Device).filter(Device.device_type == "camera").count()
        online_cameras = db.query(Device).filter(
            Device.device_type == "camera",
            Device.status == "online"
        ).count()
        offline_cameras = total_cameras - online_cameras
        
        # TODO: 实现按位置统计
        # 当前返回模拟数据
        by_location = [
            {"location": "港口区", "total": 45, "online": 42},
            {"location": "码头区", "total": 38, "online": 35},
            {"location": "办公区", "total": 23, "online": 21}
        ]
        
        return {
            "total_cameras": total_cameras,
            "online_cameras": online_cameras,
            "offline_cameras": offline_cameras,
            "by_location": by_location
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取摄像头统计数据失败: {str(e)}")

@router.get("/algorithm-stats", summary="获取算法统计")
async def get_algorithm_stats(db: Session = Depends(get_db)):
    """获取算法统计数据"""
    try:
        total_algorithms = db.query(Algorithm).count()
        active_algorithms = db.query(Algorithm).filter(Algorithm.is_enabled == True).count()
        
        # TODO: 实现按类型统计
        # 当前返回模拟数据
        by_type = [
            {"type": "目标检测", "count": 3, "accuracy": 95.2},
            {"type": "行为识别", "count": 2, "accuracy": 88.7},
            {"type": "越界检测", "count": 3, "accuracy": 92.1}
        ]
        
        return {
            "total_algorithms": total_algorithms,
            "active_algorithms": active_algorithms,
            "by_type": by_type
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取算法统计数据失败: {str(e)}")

@router.get("/event-hotspots", summary="获取事件热点统计")
async def get_event_hotspots(db: Session = Depends(get_db)):
    """获取事件热点统计数据"""
    try:
        # TODO: 实现真实的事件热点统计
        # 当前返回模拟数据
        hotspots = [
            {
                "location": "港口A区",
                "event_count": 45,
                "severity": "high",
                "coordinates": {"lat": 31.2304, "lng": 121.4737}
            },
            {
                "location": "码头B区",
                "event_count": 32,
                "severity": "medium",
                "coordinates": {"lat": 31.2404, "lng": 121.4837}
            },
            {
                "location": "办公区C区",
                "event_count": 18,
                "severity": "low",
                "coordinates": {"lat": 31.2204, "lng": 121.4637}
            }
        ]
        
        return {
            "hotspots": hotspots
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取事件热点数据失败: {str(e)}")