61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
import datetime
|
|
import os
|
|
import argparse
|
|
import subprocess
|
|
from hashlib import md5
|
|
from typing import Union
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
from config import DANMU_ASS_EXEC, VIDEO_RESOLUTION, DANMAKU_SPEED, DEFAULT_FONT_NAME
|
|
from exception.danmaku import NoDanmakuException, DanmakuFormatErrorException
|
|
from util.file import check_file_exist
|
|
|
|
|
|
def get_file_start(file: Union[os.PathLike[str], str]) -> float:
|
|
with open(file, "r", encoding="utf-8") as f:
|
|
soup = BeautifulSoup("".join(f.readlines()), "lxml")
|
|
danmaku_item = soup.find("d")
|
|
if danmaku_item is None:
|
|
# 没有弹幕？
|
|
raise NoDanmakuException()
|
|
danmaku_info = danmaku_item["p"]
|
|
split_info = danmaku_info.split(",")
|
|
if len(split_info) < 5:
|
|
raise DanmakuFormatErrorException()
|
|
bias_sec = float(split_info[0])
|
|
bias_ts_ms = int(split_info[4])
|
|
return bias_ts_ms / 1000 - bias_sec
|
|
|
|
|
|
def diff_danmaku_files(base_file: Union[os.PathLike[str], str], file: Union[os.PathLike[str], str]) -> float:
|
|
return get_file_start(file) - get_file_start(base_file)
|
|
|
|
|
|
def danmaku_to_subtitle(file: Union[os.PathLike[str], str], time_shift: float):
|
|
new_subtitle_name = md5(file.encode("utf-8")).hexdigest() + ".ass"
|
|
(_w, _h) = VIDEO_RESOLUTION.split("x")
|
|
process = subprocess.Popen((
|
|
DANMU_ASS_EXEC, "--force", "-a", "0.9", "-d", str(DANMAKU_SPEED), "--font", str(DEFAULT_FONT_NAME),
|
|
"--font-size", "40", "--lane-size", "40", "--width", _w, "--height", _h, "-o", new_subtitle_name,
|
|
"-p", "1", "--time-offset", str(time_shift), "--width-ratio", "1", file
|
|
))
|
|
process.wait()
|
|
return new_subtitle_name
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("base", help="以此为标准")
|
|
parser.add_argument("file", nargs="+", help="需要对齐的文件")
|
|
args = parser.parse_args()
|
|
check_file_exist(args.base)
|
|
base_start_ts = get_file_start(args.base)
|
|
base_start = datetime.datetime.fromtimestamp(base_start_ts)
|
|
print("[+] 基准文件[{:s}]，开始时间：{:.3f}，时间：{}".format(args.base, base_start_ts, base_start))
|
|
for _file in args.file:
|
|
check_file_exist(_file)
|
|
file_start_ts = get_file_start(_file)
|
|
diff_sec = file_start_ts - base_start_ts
|
|
print("[+] 待调整文件[{:s}]，开始时间：{:.3f}，偏差：{:.3f}".format(_file, file_start_ts, diff_sec))
|