feat: 初始化告警平台后端项目
- 创建 FastAPI 项目结构 - 实现告警数据模型(SQLAlchemy) - 实现 multipart/form-data 告警接收接口 - 实现阿里云 OSS 图片上传模块 - 实现告警查询和处理 API - 实现异步大模型分析模块
This commit is contained in:
182
app/main.py
Normal file
182
app/main.py
Normal file
@@ -0,0 +1,182 @@
|
||||
from fastapi import FastAPI, UploadFile, File, Form, HTTPException, Depends, Query
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.responses import FileResponse
|
||||
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,
|
||||
)
|
||||
from app.services.alert_service import get_alert_service
|
||||
from app.services.ai_analyzer import trigger_async_analysis
|
||||
from app.utils.logger import logger
|
||||
import json
|
||||
|
||||
|
||||
app = FastAPI(
|
||||
title="AI 告警平台",
|
||||
description="接收边缘端告警,提供告警查询与处理能力",
|
||||
version="1.0.0",
|
||||
)
|
||||
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=["*"],
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
|
||||
def get_alert_service():
|
||||
return get_alert_service()
|
||||
|
||||
|
||||
@app.on_event("startup")
|
||||
async def startup():
|
||||
init_db()
|
||||
logger.info("AI 告警平台启动")
|
||||
|
||||
|
||||
@app.get("/health", response_model=HealthResponse)
|
||||
async def health_check():
|
||||
db_status = "healthy"
|
||||
try:
|
||||
engine = get_engine()
|
||||
with engine.connect() as conn:
|
||||
conn.execute("SELECT 1")
|
||||
except Exception as e:
|
||||
db_status = f"unhealthy: {e}"
|
||||
|
||||
return HealthResponse(
|
||||
status="healthy" if db_status == "healthy" else "degraded",
|
||||
database=db_status,
|
||||
)
|
||||
|
||||
|
||||
@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_service),
|
||||
):
|
||||
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_service),
|
||||
):
|
||||
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/{alert_id}", response_model=AlertResponse)
|
||||
async def get_alert(
|
||||
alert_id: int,
|
||||
service=Depends(get_alert_service),
|
||||
):
|
||||
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_service),
|
||||
):
|
||||
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/statistics", response_model=AlertStatisticsResponse)
|
||||
async def get_statistics(
|
||||
service=Depends(get_alert_service),
|
||||
):
|
||||
stats = service.get_statistics()
|
||||
return AlertStatisticsResponse(**stats)
|
||||
|
||||
|
||||
@app.get("/api/v1/alerts/{alert_id}/image")
|
||||
async def get_alert_image(
|
||||
alert_id: int,
|
||||
service=Depends(get_alert_service),
|
||||
):
|
||||
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)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import uvicorn
|
||||
uvicorn.run(
|
||||
"app.main:app",
|
||||
host=settings.app.host,
|
||||
port=settings.app.port,
|
||||
reload=settings.app.debug,
|
||||
)
|
||||
Reference in New Issue
Block a user