56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
import datetime
|
|
import os
|
|
import argparse
|
|
from typing import Union
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
|
|
class NoDanmakuException(Exception):
|
|
...
|
|
|
|
|
|
class DanmakuFormatErrorException(Exception):
|
|
...
|
|
|
|
|
|
def check_file_exist(file):
|
|
if not os.path.isfile(file):
|
|
raise FileNotFoundError("文件不存在：%s" % file)
|
|
|
|
|
|
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)
|
|
|
|
|
|
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))
|