Compressor/_tmp_cylinder_vibration/run_timeline.py

"""生成全时段 15 分钟槽位拼接波形 HTML。"""
from __future__ import annotations

import json
import sys
from pathlib import Path

import pandas as pd

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

from _tmp_cylinder_vibration import config, hourly, labels, timeline_html


def main() -> None:
    frame = labels.prepare_frame(
        pd.read_csv(config.OUTPUT_DIR / "vibration_features_full.csv").to_dict("records"), 15
    )
    baseline = json.loads((config.OUTPUT_DIR / "baseline_full.json").read_text(encoding="utf-8"))
    intervals = pd.read_csv(config.OUTPUT_DIR / "vibration_intervals_full.csv")
    for column in ("start_time", "end_time", "max_time"):
        intervals[column] = pd.to_datetime(intervals[column], errors="coerce")

    hourly_labels = hourly.build_hourly(intervals)
    hourly_labels.to_csv(
        config.OUTPUT_DIR / "vibration_hourly_labels.csv",
        index=False,
        date_format="%Y-%m-%d %H:%M:%S",
    )
    print(f"[hourly] {len(hourly_labels)} labels")
    timeline_html.export_timeline(frame, intervals, baseline)


if __name__ == "__main__":
    main()