-
Notifications
You must be signed in to change notification settings - Fork 0
/
BahaAss.py
229 lines (198 loc) · 8.72 KB
/
BahaAss.py
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
import re
import json
import time
import random
import requests
# https://cnfczn.com/article/ass%E5%AD%97%E5%B9%95%E6%A0%BC%E5%BC%8F%E8%AF%A6%E8%A7%A3
# http://www.perlfu.co.uk/projects/asa/ass-specs.doc
template_ass = '''[Script Info]
Title: {}
Original Script: BahaAss.py
ScriptType: v4.00+
Collisions: Normal
PlayResY: 1080
PlayResX: 1920
WrapStyle: 0
[v4+ Styles]
Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
Style: DefaultStyle,Microsoft YaHei UI,{},&H80FFFFFF,&H80FFFFFF,&H80000000,&H80000000,-1,0,0,0,100,100,0,0,1,1,0,5,0,0,0,0
[Events]
Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text'''
class BahaAss(object):
def __init__(self):
self._headers = {
'Host': None,
'Referer': None,
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36',
}
self._sn_dict = {}
self._digits_num = None
self._title = None
self._font_size = 40
self._pos_end_time = [[], []]
self._pos_time = 50 # 上下弹幕默认 5s
self._move_start_time = []
self._move_text_len = []
self._move_time = 100 # 滚动弹幕默认 10s
for i in range(1080 // self._font_size):
self._pos_end_time[0].append(-1)
self._pos_end_time[1].append(-1)
self._move_start_time.append(-self._move_time)
self._move_text_len.append(0)
def _reset_aux_vars(self):
for i in range(1080 // self._font_size):
self._pos_end_time[0][i] = -1
self._pos_end_time[1][i] = -1
self._move_start_time[i] = -self._move_time
self._move_text_len[i] = 0
def _get_all_sn(self, sn):
base_url = f'https://ani.gamer.com.tw/animeVideo.php?sn={sn}'
base_response = requests.get(base_url, headers=self._headers)
# with open('base.html', 'wb') as fp:
# fp.write(base_response.content)
base_response.encoding = 'utf8'
h1_list = re.findall(
r'<h1>(.+) \[(\S+)\]</h1>', base_response.text)
self._title = h1_list[0][0]
self._sn_dict[sn] = h1_list[0][1]
sn_list = re.findall(
r'<a href=\"\?sn=(\d+)\">(\S+)</a>', base_response.text)
_max = 0
for sn in sn_list:
self._sn_dict[sn[0]] = int(sn[1])
_max = max(_max, int(sn[1]))
self._digits_num = len(str(_max))
def _get_danmu(self, sn) -> dict:
danmu_url = 'https://ani.gamer.com.tw/ajax/danmuGet.php'
danmu_data = {'sn': sn}
danmu_response = requests.post(
danmu_url, danmu_data, headers=self._headers)
danmu_json = json.loads(danmu_response.content)
# with open('danmu.json', 'w', encoding='utf8') as fp:
# json.dump(danmu_json, fp, ensure_ascii=False, indent='\t')
return danmu_json
def _time_str(self, time: int) -> str:
second = time // 10
minute = second // 60
hour = minute // 60
hour_str = str(hour)
minute_str = str(minute - 60 * hour)
second_str = str(second - 60 * minute) + '.' + str(time)[-1] + '0'
return f'{hour_str}:{minute_str}:{second_str}'
def _get_pos_str(self, start_time: int, position: int) -> str:
# \pos(<x>, <y>)
pos_str = '\pos(960,{})'
index = position - 1
y = None
min_end_time = self._pos_end_time[index][0]
min_i = 0
for i in range(len(self._pos_end_time[index])):
if position == 2 and i < 0.1 * len(self._pos_end_time[index]):
# 底部弹幕预留 10%
min_end_time = self._pos_end_time[index][i + 1]
min_i = i + 1
continue
end_time = self._pos_end_time[index][i]
if start_time > end_time:
y = i * self._font_size + (self._font_size + 1) // 2
self._pos_end_time[index][i] = start_time + self._pos_time
break
if end_time < min_end_time:
min_end_time = end_time
min_i = i
if y is None:
y = min_i * self._font_size + (self._font_size + 1) // 2
self._pos_end_time[index][min_i] = start_time + self._pos_time
print(f"警告:{self._time_str(start_time)} 时上下方弹幕过密,会有重叠")
if position == 1:
pos_str = pos_str.format(y)
else:
pos_str = pos_str.format(1080 - y)
return pos_str
def _not_overlap(self, start_time1: int, text_len1: int, start_time2: int, text_len2: int) -> bool:
if start_time2 > start_time1 + self._move_time:
# 上一个弹幕已结束
return True
else:
# 弹幕长度
l1 = text_len1 * self._font_size
l2 = text_len2 * self._font_size
# 弹幕速度
v1 = (1920 + l1) / self._move_time
v2 = (1920 + l2) / self._move_time
# 间隔时间
delta_time = start_time2 - start_time1
# 弹幕间隔
delta_l = 4 * self._font_size
if v1 * delta_time <= l1 + delta_l:
# 新的出发时,保证上一个已完全出去,并由足够间隔
return False
if v2 * (self._move_time - delta_time) >= 1920 - delta_l:
# 上个到达时,保证新的也于其有足够间隔
return False
return True
def _get_move_str(self, start_time: int, text_len: int) -> str:
# \move(<x1>, <y1>, <x2>, <y2>)
move_str = '\move({},{},{},{})'
x = text_len * self._font_size // 2
y = None
min_i = 0
min_start_time = self._move_start_time[0]
for i in range(len(self._move_start_time)):
if self._not_overlap(self._move_start_time[i], self._move_text_len[i], start_time, text_len):
y = i * self._font_size + (self._font_size + 1) // 2
self._move_start_time[i] = start_time
self._move_text_len[i] = text_len
break
if self._move_start_time[i] < min_start_time:
min_start_time = self._move_start_time[i]
min_i = i
if y is None:
y = min_i * self._font_size + (self._font_size + 1) // 2
self._move_start_time[min_i] = start_time
self._move_text_len[min_i] = text_len
print(f"警告:{self._time_str(start_time)} 时滚动弹幕过密,可能会有重叠")
return move_str.format(1920+x, y, 0-x, y)
def _parse_danmu(self, danmu: dict, sn: str):
title = '{}_{}'.format(self._title, str(
self._sn_dict[sn]).zfill(self._digits_num))
with open(f'{title}.ass', 'w', encoding='utf8') as fp:
# text 弹幕内容
# color #RGB值
# size 0->小 1->正常 2->大
# position 0->滚动 1->上方 2->下方
# time 0.1s
y1 = 25
t1 = 0
fp.write(template_ass.format(title, self._font_size))
for one_dict in danmu:
start_time = one_dict['time']
position = one_dict['position']
color = one_dict['color'][1:]
# \c&H<bbggrr>&
color_str = '\c&H{}&'.format(color)
text = one_dict['text'].strip().replace('\n', '\\n')
if position == 0:
end_time = start_time + self._move_time
move_str = self._get_move_str(start_time, len(text))
style = f'{move_str}{color_str}'
else:
end_time = start_time + self._pos_time
end_time_str = self._time_str(end_time)
pos_str = self._get_pos_str(start_time, position)
style = f'{pos_str}{color_str}'
start_time_str = self._time_str(start_time)
end_time_str = self._time_str(end_time)
fp.write('\nDialogue: 0,{},{},DefaultStyle,,0,0,0,,{{{}}}{}'.format(
start_time_str, end_time_str, style, text))
def run(self, sn):
self._get_all_sn(sn)
for sn in self._sn_dict:
self._reset_aux_vars()
danmu = self._get_danmu(sn)
self._parse_danmu(danmu, sn)
print(f'{self._title} [{self._sn_dict[sn]}] 已转换成功! 为防止被 ban,休眠中……')
time.sleep(random.uniform(10, 20))
if __name__ == '__main__':
baha_ass = BahaAss()
baha_ass.run(input("输入 sn:"))