modify: aibox_sophon_adapter plugins
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
"""
|
||||
模型导出 — aibox_sophon_adapter
|
||||
|
||||
包含 gen_models 自动生成的模型和手动定义的 board_models。
|
||||
"""
|
||||
from .pydantic import (
|
||||
AiboxSophonAdapterResponse,
|
||||
GetDeviceGetRequest,
|
||||
GetDeviceGetResponse,
|
||||
GetOfflineDevicesGetResponse,
|
||||
ListDevicesGetResponse,
|
||||
)
|
||||
from .board_models import (
|
||||
BoardPingRequest,
|
||||
HostDiskInfo,
|
||||
MediaInfo,
|
||||
MediaStatusInfo,
|
||||
AlgItemInfo,
|
||||
AlgTaskStatusInfo,
|
||||
TaskInfo,
|
||||
TpuInfo,
|
||||
DeviceSummary,
|
||||
)
|
||||
from .orm import BoardHeartbeat
|
||||
|
||||
__all__ = [
|
||||
# 自动生成
|
||||
"AiboxSophonAdapterResponse",
|
||||
"GetDeviceGetRequest",
|
||||
"GetDeviceGetResponse",
|
||||
"GetOfflineDevicesGetResponse",
|
||||
"ListDevicesGetResponse",
|
||||
# 手动定义
|
||||
"BoardPingRequest",
|
||||
"HostDiskInfo",
|
||||
"MediaInfo",
|
||||
"MediaStatusInfo",
|
||||
"AlgItemInfo",
|
||||
"AlgTaskStatusInfo",
|
||||
"TaskInfo",
|
||||
"TpuInfo",
|
||||
"DeviceSummary",
|
||||
# ORM
|
||||
"BoardHeartbeat",
|
||||
]
|
||||
@@ -0,0 +1,133 @@
|
||||
"""
|
||||
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="授权描述")
|
||||
|
||||
@@ -0,0 +1,31 @@
|
||||
"""
|
||||
AI Box Sophon Adapter — ORM 模型
|
||||
|
||||
SQLModel 定义设备心跳记录表。
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional
|
||||
|
||||
from sqlmodel import Field, SQLModel
|
||||
|
||||
|
||||
class BoardHeartbeat(SQLModel, table=True):
|
||||
"""设备心跳记录表"""
|
||||
__tablename__ = "board_heartbeat"
|
||||
|
||||
id: Optional[int] = Field(default=None, primary_key=True)
|
||||
board_id: str = Field(index=True, unique=True, description="设备唯一标识 (BoardId)")
|
||||
board_ip: str = Field(default="", description="设备 IP")
|
||||
board_platform: str = Field(default="", description="硬件类型")
|
||||
board_temp: str = Field(default="", description="芯片温度")
|
||||
board_type: str = Field(default="", description="接入方式 WAN/4G")
|
||||
status: str = Field(default="未知", description="设备状态")
|
||||
version: str = Field(default="", description="软件版本")
|
||||
system_info: str = Field(default="", description="系统版本信息")
|
||||
grant_code: int = Field(default=0, description="授权码")
|
||||
grant_desc: str = Field(default="", description="授权描述")
|
||||
last_ping_time: datetime = Field(default_factory=datetime.now, description="最后心跳时间")
|
||||
ping_data: str = Field(default="{}", description="完整心跳数据 JSON")
|
||||
is_online: bool = Field(default=True, description="是否在线")
|
||||
created_at: datetime = Field(default_factory=datetime.now, description="首次上报时间")
|
||||
updated_at: datetime = Field(default_factory=datetime.now, description="最后更新时间")
|
||||
@@ -0,0 +1,34 @@
|
||||
"""
|
||||
自动生成的 Pydantic 模型 — aibox_sophon_adapter
|
||||
来源: config/workers/aibox_sophon_adapter/
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any, Optional
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
class GetDeviceGetRequest(BaseModel):
|
||||
"""查询单个设备详情 — 请求体"""
|
||||
board_id: str = Field(...)
|
||||
|
||||
class GetDeviceGetResponse(BaseModel):
|
||||
"""查询单个设备详情 — 响应体"""
|
||||
board_id: str = Field(...)
|
||||
is_online: bool = Field(...)
|
||||
last_ping_time: Optional[str] = Field(default=None)
|
||||
ping_data: Optional[dict[str, Any]] = Field(default=None)
|
||||
|
||||
class GetOfflineDevicesGetResponse(BaseModel):
|
||||
"""查询离线/异常设备列表 — 响应体"""
|
||||
offline_devices: list[dict] = Field(...)
|
||||
total: int = Field(...)
|
||||
|
||||
class ListDevicesGetResponse(BaseModel):
|
||||
"""查询所有设备状态 — 响应体"""
|
||||
devices: list[dict] = Field(...)
|
||||
total: int = Field(...)
|
||||
|
||||
|
||||
# 所有响应模型的联合类型
|
||||
AiboxSophonAdapterResponse = ListDevicesGetResponse | GetOfflineDevicesGetResponse | GetDeviceGetResponse
|
||||
Reference in New Issue
Block a user