mirror of
https://github.com/xhlove/GetDanMu.git
synced 2025-12-16 16:15:56 +08:00
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
#!/usr/bin/env python3.7
|
||
# coding=utf-8
|
||
'''
|
||
# 作者: weimo
|
||
# 创建日期: 2020-01-04 19:17:44
|
||
# 上次编辑时间 : 2020-01-11 17:25:09
|
||
# 一个人的命运啊,当然要靠自我奋斗,但是...
|
||
'''
|
||
import os
|
||
|
||
from basic.ass import get_ass_head, check_font
|
||
from methods.assbase import ASS
|
||
from methods.sameheight import SameHeight
|
||
from pfunc.cfunc import remove_same_danmu
|
||
|
||
def write_one_video_subtitles(file_path, comments, args):
|
||
# 对于合集则每次都都得检查一次 也可以放在上一级 放在这里 考虑后面可能特殊指定字体的情况
|
||
font_path, font_style_name = check_font(args.font)
|
||
ass_head = get_ass_head(font_style_name, args.font_size)
|
||
get_xy_obj = SameHeight("那就写这一句作为初始化测试吧！", font_path=font_path, font_size=int(args.font_size))
|
||
subtitle = ASS(file_path, get_xy_obj, font=font_style_name)
|
||
comments = remove_same_danmu(comments)
|
||
for comment in comments:
|
||
subtitle.create_new_line(comment)
|
||
write_lines_to_file(ass_head, subtitle.lines, file_path)
|
||
return comments
|
||
|
||
def write_lines_to_file(ass_head, lines, file_path):
|
||
with open(file_path, "a+", encoding="utf-8") as f:
|
||
f.write(ass_head + "\n")
|
||
for line in lines:
|
||
f.write(line + "\n")
|
||
|
||
def check_file(name, skip=False, fpath=os.getcwd()):
|
||
flag = True
|
||
file_path = os.path.join(fpath, name + ".ass")
|
||
if os.path.isfile(file_path):
|
||
if skip:
|
||
os.remove(file_path)
|
||
else:
|
||
isremove = input("{}已存在，是否覆盖？(y/n)：".format(file_path))
|
||
if isremove.strip() == "y":
|
||
os.remove(file_path)
|
||
else:
|
||
flag = False
|
||
return flag, file_path
|
||
with open(file_path, "wb") as f:
|
||
pass
|
||
return flag, file_path |