126 lines
3.5 KiB
Python
126 lines
3.5 KiB
Python
|
"""ZED 分段录制 HTTP 控制（响应格式对齐 biomass：code/msg/data；不校验 ingest API Key）。"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from typing import Any, Dict, Optional
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, Body, Depends
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
from starlette.responses import JSONResponse
|
|||
|
|
|
|||
|
|
from app.services.zed_recording_control import (
|
|||
|
|
start_zed_recording,
|
|||
|
|
stop_zed_recording,
|
|||
|
|
zed_recording_is_running,
|
|||
|
|
)
|
|||
|
|
from app.settings import Settings, get_settings
|
|||
|
|
from app.state import app_state
|
|||
|
|
|
|||
|
|
router = APIRouter(prefix="/api/v1/zed/recording", tags=["zed-recording"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _ok(data: Dict[str, Any], msg: str = "成功") -> JSONResponse:
|
|||
|
|
return JSONResponse(
|
|||
|
|
content={
|
|||
|
|
"code": 200,
|
|||
|
|
"msg": msg,
|
|||
|
|
"data": data,
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _err(
|
|||
|
|
http_status: int,
|
|||
|
|
*,
|
|||
|
|
code: int,
|
|||
|
|
msg: str,
|
|||
|
|
data: Optional[Dict[str, Any]] = None,
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
body: Dict[str, Any] = {"code": code, "msg": msg}
|
|||
|
|
if data is not None:
|
|||
|
|
body["data"] = data
|
|||
|
|
return JSONResponse(status_code=http_status, content=body)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ZedRecordingStartRequest(BaseModel):
|
|||
|
|
"""可选覆盖分段时长；``fish_id`` 由服务端每次启动在库中递增分配。"""
|
|||
|
|
|
|||
|
|
segment_sec: Optional[float] = Field(
|
|||
|
|
None,
|
|||
|
|
ge=1.0,
|
|||
|
|
description="每段时长（秒）；缺省使用 ZED_SVO_SEGMENT_SEC",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/status")
|
|||
|
|
async def zed_recording_status() -> JSONResponse:
|
|||
|
|
data: Dict[str, Any] = {"running": zed_recording_is_running()}
|
|||
|
|
if app_state.zed_recording_fish_id is not None:
|
|||
|
|
data["fish_id"] = app_state.zed_recording_fish_id
|
|||
|
|
return _ok(data)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/start")
|
|||
|
|
async def zed_recording_start(
|
|||
|
|
body: ZedRecordingStartRequest = Body(default_factory=ZedRecordingStartRequest),
|
|||
|
|
settings: Settings = Depends(get_settings),
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
ok, msg, fish_id, _ = start_zed_recording(
|
|||
|
|
settings,
|
|||
|
|
segment_sec=body.segment_sec,
|
|||
|
|
)
|
|||
|
|
if not ok:
|
|||
|
|
if msg == "already_running":
|
|||
|
|
return _err(
|
|||
|
|
409,
|
|||
|
|
code=409,
|
|||
|
|
msg="已在录制中",
|
|||
|
|
data={"ok": False, "message": "已在录制中"},
|
|||
|
|
)
|
|||
|
|
if msg.startswith("session_db_error:"):
|
|||
|
|
return _err(
|
|||
|
|
500,
|
|||
|
|
code=500,
|
|||
|
|
msg="写入录制会话失败",
|
|||
|
|
data={"ok": False, "message": msg},
|
|||
|
|
)
|
|||
|
|
return _err(
|
|||
|
|
500,
|
|||
|
|
code=500,
|
|||
|
|
msg="启动录制失败",
|
|||
|
|
data={"ok": False, "message": msg},
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return _ok(
|
|||
|
|
{
|
|||
|
|
"ok": True,
|
|||
|
|
"message": "录制开始",
|
|||
|
|
"fish_id": fish_id,
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/stop")
|
|||
|
|
async def zed_recording_stop(settings: Settings = Depends(get_settings)) -> JSONResponse:
|
|||
|
|
ok, msg, fish_id = stop_zed_recording(settings)
|
|||
|
|
if not ok:
|
|||
|
|
if msg == "not_running":
|
|||
|
|
return _ok({"ok": False, "message": "未在录制"}, msg="未在录制")
|
|||
|
|
if msg == "stop_timeout":
|
|||
|
|
return _err(
|
|||
|
|
504,
|
|||
|
|
code=504,
|
|||
|
|
msg="停止超时",
|
|||
|
|
data={"ok": False, "message": "停止超时"},
|
|||
|
|
)
|
|||
|
|
return _err(
|
|||
|
|
500,
|
|||
|
|
code=500,
|
|||
|
|
msg="停止录制失败",
|
|||
|
|
data={"ok": False, "message": msg},
|
|||
|
|
)
|
|||
|
|
out: Dict[str, Any] = {"ok": True, "message": "录制停止"}
|
|||
|
|
if fish_id is not None:
|
|||
|
|
out["fish_id"] = fish_id
|
|||
|
|
return _ok(out)
|