[wat] extract dash formats

This commit is contained in:
Remita Amine 2016-09-06 20:44:45 +01:00
parent 155bc674c4
commit 948cd5b72d
1 changed files with 41 additions and 29 deletions

View File

@ -86,38 +86,50 @@ class WatIE(InfoExtractor):
def extract_url(path_template, url_type): def extract_url(path_template, url_type):
req_url = 'http://www.wat.tv/get/%s' % (path_template % video_id) req_url = 'http://www.wat.tv/get/%s' % (path_template % video_id)
head = self._request_webpage(HEADRequest(req_url), video_id, 'Extracting %s url' % url_type) head = self._request_webpage(HEADRequest(req_url), video_id, 'Extracting %s url' % url_type, fatal=False)
red_url = head.geturl() if head:
if req_url == red_url: red_url = head.geturl()
raise ExtractorError( if req_url != red_url:
'%s said: Sorry, this video is not available from your country.' % self.IE_NAME, return red_url
expected=True) return None
return red_url
def remove_bitrate_limit(manifest_url):
return re.sub(r'(?:max|min)_bitrate=\d+&?', '', manifest_url)
formats = [] formats = []
try: try:
http_url = extract_url('android5/%s.mp4', 'http') manifest_urls = self._download_json(
m3u8_url = extract_url('ipad/%s.m3u8', 'm3u8') 'http://www.wat.tv/get/webhtml/' + video_id, video_id)
m3u8_formats = self._extract_m3u8_formats( m3u8_url = manifest_urls.get('hls')
m3u8_url, video_id, 'mp4', 'm3u8_native', m3u8_id='hls') if m3u8_url:
formats.extend(m3u8_formats) m3u8_url = remove_bitrate_limit(m3u8_url)
formats.extend(self._extract_f4m_formats( m3u8_formats = self._extract_m3u8_formats(
m3u8_url.replace('ios.', 'web.').replace('.m3u8', '.f4m'), m3u8_url, video_id, 'mp4', 'm3u8_native', m3u8_id='hls', fatal=False)
video_id, f4m_id='hds', fatal=False)) if m3u8_formats:
for m3u8_format in m3u8_formats: formats.extend(m3u8_formats)
vbr, abr = m3u8_format.get('vbr'), m3u8_format.get('abr') formats.extend(self._extract_f4m_formats(
if not vbr or not abr: m3u8_url.replace('ios', 'web').replace('.m3u8', '.f4m'),
continue video_id, f4m_id='hds', fatal=False))
format_id = m3u8_format['format_id'].replace('hls', 'http') http_url = extract_url('android5/%s.mp4', 'http')
fmt_url = re.sub(r'%s-\d+00-\d+' % video_id, '%s-%d00-%d' % (video_id, round(vbr / 100), round(abr)), http_url) if http_url:
if self._is_valid_url(fmt_url, video_id, format_id): for m3u8_format in m3u8_formats:
f = m3u8_format.copy() vbr, abr = m3u8_format.get('vbr'), m3u8_format.get('abr')
f.update({ if not vbr or not abr:
'url': fmt_url, continue
'format_id': format_id, format_id = m3u8_format['format_id'].replace('hls', 'http')
'protocol': 'http', fmt_url = re.sub(r'%s-\d+00-\d+' % video_id, '%s-%d00-%d' % (video_id, round(vbr / 100), round(abr)), http_url)
}) if self._is_valid_url(fmt_url, video_id, format_id):
formats.append(f) f = m3u8_format.copy()
f.update({
'url': fmt_url,
'format_id': format_id,
'protocol': 'http',
})
formats.append(f)
mpd_url = manifest_urls.get('mpd')
if mpd_url:
formats.extend(self._extract_mpd_formats(remove_bitrate_limit(
mpd_url), video_id, mpd_id='dash', fatal=False))
self._sort_formats(formats) self._sort_formats(formats)
except ExtractorError: except ExtractorError:
abr = 64 abr = 64