mirror of https://github.com/blackjack4494/yt-dlc
100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
|
# coding: utf-8
|
||
|
from __future__ import unicode_literals
|
||
|
|
||
|
import re
|
||
|
|
||
|
from .common import InfoExtractor
|
||
|
from ..utils import (
|
||
|
int_or_none,
|
||
|
str_or_none,
|
||
|
url_or_none,
|
||
|
)
|
||
|
|
||
|
|
||
|
class ViqeoIE(InfoExtractor):
|
||
|
_VALID_URL = r'''(?x)
|
||
|
(?:
|
||
|
viqeo:|
|
||
|
https?://cdn\.viqeo\.tv/embed/*\?.*?\bvid=|
|
||
|
https?://api\.viqeo\.tv/v\d+/data/startup?.*?\bvideo(?:%5B%5D|\[\])=
|
||
|
)
|
||
|
(?P<id>[\da-f]+)
|
||
|
'''
|
||
|
_TESTS = [{
|
||
|
'url': 'https://cdn.viqeo.tv/embed/?vid=cde96f09d25f39bee837',
|
||
|
'md5': 'a169dd1a6426b350dca4296226f21e76',
|
||
|
'info_dict': {
|
||
|
'id': 'cde96f09d25f39bee837',
|
||
|
'ext': 'mp4',
|
||
|
'title': 'cde96f09d25f39bee837',
|
||
|
'thumbnail': r're:^https?://.*\.jpg$',
|
||
|
'duration': 76,
|
||
|
},
|
||
|
}, {
|
||
|
'url': 'viqeo:cde96f09d25f39bee837',
|
||
|
'only_matching': True,
|
||
|
}, {
|
||
|
'url': 'https://api.viqeo.tv/v1/data/startup?video%5B%5D=71bbec412ade45c3216c&profile=112',
|
||
|
'only_matching': True,
|
||
|
}]
|
||
|
|
||
|
@staticmethod
|
||
|
def _extract_urls(webpage):
|
||
|
return [
|
||
|
mobj.group('url')
|
||
|
for mobj in re.finditer(
|
||
|
r'<iframe[^>]+\bsrc=(["\'])(?P<url>(?:https?:)?//cdn\.viqeo\.tv/embed/*\?.*?\bvid=[\da-f]+.*?)\1',
|
||
|
webpage)]
|
||
|
|
||
|
def _real_extract(self, url):
|
||
|
video_id = self._match_id(url)
|
||
|
|
||
|
webpage = self._download_webpage(
|
||
|
'https://cdn.viqeo.tv/embed/?vid=%s' % video_id, video_id)
|
||
|
|
||
|
data = self._parse_json(
|
||
|
self._search_regex(
|
||
|
r'SLOT_DATA\s*=\s*({.+?})\s*;', webpage, 'slot data'),
|
||
|
video_id)
|
||
|
|
||
|
formats = []
|
||
|
thumbnails = []
|
||
|
for media_file in data['mediaFiles']:
|
||
|
if not isinstance(media_file, dict):
|
||
|
continue
|
||
|
media_url = url_or_none(media_file.get('url'))
|
||
|
if not media_url or not media_url.startswith(('http', '//')):
|
||
|
continue
|
||
|
media_type = str_or_none(media_file.get('type'))
|
||
|
if not media_type:
|
||
|
continue
|
||
|
media_kind = media_type.split('/')[0].lower()
|
||
|
f = {
|
||
|
'url': media_url,
|
||
|
'width': int_or_none(media_file.get('width')),
|
||
|
'height': int_or_none(media_file.get('height')),
|
||
|
}
|
||
|
format_id = str_or_none(media_file.get('quality'))
|
||
|
if media_kind == 'image':
|
||
|
f['id'] = format_id
|
||
|
thumbnails.append(f)
|
||
|
elif media_kind in ('video', 'audio'):
|
||
|
is_audio = media_kind == 'audio'
|
||
|
f.update({
|
||
|
'format_id': 'audio' if is_audio else format_id,
|
||
|
'fps': int_or_none(media_file.get('fps')),
|
||
|
'vcodec': 'none' if is_audio else None,
|
||
|
})
|
||
|
formats.append(f)
|
||
|
self._sort_formats(formats)
|
||
|
|
||
|
duration = int_or_none(data.get('duration'))
|
||
|
|
||
|
return {
|
||
|
'id': video_id,
|
||
|
'title': video_id,
|
||
|
'duration': duration,
|
||
|
'thumbnails': thumbnails,
|
||
|
'formats': formats,
|
||
|
}
|