16337
35386b8e6e
- 新增3张通知路由表模型（notify_area, camera_area_binding, area_person_binding） - 新增VLM复核服务，通过qwen3-vl-flash对告警截图二次确认 - 新增企微通知服务，告警确认后推送文本卡片给责任人 - 新增通知调度服务，编排VLM复核→查表路由→企微推送流水线 - 新增企微回调接口，支持手动结单/确认处理/标记误报 - 新增通知管理API，区域/摄像头绑定/人员绑定CRUD - 告警上报主流程（edge_compat + yudao_aiot_alarm）接入异步通知 - 扩展配置项支持VLM和企微环境变量 - 添加openai==1.68.0依赖（通过DashScope兼容端点调用）
323 lines
9.9 KiB
Python
323 lines
9.9 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, edge_compat_router
|
|
from app.routers.wechat_callback import router as wechat_callback_router
|
|
from app.routers.notify_manage import router as notify_manage_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)
|
|
|
|
# 初始化 VLM 服务
|
|
from app.services.vlm_service import get_vlm_service
|
|
vlm_svc = get_vlm_service()
|
|
vlm_svc.init(settings.vlm)
|
|
|
|
# 初始化企微服务
|
|
from app.services.wechat_service import get_wechat_service
|
|
wechat_svc = get_wechat_service()
|
|
wechat_svc.init(settings.wechat)
|
|
|
|
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)
|
|
|
|
# ==================== Edge 兼容路由 ====================
|
|
# Edge 设备使用 /api/ai/alert/edge/* 路径上报（与 WVP 一致），无需认证
|
|
app.include_router(edge_compat_router)
|
|
|
|
# ==================== 企微回调 + 通知管理路由 ====================
|
|
app.include_router(wechat_callback_router)
|
|
app.include_router(notify_manage_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")
|
|
|
|
|
|
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,
|
|
)
|