Chengfang Lu
483e027482
将 platform + CVAT 合并为单文件 docker-compose.yml，完善 .env 与 init/dev_up 脚本； 新增 docs/DEPLOY.md 与更新 README 以支持新机器部署；含数据湖示例、车队地图、 紧凑表格 UI、ADAS det_7cls 路径与批次台账等近期改动。 Co-authored-by: Cursor <cursoragent@cursor.com>
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
"""DMS 多包：data_packs.yaml + ML/workflow.registry.yaml active_packs。"""
|
||
from __future__ import annotations
|
||
|
||
from pathlib import Path
|
||
|
||
import yaml
|
||
|
||
|
||
def load_packs_registry(root: Path) -> dict:
|
||
p = root / "data_packs.yaml"
|
||
if not p.is_file():
|
||
raise SystemExit(f"缺少 {p}")
|
||
return yaml.safe_load(p.read_text(encoding="utf-8"))
|
||
|
||
|
||
def ml_workflow_path(dataset_root: Path) -> Path:
|
||
# workspace/DMS/DATASET -> DATA/HSAP/workflow.registry.yaml
|
||
return dataset_root.resolve().parent.parent.parent / "ML" / "workflow.registry.yaml"
|
||
|
||
|
||
def load_active_pack_names(dataset_root: Path, cli_packs: list[str] | None = None) -> list[str]:
|
||
if cli_packs:
|
||
return cli_packs
|
||
wf_path = ml_workflow_path(dataset_root)
|
||
if wf_path.is_file():
|
||
wf = yaml.safe_load(wf_path.read_text(encoding="utf-8"))
|
||
active = wf.get("projects", {}).get("dms", {}).get("active_packs")
|
||
if active:
|
||
return list(active)
|
||
reg = load_packs_registry(dataset_root)
|
||
return [reg["packs"][0]["name"]] if reg.get("packs") else []
|
||
|
||
|
||
def resolve_pack_dir(root: Path, pack_name: str) -> Path:
|
||
reg = load_packs_registry(root)
|
||
name = reg.get("aliases", {}).get(pack_name, pack_name)
|
||
for item in reg.get("packs", []):
|
||
if item.get("name") == name:
|
||
candidate = root / item["path"]
|
||
if candidate.is_symlink():
|
||
try:
|
||
resolved = candidate.resolve()
|
||
if resolved.is_dir():
|
||
return resolved
|
||
except OSError:
|
||
pass
|
||
if not candidate.exists():
|
||
candidate.unlink()
|
||
candidate.mkdir(parents=True, exist_ok=True)
|
||
return candidate.resolve()
|
||
candidate = root / name
|
||
if candidate.is_dir():
|
||
return candidate.resolve()
|
||
raise SystemExit(f"未知数据包: {pack_name}，已登记: {[p['name'] for p in reg.get('packs', [])]}")
|
||
|
||
|
||
def task_data_root(dataset_root: Path, pack_name: str, task_dir: str) -> Path:
|
||
return resolve_pack_dir(dataset_root, pack_name) / task_dir
|
||
|
||
|
||
def list_registered_packs(root: Path) -> list[dict]:
|
||
return load_packs_registry(root).get("packs", [])
|