mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-30 03:33:02 +00:00
[downloader/hls] immediately delegate downloading to ffmpeg in case live stream
This commit is contained in:
parent
51ef4919df
commit
fb4fc44928
7 changed files with 27 additions and 25 deletions
|
@ -30,6 +30,15 @@ class HlsFD(FragmentFD):
|
||||||
|
|
||||||
FD_NAME = 'hlsnative'
|
FD_NAME = 'hlsnative'
|
||||||
|
|
||||||
|
def _delegate_to_ffmpeg(self, filename, info_dict):
|
||||||
|
self.report_warning(
|
||||||
|
'hlsnative has detected features it does not support, '
|
||||||
|
'extraction will be delegated to ffmpeg')
|
||||||
|
fd = FFmpegFD(self.ydl, self.params)
|
||||||
|
for ph in self._progress_hooks:
|
||||||
|
fd.add_progress_hook(ph)
|
||||||
|
return fd.real_download(filename, info_dict)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def can_download(manifest, info_dict):
|
def can_download(manifest, info_dict):
|
||||||
UNSUPPORTED_FEATURES = (
|
UNSUPPORTED_FEATURES = (
|
||||||
|
@ -53,10 +62,12 @@ def can_download(manifest, info_dict):
|
||||||
)
|
)
|
||||||
check_results = [not re.search(feature, manifest) for feature in UNSUPPORTED_FEATURES]
|
check_results = [not re.search(feature, manifest) for feature in UNSUPPORTED_FEATURES]
|
||||||
check_results.append(can_decrypt_frag or '#EXT-X-KEY:METHOD=AES-128' not in manifest)
|
check_results.append(can_decrypt_frag or '#EXT-X-KEY:METHOD=AES-128' not in manifest)
|
||||||
check_results.append(not info_dict.get('is_live'))
|
|
||||||
return all(check_results)
|
return all(check_results)
|
||||||
|
|
||||||
def real_download(self, filename, info_dict):
|
def real_download(self, filename, info_dict):
|
||||||
|
if info_dict.get('is_live'):
|
||||||
|
return self._delegate_to_ffmpeg(filename, info_dict)
|
||||||
|
|
||||||
man_url = info_dict['url']
|
man_url = info_dict['url']
|
||||||
self.to_screen('[%s] Downloading m3u8 manifest' % self.FD_NAME)
|
self.to_screen('[%s] Downloading m3u8 manifest' % self.FD_NAME)
|
||||||
|
|
||||||
|
@ -68,13 +79,7 @@ def real_download(self, filename, info_dict):
|
||||||
if info_dict.get('extra_param_to_segment_url'):
|
if info_dict.get('extra_param_to_segment_url'):
|
||||||
self.report_error('pycrypto not found. Please install it.')
|
self.report_error('pycrypto not found. Please install it.')
|
||||||
return False
|
return False
|
||||||
self.report_warning(
|
return self._delegate_to_ffmpeg(filename, info_dict)
|
||||||
'hlsnative has detected features it does not support, '
|
|
||||||
'extraction will be delegated to ffmpeg')
|
|
||||||
fd = FFmpegFD(self.ydl, self.params)
|
|
||||||
for ph in self._progress_hooks:
|
|
||||||
fd.add_progress_hook(ph)
|
|
||||||
return fd.real_download(filename, info_dict)
|
|
||||||
|
|
||||||
total_frags = 0
|
total_frags = 0
|
||||||
for line in s.splitlines():
|
for line in s.splitlines():
|
||||||
|
|
|
@ -93,8 +93,7 @@ def _real_extract(self, url):
|
||||||
exts = (mimetype2ext(f.get('Type')), determine_ext(f_url, None))
|
exts = (mimetype2ext(f.get('Type')), determine_ext(f_url, None))
|
||||||
if kind == 'm3u8' or 'm3u8' in exts:
|
if kind == 'm3u8' or 'm3u8' in exts:
|
||||||
formats.extend(self._extract_m3u8_formats(
|
formats.extend(self._extract_m3u8_formats(
|
||||||
f_url, video_id, 'mp4',
|
f_url, video_id, 'mp4', 'm3u8_native',
|
||||||
entry_protocol='m3u8' if is_live else 'm3u8_native',
|
|
||||||
m3u8_id=kind, fatal=False, live=is_live))
|
m3u8_id=kind, fatal=False, live=is_live))
|
||||||
elif kind == 'flash' or 'f4m' in exts:
|
elif kind == 'flash' or 'f4m' in exts:
|
||||||
formats.extend(self._extract_f4m_formats(
|
formats.extend(self._extract_f4m_formats(
|
||||||
|
|
|
@ -160,8 +160,7 @@ def _real_extract(self, url):
|
||||||
for format_id, stream_url in item.get('streamUrls', {}).items():
|
for format_id, stream_url in item.get('streamUrls', {}).items():
|
||||||
if 'playerType=flash' in stream_url:
|
if 'playerType=flash' in stream_url:
|
||||||
stream_formats = self._extract_m3u8_formats(
|
stream_formats = self._extract_m3u8_formats(
|
||||||
stream_url, playlist_id, 'mp4',
|
stream_url, playlist_id, 'mp4', 'm3u8_native',
|
||||||
entry_protocol='m3u8' if is_live else 'm3u8_native',
|
|
||||||
m3u8_id='hls-%s' % format_id, fatal=False)
|
m3u8_id='hls-%s' % format_id, fatal=False)
|
||||||
else:
|
else:
|
||||||
stream_formats = self._extract_mpd_formats(
|
stream_formats = self._extract_mpd_formats(
|
||||||
|
|
|
@ -54,7 +54,7 @@ def _add_ns(path):
|
||||||
'id': video_id,
|
'id': video_id,
|
||||||
'title': title,
|
'title': title,
|
||||||
'formats': self._extract_m3u8_formats(
|
'formats': self._extract_m3u8_formats(
|
||||||
m3u8_url, video_id, 'mp4', 'm3u8' if is_live else 'm3u8_native'),
|
m3u8_url, video_id, 'mp4', 'm3u8_native'),
|
||||||
'description': xpath_text(video_data, _add_ns('Description')),
|
'description': xpath_text(video_data, _add_ns('Description')),
|
||||||
'duration': parse_duration(xpath_text(video_data, _add_ns('Duration'))),
|
'duration': parse_duration(xpath_text(video_data, _add_ns('Duration'))),
|
||||||
'uploader': xpath_text(video_data, _add_ns('Createur')),
|
'uploader': xpath_text(video_data, _add_ns('Createur')),
|
||||||
|
|
|
@ -56,9 +56,8 @@ def _real_extract(self, url):
|
||||||
is_live = info.get('liveStreamUrl') is not None
|
is_live = info.get('liveStreamUrl') is not None
|
||||||
|
|
||||||
formats = self._extract_m3u8_formats(
|
formats = self._extract_m3u8_formats(
|
||||||
stream_url, video_id, ext='mp4',
|
stream_url, video_id, 'mp4',
|
||||||
entry_protocol='m3u8' if is_live else 'm3u8_native',
|
'm3u8_native', m3u8_id='hls')
|
||||||
m3u8_id='hls')
|
|
||||||
|
|
||||||
if is_live:
|
if is_live:
|
||||||
title = self._live_title(title)
|
title = self._live_title(title)
|
||||||
|
|
|
@ -119,7 +119,8 @@ def _extract_video_info(self, video_data):
|
||||||
m3u8_url = video_data.get('m3u8_url')
|
m3u8_url = video_data.get('m3u8_url')
|
||||||
if m3u8_url:
|
if m3u8_url:
|
||||||
formats.extend(self._extract_m3u8_formats(
|
formats.extend(self._extract_m3u8_formats(
|
||||||
m3u8_url, video_id, 'mp4', 'm3u8_native', m3u8_id='hls', fatal=False))
|
m3u8_url, video_id, 'mp4', 'm3u8_native',
|
||||||
|
m3u8_id='hls', fatal=False))
|
||||||
|
|
||||||
f4m_url = video_data.get('f4m_url')
|
f4m_url = video_data.get('f4m_url')
|
||||||
if f4m_url:
|
if f4m_url:
|
||||||
|
@ -158,11 +159,11 @@ def _extract_stream_info(self, stream_info):
|
||||||
if smil_url:
|
if smil_url:
|
||||||
formats.extend(self._extract_smil_formats(smil_url, broadcast_id))
|
formats.extend(self._extract_smil_formats(smil_url, broadcast_id))
|
||||||
|
|
||||||
entry_protocol = 'm3u8' if is_live else 'm3u8_native'
|
|
||||||
m3u8_url = stream_info.get('m3u8_url')
|
m3u8_url = stream_info.get('m3u8_url')
|
||||||
if m3u8_url:
|
if m3u8_url:
|
||||||
formats.extend(self._extract_m3u8_formats(
|
formats.extend(self._extract_m3u8_formats(
|
||||||
m3u8_url, broadcast_id, 'mp4', entry_protocol, m3u8_id='hls', fatal=False))
|
m3u8_url, broadcast_id, 'mp4', 'm3u8_native',
|
||||||
|
m3u8_id='hls', fatal=False))
|
||||||
|
|
||||||
rtsp_url = stream_info.get('rtsp_url')
|
rtsp_url = stream_info.get('rtsp_url')
|
||||||
if rtsp_url:
|
if rtsp_url:
|
||||||
|
@ -276,7 +277,7 @@ def _extract_video_info(self, user, video_id):
|
||||||
'view_count': view_count,
|
'view_count': view_count,
|
||||||
}
|
}
|
||||||
|
|
||||||
def _extract_video_formats(self, video_data, video_id, entry_protocol):
|
def _extract_video_formats(self, video_data, video_id):
|
||||||
formats = []
|
formats = []
|
||||||
|
|
||||||
progressive_url = video_data.get('progressiveUrl')
|
progressive_url = video_data.get('progressiveUrl')
|
||||||
|
@ -289,7 +290,8 @@ def _extract_video_formats(self, video_data, video_id, entry_protocol):
|
||||||
m3u8_url = video_data.get('httpUrl')
|
m3u8_url = video_data.get('httpUrl')
|
||||||
if m3u8_url:
|
if m3u8_url:
|
||||||
formats.extend(self._extract_m3u8_formats(
|
formats.extend(self._extract_m3u8_formats(
|
||||||
m3u8_url, video_id, 'mp4', entry_protocol, m3u8_id='hls', fatal=False))
|
m3u8_url, video_id, 'mp4', 'm3u8_native',
|
||||||
|
m3u8_id='hls', fatal=False))
|
||||||
|
|
||||||
rtsp_url = video_data.get('rtspUrl')
|
rtsp_url = video_data.get('rtspUrl')
|
||||||
if rtsp_url:
|
if rtsp_url:
|
||||||
|
@ -340,11 +342,10 @@ def _real_extract(self, url):
|
||||||
}
|
}
|
||||||
video_data = self._download_json(stream_url, content_id)
|
video_data = self._download_json(stream_url, content_id)
|
||||||
is_live = video_data.get('isLive')
|
is_live = video_data.get('isLive')
|
||||||
entry_protocol = 'm3u8' if is_live else 'm3u8_native'
|
|
||||||
info.update({
|
info.update({
|
||||||
'id': content_id,
|
'id': content_id,
|
||||||
'title': self._live_title(info['title']) if is_live else info['title'],
|
'title': self._live_title(info['title']) if is_live else info['title'],
|
||||||
'formats': self._extract_video_formats(video_data, content_id, entry_protocol),
|
'formats': self._extract_video_formats(video_data, content_id),
|
||||||
'is_live': is_live,
|
'is_live': is_live,
|
||||||
})
|
})
|
||||||
return info
|
return info
|
||||||
|
|
|
@ -432,8 +432,7 @@ def _real_extract(self, url):
|
||||||
})
|
})
|
||||||
elif format_id == 'hls':
|
elif format_id == 'hls':
|
||||||
formats.extend(self._extract_m3u8_formats(
|
formats.extend(self._extract_m3u8_formats(
|
||||||
format_url, video_id, 'mp4',
|
format_url, video_id, 'mp4', 'm3u8_native',
|
||||||
entry_protocol='m3u8' if is_live else 'm3u8_native',
|
|
||||||
m3u8_id=format_id, fatal=False, live=is_live))
|
m3u8_id=format_id, fatal=False, live=is_live))
|
||||||
elif format_id == 'rtmp':
|
elif format_id == 'rtmp':
|
||||||
formats.append({
|
formats.append({
|
||||||
|
|
Loading…
Reference in a new issue