mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-27 02:42:30 +00:00
86e5f3ed2e
Using https://github.com/asottile/pyupgrade 1. `__future__` imports and `coding: utf-8` were removed 2. Files were rewritten with `pyupgrade --py36-plus --keep-percent-format` 3. f-strings were cherry-picked from `pyupgrade --py36-plus` Extractors are left untouched (except removing header) to avoid unnecessary merge conflicts
97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
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,
|
|
}
|