]>
jfr.im git - yt-dlp.git/blob - youtube_dl/extractor/subtitles.py
1 from .common
import InfoExtractor
9 class SubtitlesInfoExtractor(InfoExtractor
):
11 def _have_to_download_any_subtitles(self
):
12 return any([self
._downloader
.params
.get('writesubtitles', False),
13 self
._downloader
.params
.get('writeautomaticsub')])
15 def _list_available_subtitles(self
, video_id
, webpage
):
16 """ outputs the available subtitles for the video """
17 sub_lang_list
= self
._get
_available
_subtitles
(video_id
, webpage
)
18 auto_captions_list
= self
._get
_available
_automatic
_caption
(video_id
, webpage
)
19 sub_lang
= ",".join(list(sub_lang_list
.keys()))
20 self
.to_screen(u
'%s: Available subtitles for video: %s' %
22 auto_lang
= ",".join(auto_captions_list
.keys())
23 self
.to_screen(u
'%s: Available automatic captions for video: %s' %
24 (video_id
, auto_lang
))
26 def extract_subtitles(self
, video_id
, webpage
):
28 returns {sub_lang: sub} ,{} if subtitles not found or None if the
29 subtitles aren't requested.
31 if not self
._have
_to
_download
_any
_subtitles
:
33 available_subs_list
= {}
34 if self
._downloader
.params
.get('writeautomaticsub', False):
35 available_subs_list
.update(self
._get
_available
_automatic
_caption
(video_id
, webpage
))
36 if self
._downloader
.params
.get('writesubtitles', False):
37 available_subs_list
.update(self
._get
_available
_subtitles
(video_id
, webpage
))
39 if not available_subs_list
: # error, it didn't get the available subtitles
41 if self
._downloader
.params
.get('allsubtitles', False):
42 sub_lang_list
= available_subs_list
44 if self
._downloader
.params
.get('subtitleslangs', False):
45 requested_langs
= self
._downloader
.params
.get('subtitleslangs')
46 elif 'en' in available_subs_list
:
47 requested_langs
= ['en']
49 requested_langs
= [list(available_subs_list
.keys())[0]]
52 for sub_lang
in requested_langs
:
53 if not sub_lang
in available_subs_list
:
54 self
._downloader
.report_warning(u
'no closed captions found in the specified language "%s"' % sub_lang
)
56 sub_lang_list
[sub_lang
] = available_subs_list
[sub_lang
]
59 for sub_lang
, url
in sub_lang_list
.items():
60 subtitle
= self
._request
_subtitle
_url
(sub_lang
, url
)
62 subtitles
[sub_lang
] = subtitle
65 def _download_subtitle_url(self
, sub_lang
, url
):
66 return self
._download
_webpage
(url
, None, note
=False)
68 def _request_subtitle_url(self
, sub_lang
, url
):
69 """ makes the http request for the subtitle """
71 return self
._download
_subtitle
_url
(sub_lang
, url
)
72 except ExtractorError
as err
:
73 self
._downloader
.report_warning(u
'unable to download video subtitles for %s: %s' % (sub_lang
, compat_str(err
)))
76 self
._downloader
.report_warning(u
'Did not fetch video subtitles')
79 def _get_available_subtitles(self
, video_id
, webpage
):
81 returns {sub_lang: url} or {} if not available
82 Must be redefined by the subclasses
85 # By default, allow implementations to simply pass in the result
86 assert isinstance(webpage
, dict), \
87 '_get_available_subtitles not implemented'
90 def _get_available_automatic_caption(self
, video_id
, webpage
):
92 returns {sub_lang: url} or {} if not available
93 Must be redefined by the subclasses that support automatic captions,
94 otherwise it will return {}
96 self
._downloader
.report_warning(u
'Automatic Captions not supported by this server')