mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-31 10:18:12 +00:00
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
# 公共方法
|
||
import json
|
||
import requests # 新增
|
||
import pysrt
|
||
from loguru import logger
|
||
from typing import List, Dict
|
||
|
||
|
||
def load_srt(file_path: str) -> List[Dict]:
|
||
"""加载并解析SRT文件（使用 pysrt 库，支持多种编码和格式）
|
||
|
||
Args:
|
||
file_path: SRT文件路径
|
||
|
||
Returns:
|
||
字幕内容列表，格式：
|
||
[
|
||
{
|
||
'number': int, # 字幕序号
|
||
'timestamp': str, # "00:00:01,000 --> 00:00:03,000"
|
||
'text': str, # 字幕文本
|
||
'start_time': str, # "00:00:01,000"
|
||
'end_time': str # "00:00:03,000"
|
||
},
|
||
...
|
||
]
|
||
|
||
Raises:
|
||
FileNotFoundError: 文件不存在
|
||
ValueError: 文件编码不支持或格式错误
|
||
"""
|
||
# 编码自动检测：依次尝试常见编码
|
||
encodings = ['utf-8', 'utf-8-sig', 'gbk', 'gb2312']
|
||
subs = None
|
||
detected_encoding = None
|
||
|
||
for encoding in encodings:
|
||
try:
|
||
subs = pysrt.open(file_path, encoding=encoding)
|
||
detected_encoding = encoding
|
||
logger.info(f"成功加载字幕文件 {file_path}，编码：{encoding}，共 {len(subs)} 条")
|
||
break
|
||
except UnicodeDecodeError:
|
||
continue
|
||
except Exception as e:
|
||
logger.warning(f"使用编码 {encoding} 加载失败: {e}")
|
||
continue
|
||
|
||
if subs is None:
|
||
# 所有编码都失败
|
||
raise ValueError(
|
||
f"无法读取字幕文件 {file_path}，"
|
||
f"请检查文件编码（支持 UTF-8、GBK、GB2312）"
|
||
)
|
||
|
||
# 检查是否为空
|
||
if not subs:
|
||
logger.warning(f"字幕文件 {file_path} 解析后无有效内容")
|
||
return []
|
||
|
||
# 转换为原格式（向后兼容）
|
||
subtitles = []
|
||
for sub in subs:
|
||
# 合并多行文本为单行（某些 SRT 文件会有换行）
|
||
text = sub.text.replace('\n', ' ').strip()
|
||
|
||
# 跳过空字幕
|
||
if not text:
|
||
continue
|
||
|
||
subtitles.append({
|
||
'number': sub.index,
|
||
'timestamp': f"{sub.start} --> {sub.end}",
|
||
'text': text,
|
||
'start_time': str(sub.start),
|
||
'end_time': str(sub.end)
|
||
})
|
||
|
||
logger.info(f"成功解析 {len(subtitles)} 条有效字幕")
|
||
return subtitles
|