0c05463617
- 语音：序数解析（第一个/第二个等）、解析失败计数与 API detail.retry_remaining； 百度 ASR 固定 dev_pid 为普通话；SurgeryPipelineError 支持 extra 并入 HTTP detail。 - Demo：demo 路由与假 RTSP、客户端 index 与 README；BackendResolver 与配置调整。 - 可观测：消耗 TSV 日志、语音文件日志、终端 Markdown 辅助；相关测试与依赖更新。 - 注意：.env 仍被 gitignore，本地密钥不会进入本提交。 Made-with: Cursor
242 lines
8.0 KiB
Python
242 lines
8.0 KiB
Python
"""每例手术一个文本文件（制表符列）：`start_surgery` 时截断并写表头，每次时间窗识别**追加**一行。终端 Markdown 中时间戳为可读形式；落盘行内仍为 ISO 便于程序解析。
|
||
|
||
时间戳：在拉流起点记录 `time.time()`，与 `time.monotonic()` 时间窗对齐。直播 RTSP 经 OpenCV 一般无可靠绝对时码，以本机接收时刻为准。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import re
|
||
import threading
|
||
from datetime import datetime, timezone
|
||
from pathlib import Path
|
||
from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
|
||
|
||
from loguru import logger
|
||
|
||
from app.config import settings
|
||
from app.services.consumable_vision_algorithm import ClsTop3
|
||
from app.terminal_markdown import print_markdown_stderr
|
||
|
||
# 制表符分隔；时间范围用 U+2013 连接；Top2/3 仅名称；本窗消耗数量恒为 1
|
||
HEADER = "物品id\t物品名称\tTop2物品名称\tTop3物品名称\t消耗数量\t医生id\t时间戳\n"
|
||
_RANGE_SEP = "\u2013" # en dash，与样例 `00:00:00.000–00:00:45.000` 一致
|
||
|
||
_lock = threading.Lock()
|
||
|
||
|
||
def _consumption_tzinfo():
|
||
raw = (settings.consumption_log_timezone or "").strip()
|
||
if not raw:
|
||
lt = datetime.now().astimezone().tzinfo
|
||
return lt if lt is not None else timezone.utc
|
||
try:
|
||
return ZoneInfo(raw)
|
||
except ZoneInfoNotFoundError:
|
||
logger.warning("无效的 consumption_log_timezone={!r}，回退为 UTC", raw)
|
||
return timezone.utc
|
||
|
||
|
||
def format_consumption_timestamp(
|
||
camera_id: str,
|
||
wall_start_epoch: float,
|
||
wall_end_epoch: float,
|
||
) -> str:
|
||
"""落盘用：墙钟 + 配置时区 → `camXX@ISO8601–ISO8601`。"""
|
||
tz = _consumption_tzinfo()
|
||
a = datetime.fromtimestamp(wall_start_epoch, tz=tz)
|
||
b = datetime.fromtimestamp(wall_end_epoch, tz=tz)
|
||
cam = short_camera_label(camera_id)
|
||
return f"{cam}@{a.isoformat(timespec='milliseconds')}{_RANGE_SEP}{b.isoformat(timespec='milliseconds')}"
|
||
|
||
|
||
def format_consumption_timestamp_readable(
|
||
camera_id: str,
|
||
wall_start_epoch: float,
|
||
wall_end_epoch: float,
|
||
) -> str:
|
||
"""仅终端 Rich：不含 `T` 的本地可读区间 + 摄像头简名，便于人眼对时。"""
|
||
tz = _consumption_tzinfo()
|
||
a = datetime.fromtimestamp(wall_start_epoch, tz=tz)
|
||
b = datetime.fromtimestamp(wall_end_epoch, tz=tz)
|
||
cam = short_camera_label(camera_id)
|
||
|
||
def _fmt(d: datetime) -> str:
|
||
return d.strftime("%Y-%m-%d %H:%M:%S") + f".{d.microsecond // 1000:03d}"
|
||
|
||
return f"{_fmt(a)} {_RANGE_SEP} {_fmt(b)} · {cam}"
|
||
|
||
|
||
def short_camera_label(camera_id: str) -> str:
|
||
s = (camera_id or "").strip()
|
||
m = re.match(r"^or-cam-(\d+)$", s, re.IGNORECASE)
|
||
if m:
|
||
return f"cam{int(m.group(1)):02d}"
|
||
m2 = re.match(r"^cam-?0*(\d+)$", s, re.IGNORECASE)
|
||
if m2:
|
||
return f"cam{int(m2.group(1)):02d}"
|
||
alnum = re.sub(r"[^\w-]", "", s)[:12]
|
||
return alnum or "cam"
|
||
|
||
|
||
def _encode_cell(value: str) -> str:
|
||
s = (value or "").replace("\r", " ").replace("\n", " ").replace("\t", " ")
|
||
return s
|
||
|
||
|
||
def _item_id_for_row(name: str, pid: str, name_to_code: dict[str, str]) -> str:
|
||
p = (pid or "").strip()
|
||
if p:
|
||
return p
|
||
n = (name or "").strip()
|
||
if n in name_to_code:
|
||
return (name_to_code.get(n) or n).strip()
|
||
return n
|
||
|
||
|
||
def build_tsv_line(
|
||
*,
|
||
name_to_code: dict[str, str],
|
||
best: ClsTop3,
|
||
doctor_id: str,
|
||
camera_id: str,
|
||
wall_start_epoch: float,
|
||
wall_end_epoch: float,
|
||
) -> str:
|
||
id1 = _item_id_for_row(best.t1_name, best.t1_pid, name_to_code)
|
||
# 与历史样例：Top1 为「名称 置信度」四位小数
|
||
name1 = f"{(best.t1_name or '').strip()} {best.t1_conf:.4f}".strip()
|
||
n2 = (best.t2_name or "").strip()
|
||
n3 = (best.t3_name or "").strip()
|
||
ts = format_consumption_timestamp(camera_id, wall_start_epoch, wall_end_epoch)
|
||
row = [
|
||
_encode_cell(id1),
|
||
_encode_cell(name1),
|
||
_encode_cell(n2),
|
||
_encode_cell(n3),
|
||
"1",
|
||
_encode_cell(doctor_id),
|
||
_encode_cell(ts),
|
||
]
|
||
return "\t".join(row) + "\n"
|
||
|
||
|
||
def _safe_surgery_path_segment(surgery_id: str) -> str:
|
||
s = (surgery_id or "unknown").strip() or "unknown"
|
||
s = re.sub(r"[^\w\-.@]", "_", s)
|
||
return s[:200] if len(s) > 200 else s
|
||
|
||
|
||
def resolved_consumption_log_path(surgery_id: str) -> Path:
|
||
raw = (settings.consumption_tsv_log_path or "logs/consumption_{surgery_id}.txt").strip()
|
||
safe = _safe_surgery_path_segment(surgery_id)
|
||
if "{surgery_id}" in raw:
|
||
raw = raw.replace("{surgery_id}", safe)
|
||
else:
|
||
p0 = Path(raw)
|
||
if p0.suffix:
|
||
raw = str(p0.with_name(f"{p0.stem}_{safe}{p0.suffix}"))
|
||
else:
|
||
raw = f"{raw.rstrip('/')}_{safe}.txt"
|
||
p = Path(raw).expanduser()
|
||
if not p.is_absolute():
|
||
p = Path.cwd() / p
|
||
return p
|
||
|
||
|
||
def init_consumption_log_file(surgery_id: str) -> None:
|
||
"""新手术开始：截断该手术对应文件并写入表头（一次）。"""
|
||
if not settings.consumption_tsv_log_enabled:
|
||
return
|
||
path = resolved_consumption_log_path(surgery_id)
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
with _lock:
|
||
with path.open("w", encoding="utf-8") as f:
|
||
f.write(HEADER)
|
||
|
||
|
||
def append_consumption_tsv_line(surgery_id: str, line: str) -> None:
|
||
if not settings.consumption_tsv_log_enabled:
|
||
return
|
||
path = resolved_consumption_log_path(surgery_id)
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
with _lock:
|
||
with path.open("a", encoding="utf-8") as f:
|
||
f.write(line)
|
||
|
||
|
||
def _md_cell(value: str) -> str:
|
||
"""避免破坏 Markdown 表格的 | 与换行。"""
|
||
s = (value or "").replace("\r", " ").replace("\n", " ").replace("|", "｜")
|
||
return s
|
||
|
||
|
||
def build_consumption_markdown(
|
||
*,
|
||
name_to_code: dict[str, str],
|
||
best: ClsTop3,
|
||
doctor_id: str,
|
||
camera_id: str,
|
||
wall_start_epoch: float,
|
||
wall_end_epoch: float,
|
||
) -> str:
|
||
"""终端用：Top1 含 id/名称/置信度；Top2/3 仅名称；消耗数量恒为 1。"""
|
||
id1 = _item_id_for_row(best.t1_name, best.t1_pid, name_to_code)
|
||
n1 = (best.t1_name or "").strip()
|
||
has2 = bool((best.t2_name or "").strip())
|
||
has3 = bool((best.t3_name or "").strip())
|
||
n2 = (best.t2_name or "").strip() if has2 else ""
|
||
n3 = (best.t3_name or "").strip() if has3 else ""
|
||
dash = "—"
|
||
ts = format_consumption_timestamp_readable(camera_id, wall_start_epoch, wall_end_epoch)
|
||
return "\n".join(
|
||
[
|
||
"| Top1 物品id | Top1 物品名称 | Top1 置信度 | Top2 物品名称 | Top3 物品名称 | 消耗数量 | 医生id | 时间戳 |",
|
||
"| :--- | :--- | ---: | :--- | :--- | ---: | :--- | :--- |",
|
||
"| {} | {} | {:.4f} | {} | {} | 1 | {} | {} |".format(
|
||
_md_cell(id1),
|
||
_md_cell(n1),
|
||
best.t1_conf,
|
||
_md_cell(n2) if has2 else dash,
|
||
_md_cell(n3) if has3 else dash,
|
||
_md_cell(doctor_id),
|
||
_md_cell(ts),
|
||
),
|
||
"",
|
||
]
|
||
)
|
||
|
||
|
||
def append_consumption_window(
|
||
*,
|
||
surgery_id: str,
|
||
name_to_code: dict[str, str],
|
||
best: ClsTop3,
|
||
doctor_id: str,
|
||
camera_id: str,
|
||
wall_start_epoch: float,
|
||
wall_end_epoch: float,
|
||
) -> None:
|
||
if not settings.consumption_tsv_log_enabled and not settings.consumption_log_markdown_terminal:
|
||
return
|
||
if settings.consumption_tsv_log_enabled:
|
||
line = build_tsv_line(
|
||
name_to_code=name_to_code,
|
||
best=best,
|
||
doctor_id=doctor_id,
|
||
camera_id=camera_id,
|
||
wall_start_epoch=wall_start_epoch,
|
||
wall_end_epoch=wall_end_epoch,
|
||
)
|
||
append_consumption_tsv_line(surgery_id, line)
|
||
if settings.consumption_log_markdown_terminal:
|
||
print_markdown_stderr(
|
||
build_consumption_markdown(
|
||
name_to_code=name_to_code,
|
||
best=best,
|
||
doctor_id=doctor_id,
|
||
camera_id=camera_id,
|
||
wall_start_epoch=wall_start_epoch,
|
||
wall_end_epoch=wall_end_epoch,
|
||
),
|
||
)
|