16337
0f5e3ebce2
- 挂载 Edge 截图目录为 /captures 静态文件（COS 不可用时回退） - 挂载 /uploads 静态文件目录 - _alarm_to_camel 支持 local: 前缀转 /captures/ URL - 告警分页/详情/处理/删除接口兼容前端旧字段名（id、cameraId、status 等） - 设备告警汇总添加前端兼容别名（cameraId、pendingCount 等） Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
308 lines
9.4 KiB
Python
308 lines
9.4 KiB
Python
import asyncio
|
||
from contextlib import asynccontextmanager
|
||
from fastapi import FastAPI, UploadFile, File, Form, HTTPException, Depends, Query, WebSocket, WebSocketDisconnect
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from fastapi.responses import FileResponse
|
||
from fastapi.staticfiles import StaticFiles
|
||
from typing import Optional
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
from app.config import settings
|
||
from app.models import init_db, get_engine, AlertStatus
|
||
from app.schemas import (
|
||
AlertCreate,
|
||
AlertResponse,
|
||
AlertListResponse,
|
||
AlertHandleRequest,
|
||
AlertStatisticsResponse,
|
||
HealthResponse,
|
||
DeviceResponse,
|
||
DeviceListResponse,
|
||
DeviceStatisticsResponse,
|
||
)
|
||
from app.services.alert_service import alert_service, get_alert_service
|
||
from app.services.alarm_event_service import alarm_event_service, get_alarm_event_service
|
||
from app.services.ai_analyzer import trigger_async_analysis
|
||
from app.services.notification_service import get_notification_service
|
||
from app.services.device_service import get_device_service
|
||
from app.utils.logger import logger
|
||
from app.routers import yudao_alert_router, yudao_auth_router, yudao_aiot_alarm_router, yudao_aiot_edge_router, yudao_aiot_storage_router
|
||
from app.yudao_compat import yudao_exception_handler
|
||
import json
|
||
|
||
|
||
# 全局服务实例
|
||
notification_service = get_notification_service()
|
||
device_service = get_device_service()
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
"""应用生命周期管理"""
|
||
# 启动
|
||
init_db()
|
||
logger.info("数据库初始化完成")
|
||
|
||
# 设置事件循环（用于从同步代码调用异步方法，如 WebSocket 通知）
|
||
loop = asyncio.get_event_loop()
|
||
notification_service.set_event_loop(loop)
|
||
|
||
logger.info("AI 告警平台启动完成")
|
||
|
||
yield
|
||
|
||
# 关闭
|
||
logger.info("AI 告警平台已关闭")
|
||
|
||
|
||
app = FastAPI(
|
||
title="AI 告警平台",
|
||
description="接收边缘端告警，提供告警查询与处理能力，支持 WebSocket 实时推送",
|
||
version="2.0.0",
|
||
lifespan=lifespan,
|
||
)
|
||
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"],
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
|
||
)
|
||
|
||
# ==================== 芋道兼容路由 ====================
|
||
# 提供与芋道主平台一致的接口规范，便于前端统一对接
|
||
app.include_router(yudao_auth_router)
|
||
app.include_router(yudao_alert_router)
|
||
|
||
# ==================== AIoT 统一路由 ====================
|
||
# aiot 命名空间下的新路由，与旧路由并存
|
||
app.include_router(yudao_aiot_alarm_router)
|
||
app.include_router(yudao_aiot_edge_router)
|
||
app.include_router(yudao_aiot_storage_router)
|
||
|
||
# 注册芋道格式异常处理器
|
||
app.add_exception_handler(HTTPException, yudao_exception_handler)
|
||
|
||
# ==================== 静态文件（本地截图） ====================
|
||
_uploads_dir = Path("uploads")
|
||
_uploads_dir.mkdir(parents=True, exist_ok=True)
|
||
app.mount("/uploads", StaticFiles(directory=str(_uploads_dir)), name="uploads")
|
||
|
||
# Edge 本地截图目录（COS 未配置时的回退访问路径）
|
||
_edge_captures_dir = Path(r"C:\Users\16337\PycharmProjects\ai_edge\data\captures")
|
||
if _edge_captures_dir.exists():
|
||
app.mount("/captures", StaticFiles(directory=str(_edge_captures_dir)), name="captures")
|
||
|
||
|
||
def get_alert_svc():
|
||
return alert_service
|
||
|
||
|
||
def get_device_svc():
|
||
return device_service
|
||
|
||
|
||
@app.get("/health", response_model=HealthResponse)
|
||
async def health_check():
|
||
from sqlalchemy import text
|
||
|
||
db_status = "healthy"
|
||
try:
|
||
engine = get_engine()
|
||
with engine.connect() as conn:
|
||
conn.execute(text("SELECT 1"))
|
||
except Exception as e:
|
||
db_status = f"unhealthy: {e}"
|
||
|
||
return HealthResponse(
|
||
status="healthy" if db_status == "healthy" else "degraded",
|
||
database=db_status,
|
||
websocket_connections=notification_service.manager.connection_count,
|
||
)
|
||
|
||
|
||
@app.post("/api/v1/alerts", response_model=AlertResponse)
|
||
async def create_alert(
|
||
data: str = Form(..., description="JSON数据"),
|
||
snapshot: Optional[UploadFile] = File(None),
|
||
service=Depends(get_alert_svc),
|
||
):
|
||
try:
|
||
alert_data = json.loads(data)
|
||
alert_create = AlertCreate(**alert_data)
|
||
except json.JSONDecodeError as e:
|
||
raise HTTPException(status_code=400, detail=f"JSON解析失败: {e}")
|
||
|
||
snapshot_data = None
|
||
if snapshot:
|
||
snapshot_data = await snapshot.read()
|
||
|
||
alert = service.create_alert(alert_create, snapshot_data)
|
||
|
||
if alert.snapshot_url and settings.ai_model.endpoint:
|
||
trigger_async_analysis(
|
||
alert_id=alert.id,
|
||
snapshot_url=alert.snapshot_url,
|
||
alert_info={
|
||
"camera_id": alert.camera_id,
|
||
"roi_id": alert.roi_id,
|
||
"alert_type": alert.alert_type,
|
||
"confidence": alert.confidence,
|
||
"duration_minutes": alert.duration_minutes,
|
||
},
|
||
)
|
||
|
||
return AlertResponse(**alert.to_dict())
|
||
|
||
|
||
@app.get("/api/v1/alerts", response_model=AlertListResponse)
|
||
async def list_alerts(
|
||
camera_id: Optional[str] = Query(None, description="摄像头ID"),
|
||
alert_type: Optional[str] = Query(None, description="告警类型"),
|
||
status: Optional[str] = Query(None, description="处理状态"),
|
||
start_time: Optional[datetime] = Query(None, description="开始时间"),
|
||
end_time: Optional[datetime] = Query(None, description="结束时间"),
|
||
page: int = Query(1, ge=1),
|
||
page_size: int = Query(20, ge=1, le=100),
|
||
service=Depends(get_alert_svc),
|
||
):
|
||
alerts, total = service.get_alerts(
|
||
camera_id=camera_id,
|
||
alert_type=alert_type,
|
||
status=status,
|
||
start_time=start_time,
|
||
end_time=end_time,
|
||
page=page,
|
||
page_size=page_size,
|
||
)
|
||
|
||
return AlertListResponse(
|
||
alerts=[AlertResponse(**a.to_dict()) for a in alerts],
|
||
total=total,
|
||
page=page,
|
||
page_size=page_size,
|
||
)
|
||
|
||
|
||
@app.get("/api/v1/alerts/statistics", response_model=AlertStatisticsResponse)
|
||
async def get_statistics(
|
||
service=Depends(get_alert_svc),
|
||
):
|
||
stats = service.get_statistics()
|
||
return AlertStatisticsResponse(**stats)
|
||
|
||
|
||
@app.get("/api/v1/alerts/{alert_id}", response_model=AlertResponse)
|
||
async def get_alert(
|
||
alert_id: int,
|
||
service=Depends(get_alert_svc),
|
||
):
|
||
alert = service.get_alert(alert_id)
|
||
if not alert:
|
||
raise HTTPException(status_code=404, detail="告警不存在")
|
||
return AlertResponse(**alert.to_dict())
|
||
|
||
|
||
@app.put("/api/v1/alerts/{alert_id}/handle", response_model=AlertResponse)
|
||
async def handle_alert(
|
||
alert_id: int,
|
||
handle_data: AlertHandleRequest,
|
||
handled_by: Optional[str] = Query(None, description="处理人"),
|
||
service=Depends(get_alert_svc),
|
||
):
|
||
alert = service.handle_alert(alert_id, handle_data, handled_by)
|
||
if not alert:
|
||
raise HTTPException(status_code=404, detail="告警不存在")
|
||
return AlertResponse(**alert.to_dict())
|
||
|
||
|
||
@app.get("/api/v1/alerts/{alert_id}/image")
|
||
async def get_alert_image(
|
||
alert_id: int,
|
||
service=Depends(get_alert_svc),
|
||
):
|
||
alert = service.get_alert(alert_id)
|
||
if not alert:
|
||
raise HTTPException(status_code=404, detail="告警不存在")
|
||
|
||
if not alert.snapshot_path:
|
||
raise HTTPException(status_code=404, detail="图片不存在")
|
||
|
||
return FileResponse(alert.snapshot_path)
|
||
|
||
|
||
# ==================== WebSocket 端点 ====================
|
||
|
||
@app.websocket("/ws/alerts")
|
||
async def websocket_alerts(websocket: WebSocket):
|
||
"""WebSocket 连接，接收实时告警和设备状态推送"""
|
||
await notification_service.manager.connect(websocket)
|
||
try:
|
||
while True:
|
||
# 保持连接，接收客户端心跳
|
||
data = await websocket.receive_text()
|
||
if data == "ping":
|
||
await websocket.send_text("pong")
|
||
except WebSocketDisconnect:
|
||
notification_service.manager.disconnect(websocket)
|
||
except Exception as e:
|
||
logger.warning(f"WebSocket 异常: {e}")
|
||
notification_service.manager.disconnect(websocket)
|
||
|
||
|
||
# ==================== 设备管理端点 ====================
|
||
|
||
@app.get("/api/v1/devices", response_model=DeviceListResponse)
|
||
async def list_devices(
|
||
status: Optional[str] = Query(None, description="设备状态: online/offline/error"),
|
||
page: int = Query(1, ge=1),
|
||
page_size: int = Query(20, ge=1, le=100),
|
||
service=Depends(get_device_svc),
|
||
):
|
||
"""获取设备列表"""
|
||
devices, total = service.get_devices(
|
||
status=status,
|
||
page=page,
|
||
page_size=page_size,
|
||
)
|
||
return DeviceListResponse(
|
||
devices=[DeviceResponse(**d.to_dict()) for d in devices],
|
||
total=total,
|
||
page=page,
|
||
page_size=page_size,
|
||
)
|
||
|
||
|
||
@app.get("/api/v1/devices/statistics", response_model=DeviceStatisticsResponse)
|
||
async def get_device_statistics(
|
||
service=Depends(get_device_svc),
|
||
):
|
||
"""获取设备统计"""
|
||
stats = service.get_statistics()
|
||
return DeviceStatisticsResponse(**stats)
|
||
|
||
|
||
@app.get("/api/v1/devices/{device_id}", response_model=DeviceResponse)
|
||
async def get_device(
|
||
device_id: str,
|
||
service=Depends(get_device_svc),
|
||
):
|
||
"""获取设备详情"""
|
||
device = service.get_device(device_id)
|
||
if not device:
|
||
raise HTTPException(status_code=404, detail="设备不存在")
|
||
return DeviceResponse(**device.to_dict())
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
uvicorn.run(
|
||
"app.main:app",
|
||
host=settings.app.host,
|
||
port=settings.app.port,
|
||
reload=settings.app.debug,
|
||
)
|