vent_ai_platform/scripts/seed_video_platform.py

"""向当前数据库写入工业视频平台模拟演示数据。

用法（在项目根目录执行）：

    python scripts/seed_video_platform.py

脚本幂等：先确保矿井主数据存在（已存在则跳过基础演示数据），
再写入视频通道、识别抓拍、通行与操作记录、分析事件。
"""

from __future__ import annotations

import sys
from pathlib import Path

PROJECT_ROOT = Path(__file__).resolve().parents[1]
if str(PROJECT_ROOT) not in sys.path:
    sys.path.insert(0, str(PROJECT_ROOT))

from app.db.init_db import init_db  # noqa: E402
from app.db.seed import seed_demo_data  # noqa: E402
from app.db.seed_video_platform import seed_video_platform_demo  # noqa: E402
from app.db.session import SessionLocal  # noqa: E402


def main() -> None:
    init_db()
    with SessionLocal() as db:
        seed_demo_data(db)
        result = seed_video_platform_demo(db)
    print("工业视频平台演示数据：", result)


if __name__ == "__main__":
    main()