97 lines
3.3 KiB
Python
97 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
从 数据.jsonl 中读取成功记录，将 MP4 和 SRT 文件下载到本地 ./成功/ 目录。
|
||
用法：python download_srt.py [数据.jsonl路径] [本地输出目录]
|
||
默认：python download_srt.py 数据.jsonl ./成功
|
||
"""
|
||
import json
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
from config.config_utils import config
|
||
|
||
JSONL_PATH = sys.argv[1] if len(sys.argv) > 1 else "数据.jsonl"
|
||
OUTPUT_DIR = sys.argv[2] if len(sys.argv) > 2 else "./成功"
|
||
|
||
|
||
def download(coscli_path, cos_config_path, cos_src, local_path):
|
||
"""下载单个文件，返回是否成功"""
|
||
if os.path.exists(local_path):
|
||
return "skip"
|
||
cmd = [coscli_path, "-c", cos_config_path, "cp", cos_src, local_path]
|
||
result = subprocess.run(cmd, capture_output=True, text=True, check=False)
|
||
return "ok" if result.returncode == 0 else result.stderr.strip()
|
||
|
||
|
||
def main():
|
||
if not os.path.exists(JSONL_PATH):
|
||
print(f"找不到文件: {JSONL_PATH}")
|
||
return
|
||
|
||
cfg = config.get_splitframe_config()
|
||
coscli_path = cfg.get('coscli_path', './coscli-linux-amd64')
|
||
cos_config_path = cfg.get('cosconfig_path', '.cos.yaml')
|
||
|
||
# 读取所有成功记录
|
||
records = []
|
||
with open(JSONL_PATH, 'r', encoding='utf-8') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if not line:
|
||
continue
|
||
try:
|
||
rec = json.loads(line)
|
||
if rec.get('status') == 'success' and rec.get('srt_cos_path'):
|
||
records.append(rec)
|
||
except json.JSONDecodeError:
|
||
pass
|
||
records = records[:50]
|
||
total = len(records)
|
||
print(f"共找到 {total} 条成功记录，开始下载...")
|
||
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
||
|
||
success = skip = failed = 0
|
||
|
||
for i, rec in enumerate(records, 1):
|
||
video_cos = rec.get('video_cos_path').replace(".mp4", ".mp3")
|
||
srt_cos = rec.get('srt_cos_path')
|
||
|
||
video_name = os.path.splitext(os.path.basename(video_cos))[0] if video_cos else None
|
||
|
||
print(f"[{i}/{total}] {video_name}")
|
||
|
||
# 下载 MP4
|
||
if video_cos:
|
||
local_mp4 = os.path.join(OUTPUT_DIR, os.path.basename(video_cos))
|
||
ret = download(coscli_path, cos_config_path, video_cos, local_mp4)
|
||
if ret == "ok":
|
||
success += 1
|
||
print(f" ✓ 视频: {os.path.basename(video_cos)}")
|
||
elif ret == "skip":
|
||
skip += 1
|
||
print(f" - 视频已存在，跳过")
|
||
else:
|
||
failed += 1
|
||
print(f" ✗ 视频下载失败: {ret}")
|
||
|
||
# 下载 SRT
|
||
if srt_cos:
|
||
local_srt = os.path.join(OUTPUT_DIR, os.path.basename(srt_cos))
|
||
ret = download(coscli_path, cos_config_path, srt_cos, local_srt)
|
||
if ret == "ok":
|
||
success += 1
|
||
print(f" ✓ 字幕: {os.path.basename(srt_cos)}")
|
||
elif ret == "skip":
|
||
skip += 1
|
||
print(f" - 字幕已存在，跳过")
|
||
else:
|
||
failed += 1
|
||
print(f" ✗ 字幕下载失败: {ret}")
|
||
|
||
print(f"\n完成 — 下载: {success} 失败: {failed} 跳过(已存在): {skip}")
|
||
print(f"文件保存在: {os.path.abspath(OUTPUT_DIR)}")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|