提交必要文件

convert.py

@@ -0,0 +1,53 @@
+from __future__ import annotations
+
+import argparse
+import threading
+
+from video_pipeline import ConversionConfig, run_conversion_jobs
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(description="批量视频转 HLS 切片")
+    parser.add_argument("--root", default=".", help="任务根目录")
+    parser.add_argument("--jobs", type=int, default=None, help="并发转码任务数")
+    parser.add_argument("--resolution", default="1280:720", help="输出分辨率")
+    parser.add_argument("--video-bitrate", default="2000k", help="视频码率")
+    parser.add_argument("--audio-bitrate", default="128k", help="音频码率")
+    parser.add_argument("--hls-time", type=int, default=10, help="切片时长")
+    parser.add_argument("--preset", default="p4", help="NVENC preset")
+    parser.add_argument("--flat-output", action="store_true", help="关闭单集子目录模式")
+    return parser
+
+
+def main() -> int:
+    parser = build_parser()
+    args = parser.parse_args()
+
+    config = ConversionConfig(
+        root_dir=args.root,
+        parallel_jobs=args.jobs or ConversionConfig().parallel_jobs,
+        resolution=args.resolution,
+        video_bitrate=args.video_bitrate,
+        audio_bitrate=args.audio_bitrate,
+        hls_time=max(1, args.hls_time),
+        preset=args.preset,
+        episode_folder_mode=not args.flat_output,
+    )
+
+    def on_event(payload: dict) -> None:
+        kind = payload.get("kind")
+        if kind == "log":
+            line = payload.get("line")
+            if line:
+                print(line)
+
+    summary = run_conversion_jobs(config, event_callback=on_event, stop_event=threading.Event())
+    print(
+        f"\n完成：成功 {summary.succeeded}/{summary.total}，失败 {summary.failed}，"
+        f"日志 {summary.log_file}"
+    )
+    return 0 if summary.failed == 0 else 1
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())