178 lines
6.5 KiB
Python
178 lines
6.5 KiB
Python
|
"""
|
|||
|
|
AI Box Sophon Adapter — 自定义 FastAPI 路由
|
|||
|
|
|
|||
|
|
处理 TOML 无法描述的特殊端点：
|
|||
|
|
- POST /aibox/video_upload 视频上传（FORM-DATA）
|
|||
|
|
- POST /aibox/alarm_report 告警上报（JSON，手动解析）
|
|||
|
|
|
|||
|
|
路由前缀: /aibox（网关自动加 /api/v1）
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
import json
|
|||
|
|
import uuid
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, UploadFile, File, Form, Request
|
|||
|
|
from aiofile import AIOFile
|
|||
|
|
|
|||
|
|
from .models.video_models import VideoUploadResponse, VideoUploadResult
|
|||
|
|
from .models.alarm_models import (
|
|||
|
|
AlarmReportRequest,
|
|||
|
|
AlarmReportResponse,
|
|||
|
|
AlarmReportResult,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
# 自定义路由器，前缀与 config.toml 中的 router_prefix 保持一致
|
|||
|
|
# 最终路径由网关 register_routes() 加上全局 prefix (/api/v1) 组合而成
|
|||
|
|
router = APIRouter(prefix="/aibox", tags=["aibox_sophon_adapter"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post(
|
|||
|
|
"/video_upload",
|
|||
|
|
summary="视频上传接口",
|
|||
|
|
description="盒子告警时通过 HTTP FORM-DATA 上报视频文件，返回唯一 VideoId",
|
|||
|
|
response_model=VideoUploadResponse,
|
|||
|
|
)
|
|||
|
|
async def video_upload(
|
|||
|
|
request: Request,
|
|||
|
|
BoardIp: str = Form(..., description="盒子网络地址"),
|
|||
|
|
BoardId: str = Form(..., description="盒子唯一标识"),
|
|||
|
|
TaskSession: str = Form(..., description="所属任务标识"),
|
|||
|
|
GBDeviceId: str = Form(default="", description="盒子国标设备号（不存在则为空字符串）"),
|
|||
|
|
GBTaskChnId: str = Form(default="", description="所属任务国标通道号（不存在则为空字符串）"),
|
|||
|
|
Video: UploadFile = File(..., description="视频文件"),
|
|||
|
|
) -> VideoUploadResponse:
|
|||
|
|
"""
|
|||
|
|
视频上传接口
|
|||
|
|
|
|||
|
|
接收盒子告警时录制的视频文件，保存到本地存储，返回唯一 VideoId。
|
|||
|
|
上报告警报文时可携带该 VideoId 关联视频文件。
|
|||
|
|
|
|||
|
|
复用网关 upload 接口的本地存储机制（data/uploads/ 目录）。
|
|||
|
|
"""
|
|||
|
|
# 生成唯一 VideoId
|
|||
|
|
video_id = uuid.uuid4().hex
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 复用网关存储路径约定: data/uploads/<worker_type>/<uuid>/<filename>
|
|||
|
|
upload_root = request.app.state.upload_root
|
|||
|
|
worker_type = "aibox_sophon_adapter"
|
|||
|
|
object_key = f"{worker_type}/{video_id}/{Video.filename}"
|
|||
|
|
dest = os.path.join(upload_root, object_key)
|
|||
|
|
os.makedirs(os.path.dirname(dest), exist_ok=True)
|
|||
|
|
|
|||
|
|
# 流式写入文件（与网关 /api/v1/storage/upload 一致）
|
|||
|
|
async with AIOFile(dest, "wb") as afp:
|
|||
|
|
while chunk := await Video.read(1024 * 1024):
|
|||
|
|
await afp.write(chunk)
|
|||
|
|
|
|||
|
|
file_size = os.path.getsize(dest)
|
|||
|
|
|
|||
|
|
logger.info(
|
|||
|
|
f"[VideoUpload] 视频已保存: VideoId={video_id}, "
|
|||
|
|
f"BoardId={BoardId}, BoardIp={BoardIp}, "
|
|||
|
|
f"TaskSession={TaskSession}, File={Video.filename}, "
|
|||
|
|
f"Size={file_size} bytes, Path={object_key}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return VideoUploadResponse(
|
|||
|
|
Result=VideoUploadResult(Code=0, Desc=""),
|
|||
|
|
VideoId=video_id,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(
|
|||
|
|
f"[VideoUpload] 上传失败: BoardId={BoardId}, "
|
|||
|
|
f"BoardIp={BoardIp}, Error={e}"
|
|||
|
|
)
|
|||
|
|
return VideoUploadResponse(
|
|||
|
|
Result=VideoUploadResult(Code=-1, Desc=str(e)),
|
|||
|
|
VideoId="",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post(
|
|||
|
|
"/alarm_report",
|
|||
|
|
summary="告警上报接口",
|
|||
|
|
description="接收盒子告警报文（JSON），所有字段已在 OpenAPI 中声明，未知字段自动保留",
|
|||
|
|
response_model=AlarmReportResponse,
|
|||
|
|
)
|
|||
|
|
async def alarm_report(body: AlarmReportRequest) -> AlarmReportResponse:
|
|||
|
|
"""
|
|||
|
|
告警上报接口
|
|||
|
|
|
|||
|
|
盒子检测到违规事件后，先上传视频再上报 JSON 告警报文。
|
|||
|
|
请求体所有字段已在 Pydantic 模型中声明（含默认值），
|
|||
|
|
model_config extra=allow 确保盒子固件升级新增字段不会丢失。
|
|||
|
|
|
|||
|
|
业务处理：
|
|||
|
|
1. Pydantic 验证基础字段类型
|
|||
|
|
2. model_dump() 获取完整 dict（含额外字段）
|
|||
|
|
3. 提取关键索引字段存入 alarm_record 表
|
|||
|
|
4. 完整报文 JSON 字符串存入 alarm_data
|
|||
|
|
|
|||
|
|
返回格式：{"Result": {"Code": 0, "Desc": ""}}
|
|||
|
|
仅当 HTTP 200 + Code 非 0 时，盒子认为上报失败。
|
|||
|
|
"""
|
|||
|
|
# 获取完整报文（含 extra 字段）
|
|||
|
|
full_data = body.model_dump()
|
|||
|
|
|
|||
|
|
board_id = full_data.get("BoardId", "")
|
|||
|
|
board_ip = full_data.get("BoardIp", "")
|
|||
|
|
alarm_id = full_data.get("AlarmId") or full_data.get("UniqueId", "")
|
|||
|
|
summary = full_data.get("Summary", "")
|
|||
|
|
task_session = full_data.get("TaskSession", "")
|
|||
|
|
video_file = full_data.get("VideoFile", "")
|
|||
|
|
|
|||
|
|
media = full_data.get("Media", {})
|
|||
|
|
media_name = media.get("MediaName", "") if isinstance(media, dict) else ""
|
|||
|
|
|
|||
|
|
alarm_time = full_data.get("Time", "")
|
|||
|
|
alarm_timestamp = full_data.get("TimeStamp", 0)
|
|||
|
|
|
|||
|
|
alarm_data_str = json.dumps(full_data, ensure_ascii=False)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from .global_var import db
|
|||
|
|
if db is not None:
|
|||
|
|
async with db.get_session() as session:
|
|||
|
|
from .models.orm import AlarmRecord
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
|
|||
|
|
stmt = select(AlarmRecord).where(AlarmRecord.alarm_id == alarm_id) # type: ignore
|
|||
|
|
result = await session.execute(stmt)
|
|||
|
|
existing = result.scalar_one_or_none()
|
|||
|
|
|
|||
|
|
if existing is None:
|
|||
|
|
record = AlarmRecord(
|
|||
|
|
board_id=board_id,
|
|||
|
|
board_ip=board_ip,
|
|||
|
|
alarm_id=alarm_id,
|
|||
|
|
summary=summary,
|
|||
|
|
task_session=task_session,
|
|||
|
|
media_name=media_name,
|
|||
|
|
video_file=video_file,
|
|||
|
|
alarm_time=alarm_time,
|
|||
|
|
alarm_timestamp=alarm_timestamp,
|
|||
|
|
alarm_data=alarm_data_str,
|
|||
|
|
)
|
|||
|
|
session.add(record)
|
|||
|
|
await session.commit()
|
|||
|
|
logger.info(
|
|||
|
|
f"[AlarmReport] 新告警: AlarmId={alarm_id}, "
|
|||
|
|
f"BoardId={board_id}, Summary={summary}, "
|
|||
|
|
f"Media={media_name}, VideoFile={video_file}"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
logger.warning(
|
|||
|
|
f"[AlarmReport] 重复告警，已忽略: AlarmId={alarm_id}"
|
|||
|
|
)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"[AlarmReport] 数据库写入失败: {e}")
|
|||
|
|
|
|||
|
|
return AlarmReportResponse(
|
|||
|
|
Result=AlarmReportResult(Code=0, Desc="")
|
|||
|
|
)
|