134 lines
6.1 KiB
Python
134 lines
6.1 KiB
Python
"""
|
|
AI Box Sophon Adapter — 设备心跳数据 Pydantic 模型（手动定义）
|
|
|
|
定义 BoardPingRequest 及其嵌套子结构。
|
|
这些模型无法通过 gen_models 自动生成（嵌套层次过深），需要手动维护。
|
|
|
|
对应 MQTT 主题 /board_ping 的上报数据结构。
|
|
|
|
命名约定:
|
|
- Pydantic 模型类名以 Info 结尾（如 HostDiskInfo），避免与 JSON 字段名冲突
|
|
- 字段名直接对应 JSON key，无需 alias
|
|
"""
|
|
from typing import Optional
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
# ============================================================
|
|
# 心跳数据子结构
|
|
# ============================================================
|
|
|
|
class HostDiskInfo(BaseModel):
|
|
"""设备硬盘使用情况"""
|
|
Available: int = Field(..., description="可用空间 (kB)")
|
|
Total: int = Field(..., description="总空间 (kB)")
|
|
Used: int = Field(..., description="已用空间 (kB)")
|
|
|
|
|
|
class MediaStatusInfo(BaseModel):
|
|
"""通道状态描述"""
|
|
label: str = Field(..., description="状态标签，如 '正常'")
|
|
type: int = Field(..., description="状态类型码")
|
|
|
|
|
|
class MediaInfo(BaseModel):
|
|
"""设备通道信息"""
|
|
MediaName: str = Field(..., description="通道名称")
|
|
MediaStatus: MediaStatusInfo = Field(..., description="通道状态")
|
|
MediaUrl: str = Field(..., description="通道视频流地址")
|
|
SubId: str = Field(default="", description="关联 GB28181 通道号")
|
|
|
|
|
|
class AlgItemInfo(BaseModel):
|
|
"""算法配置信息"""
|
|
lineDesc: str = Field(default="", description="线描述")
|
|
lineRequired: bool = Field(default=False, description="是否需要线")
|
|
majorId: int = Field(..., description="主 ID")
|
|
majorOnly: bool = Field(default=False, description="仅主算法")
|
|
minorId: int = Field(..., description="副 ID")
|
|
name: str = Field(..., description="算法名称")
|
|
zoneDesc: str = Field(default="", description="区域描述")
|
|
zoneRequired: bool = Field(default=False, description="是否需要区域")
|
|
|
|
|
|
class AlgTaskStatusInfo(BaseModel):
|
|
"""任务状态描述"""
|
|
label: str = Field(..., description="状态标签，如 '运行中'")
|
|
type: int = Field(..., description="状态类型码")
|
|
|
|
|
|
class TaskInfo(BaseModel):
|
|
"""设备任务信息"""
|
|
AlgItem: list[AlgItemInfo] = Field(default_factory=list, description="算法配置列表")
|
|
AlgTaskSession: str = Field(..., description="任务号")
|
|
AlgTaskStatus: AlgTaskStatusInfo = Field(..., description="任务状态")
|
|
|
|
|
|
class TpuInfo(BaseModel):
|
|
"""TPU 算力资源使用情况"""
|
|
mem_total: int = Field(..., description="内存总量")
|
|
mem_usage: int = Field(..., description="内存使用量")
|
|
tpu_usage: int = Field(..., description="TPU 使用率")
|
|
|
|
|
|
# ============================================================
|
|
# 心跳请求体（MQTT 上报的完整数据结构）
|
|
# ============================================================
|
|
|
|
class BoardPingRequest(BaseModel):
|
|
"""
|
|
AI Box 设备心跳上报数据
|
|
|
|
设备通过 MQTT 主题 /board_ping 定期上报。
|
|
上报间隔默认 5 秒。
|
|
|
|
包含字段：设备标识、IP、温度、硬盘、内存、GPS、
|
|
通道列表、任务列表、TPU 算力等。
|
|
"""
|
|
BoardId: str = Field(..., description="当前盒子唯一标识")
|
|
BoardIp: str = Field(..., description="设备 IP 地址")
|
|
BoardPlatform: str = Field(..., description="当前设备硬件类型，如 'BM-BOX'")
|
|
BoardTemp: str = Field(..., description="当前设备芯片温度，如 '44 (C)'")
|
|
BoardType: str = Field(..., description="当前设备接入方式 WAN/4G")
|
|
GBClientId: str = Field(default="", description="GB28181 设备号，配置后透传")
|
|
GrantCode: int = Field(..., description="授权码")
|
|
GrantDesc: str = Field(..., description="授权描述，如 '已授权'")
|
|
HostDisk: HostDiskInfo = Field(..., description="当前设备硬盘情况")
|
|
HostMemory: list[float] = Field(..., description="当前设备内存使用 [已使用(kB), 总利用率]")
|
|
Key: str = Field(..., description="运行状态关键字，如 '运行中'")
|
|
Latitude: float = Field(default=0.0, description="GPS 纬度")
|
|
Longitude: float = Field(default=0.0, description="GPS 经度")
|
|
Malloc: int = Field(..., description="当前应用 VmRSS 信息 (kB)")
|
|
Medias: list[MediaInfo] = Field(default_factory=list, description="当前设备通道信息")
|
|
Pid: int = Field(..., description="本次应用 PID")
|
|
Se6ip: str = Field(default="", description="特定设备时透传主控端 IP")
|
|
Status: str = Field(..., description="设备状态，如 '在线'")
|
|
System: str = Field(..., description="当前设备系统版本")
|
|
Tasks: list[TaskInfo] = Field(default_factory=list, description="当前设备任务信息")
|
|
Time: int = Field(..., description="当前时间戳（毫秒）")
|
|
Tpu: list[TpuInfo] = Field(default_factory=list, description="当前设备算力资源使用情况")
|
|
Version: str = Field(..., description="软件版本")
|
|
angleCourse: float = Field(default=0.0, description="GPS 角度")
|
|
kSpeed: float = Field(default=0.0, description="GPS 速度 (km/h)")
|
|
nSpeed: float = Field(default=0.0, description="GPS 速度 (m/s)")
|
|
|
|
|
|
# ============================================================
|
|
# 设备状态摘要（API 响应用）
|
|
# ============================================================
|
|
|
|
class DeviceSummary(BaseModel):
|
|
"""设备状态摘要（用于列表 API 响应）"""
|
|
board_id: str = Field(..., description="设备唯一标识")
|
|
board_ip: str = Field(..., description="设备 IP")
|
|
board_platform: str = Field(..., description="硬件类型")
|
|
board_type: str = Field(..., description="接入方式")
|
|
board_temp: str = Field(..., description="芯片温度")
|
|
status: str = Field(..., description="设备状态")
|
|
version: str = Field(..., description="软件版本")
|
|
is_online: bool = Field(..., description="是否在线")
|
|
last_ping_time: Optional[str] = Field(default=None, description="最后心跳时间")
|
|
grant_code: int = Field(..., description="授权码")
|
|
grant_desc: str = Field(..., description="授权描述")
|
|
|