Spaces:
Sleeping
Sleeping
File size: 1,926 Bytes
db4f540 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
"""SRT 字幕文件解析器"""
import re
from dataclasses import dataclass
@dataclass
class SubtitleEntry:
"""字幕条目"""
index: int
start_time: float # 秒
end_time: float # 秒
text: str
def parse_timestamp(timestamp: str) -> float:
"""将 HH:MM:SS,mmm 格式转换为秒数
Args:
timestamp: 时间戳字符串，如 "00:01:23,456"
Returns:
float: 秒数
"""
# 匹配格式：HH:MM:SS,mmm
match = re.match(r"(\d{2}):(\d{2}):(\d{2}),(\d{3})", timestamp)
if not match:
raise ValueError(f"无效的时间戳格式: {timestamp}")
hours, minutes, seconds, milliseconds = map(int, match.groups())
total_seconds = hours * 3600 + minutes * 60 + seconds + milliseconds / 1000
return total_seconds
def parse_srt(file_path: str) -> list[SubtitleEntry]:
"""解析 SRT 字幕文件
Args:
file_path: SRT 文件路径
Returns:
List[SubtitleEntry]: 字幕条目列表
"""
with open(file_path, encoding="utf-8") as f:
content = f.read()
entries = []
# 按空行分割字幕块
blocks = content.strip().split("\n\n")
for block in blocks:
lines = block.strip().split("\n")
if len(lines) < 3:
continue
# 第一行是序号
try:
index = int(lines[0])
except ValueError:
continue
# 第二行是时间戳
timestamp_line = lines[1]
match = re.match(r"(.+?)\s*-->\s*(.+)", timestamp_line)
if not match:
continue
start_str, end_str = match.groups()
start_time = parse_timestamp(start_str.strip())
end_time = parse_timestamp(end_str.strip())
# 剩余行是文本内容
text = " ".join(lines[2:])
entries.append(SubtitleEntry(index=index, start_time=start_time, end_time=end_time, text=text))
return entries
|