[iwara] Improve extraction

This commit is contained in:
Sergey M․ 2018-07-21 18:02:41 +07:00
parent af03000ad5
commit 4ecf300d13
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D

View File

@ -7,6 +7,7 @@ from ..utils import (
int_or_none, int_or_none,
mimetype2ext, mimetype2ext,
remove_end, remove_end,
url_or_none,
) )
@ -73,11 +74,14 @@ class IwaraIE(InfoExtractor):
formats = [] formats = []
for a_format in video_data: for a_format in video_data:
format_uri = url_or_none(a_format.get('uri'))
if not format_uri:
continue
format_id = a_format.get('resolution') format_id = a_format.get('resolution')
height = int_or_none(self._search_regex( height = int_or_none(self._search_regex(
r'(\d+)p', format_id, 'height', default=None)) r'(\d+)p', format_id, 'height', default=None))
formats.append({ formats.append({
'url': self._proto_relative_url(a_format['uri'], 'https:'), 'url': self._proto_relative_url(format_uri, 'https:'),
'format_id': format_id, 'format_id': format_id,
'ext': mimetype2ext(a_format.get('mime')) or 'mp4', 'ext': mimetype2ext(a_format.get('mime')) or 'mp4',
'height': height, 'height': height,