pepepper/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl synced 2024-11-18 11:30:11 +09:00
Code Issues Projects Releases Wiki Activity

add bilibili subtitle download

Browse Source
This commit is contained in:
Du, Song 2021-06-04 18:16:27 -04:00
parent a803582717
commit cc6dec72b1
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
youtube_dl/extractor/bilibili.py
View File

@ -112,6 +112,14 @@ class BiliBiliIE(InfoExtractor):
_APP_KEY = 'iVGUTjsxvpLeuDCf'
_BILIBILI_KEY = 'aHRmhWMLkdeMuILqORnYZocwMBpMEOdt'
def convert_timestamp_format(self, cc_timestamp):
s, mm = int(cc_timestamp), 0
if "." in str(cc_timestamp):
s, mm = map(int, str(cc_timestamp).split("."))
m, s = divmod(s, 60)
h, m = divmod(m, 60)
return "%02d:%02d:%02d,%03d" % (h, m, s, mm)
def _report_error(self, result):
if 'message' in result:
raise ExtractorError('%s said: %s' % (self.IE_NAME, result['message']), expected=True)
@ -155,6 +163,21 @@ class BiliBiliIE(InfoExtractor):
self._report_error(js)
cid = js['result']['cid']
subtitles = {}
for sub in re.findall(r'"lan":\s*"([^"]+)"[^}]+"subtitle_url":\s*"([^"]+)"', webpage, re.DOTALL):
subtitles[sub[0]] = [{'url': sub[1]}]
j = self._download_json(sub[1].replace("\\u002F", "/"), video_id, note='Downloading raw subtitle: %s' % (sub[0]))
srt = []
for idx, line in enumerate(j["body"]):
from_timestamp = self.convert_timestamp_format(line["from"])
to_timestamp = self.convert_timestamp_format(line["to"])
srt_tmp = (
"""%d\n%s --> %s\n%s\n""" % (idx, from_timestamp, to_timestamp, line["content"])
)
srt.append(srt_tmp)
subtitles[sub[0]][0]["data"] = "\n".join(srt)
subtitles[sub[0]][0]["ext"] = "srt"
headers = {
'Accept': 'application/json',
'Referer': url
@ -204,6 +227,7 @@ class BiliBiliIE(InfoExtractor):
'id': '%s_part%s' % (video_id, idx),
'duration': float_or_none(durl.get('length'), 1000),
'formats': formats,
'subtitles': subtitles
})
break