fengchuanhn@gmail.com
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
批量生成封面模板预览图（CLI，对齐 Electron 资源目录生成）
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import argparse
|
|
import subprocess
|
|
|
|
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
COVER_GENERATOR = os.path.join(SCRIPT_DIR, "cover_generator.py")
|
|
|
|
COVER_STYLES = ["default", "blur-bg", "outline", "gradient", "split"]
|
|
|
|
STYLE_CONFIGS = {
|
|
"default": {"titlePosition": "top", "titleColor": "#FFFFFF", "titleFontSize": 72},
|
|
"blur-bg": {"titlePosition": "top", "titleColor": "#FFFFFF", "backgroundBlurEnabled": True},
|
|
"outline": {
|
|
"titlePosition": "top",
|
|
"titleColor": "#FFFFFF",
|
|
"titleStrokeWidth": 4,
|
|
"titleStrokeColor": "#000000",
|
|
},
|
|
"gradient": {"titlePosition": "top", "titleColor": "#FFFFFF"},
|
|
"split": {"titlePosition": "center", "titleColor": "#FFFFFF"},
|
|
}
|
|
|
|
|
|
def generate_all_cover_previews(video_path: str, preview_dir: str, demo_title: str) -> bool:
|
|
os.makedirs(preview_dir, exist_ok=True)
|
|
python_exe = os.environ.get("AICLIENT_PYTHON_PATH") or sys.executable
|
|
ok_count = 0
|
|
|
|
for style in COVER_STYLES:
|
|
out_path = os.path.join(preview_dir, f"cover-preview-{style}.jpg")
|
|
config = dict(STYLE_CONFIGS.get(style, STYLE_CONFIGS["default"]))
|
|
cmd = [
|
|
python_exe,
|
|
COVER_GENERATOR,
|
|
"--video",
|
|
video_path,
|
|
"--title",
|
|
demo_title,
|
|
"--output",
|
|
out_path,
|
|
"--config",
|
|
json.dumps(config, ensure_ascii=False),
|
|
"--timestamp",
|
|
"1",
|
|
]
|
|
try:
|
|
r = subprocess.run(cmd, cwd=SCRIPT_DIR, capture_output=True, text=True, timeout=120)
|
|
if r.returncode == 0 and os.path.isfile(out_path):
|
|
print(f"✓ {out_path}", file=sys.stderr)
|
|
ok_count += 1
|
|
else:
|
|
print(f"✗ {style}: {r.stderr or r.stdout}", file=sys.stderr)
|
|
except Exception as e:
|
|
print(f"✗ {style}: {e}", file=sys.stderr)
|
|
|
|
return ok_count > 0
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Generate cover template preview images")
|
|
parser.add_argument("--video", required=True, help="Source video path")
|
|
parser.add_argument("--output-dir", required=True, help="Output directory")
|
|
parser.add_argument("--title", default="封面预览标题示例", help="Demo title text")
|
|
args = parser.parse_args()
|
|
|
|
if not os.path.isfile(args.video):
|
|
print(f"Error: video not found: {args.video}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
success = generate_all_cover_previews(args.video, args.output_dir, args.title)
|
|
sys.exit(0 if success else 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|