mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-12-04 17:30:49 +01: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
132 lines
4.8 KiB
Python
132 lines
4.8 KiB
Python
from .common import InfoExtractor
|
|
from ..utils import (
|
|
clean_html,
|
|
extract_attributes,
|
|
float_or_none,
|
|
int_or_none,
|
|
try_get,
|
|
)
|
|
|
|
|
|
class EllenTubeBaseIE(InfoExtractor):
|
|
def _extract_data_config(self, webpage, video_id):
|
|
details = self._search_regex(
|
|
r'(<[^>]+\bdata-component=(["\'])[Dd]etails.+?></div>)', webpage,
|
|
'details')
|
|
return self._parse_json(
|
|
extract_attributes(details)['data-config'], video_id)
|
|
|
|
def _extract_video(self, data, video_id):
|
|
title = data['title']
|
|
|
|
formats = []
|
|
duration = None
|
|
for entry in data.get('media'):
|
|
if entry.get('id') == 'm3u8':
|
|
formats, subtitles = self._extract_m3u8_formats_and_subtitles(
|
|
entry['url'], video_id, 'mp4',
|
|
entry_protocol='m3u8_native', m3u8_id='hls')
|
|
duration = int_or_none(entry.get('duration'))
|
|
break
|
|
self._sort_formats(formats)
|
|
|
|
def get_insight(kind):
|
|
return int_or_none(try_get(
|
|
data, lambda x: x['insight']['%ss' % kind]))
|
|
|
|
return {
|
|
'extractor_key': EllenTubeIE.ie_key(),
|
|
'id': video_id,
|
|
'title': title,
|
|
'description': data.get('description'),
|
|
'duration': duration,
|
|
'thumbnail': data.get('thumbnail'),
|
|
'timestamp': float_or_none(data.get('publishTime'), scale=1000),
|
|
'view_count': get_insight('view'),
|
|
'like_count': get_insight('like'),
|
|
'formats': formats,
|
|
'subtitles': subtitles,
|
|
}
|
|
|
|
|
|
class EllenTubeIE(EllenTubeBaseIE):
|
|
_VALID_URL = r'''(?x)
|
|
(?:
|
|
ellentube:|
|
|
https://api-prod\.ellentube\.com/ellenapi/api/item/
|
|
)
|
|
(?P<id>[\da-f]{8}-[\da-f]{4}-[\da-f]{4}-[\da-f]{4}-[\da-f]{12})
|
|
'''
|
|
_TESTS = [{
|
|
'url': 'https://api-prod.ellentube.com/ellenapi/api/item/0822171c-3829-43bf-b99f-d77358ae75e3',
|
|
'md5': '2fabc277131bddafdd120e0fc0f974c9',
|
|
'info_dict': {
|
|
'id': '0822171c-3829-43bf-b99f-d77358ae75e3',
|
|
'ext': 'mp4',
|
|
'title': 'Ellen Meets Las Vegas Survivors Jesus Campos and Stephen Schuck',
|
|
'description': 'md5:76e3355e2242a78ad9e3858e5616923f',
|
|
'thumbnail': r're:^https?://.+?',
|
|
'duration': 514,
|
|
'timestamp': 1508505120,
|
|
'upload_date': '20171020',
|
|
'view_count': int,
|
|
'like_count': int,
|
|
}
|
|
}, {
|
|
'url': 'ellentube:734a3353-f697-4e79-9ca9-bfc3002dc1e0',
|
|
'only_matching': True,
|
|
}]
|
|
|
|
def _real_extract(self, url):
|
|
video_id = self._match_id(url)
|
|
data = self._download_json(
|
|
'https://api-prod.ellentube.com/ellenapi/api/item/%s' % video_id,
|
|
video_id)
|
|
return self._extract_video(data, video_id)
|
|
|
|
|
|
class EllenTubeVideoIE(EllenTubeBaseIE):
|
|
_VALID_URL = r'https?://(?:www\.)?ellentube\.com/video/(?P<id>.+?)\.html'
|
|
_TEST = {
|
|
'url': 'https://www.ellentube.com/video/ellen-meets-las-vegas-survivors-jesus-campos-and-stephen-schuck.html',
|
|
'only_matching': True,
|
|
}
|
|
|
|
def _real_extract(self, url):
|
|
display_id = self._match_id(url)
|
|
webpage = self._download_webpage(url, display_id)
|
|
video_id = self._extract_data_config(webpage, display_id)['id']
|
|
return self.url_result(
|
|
'ellentube:%s' % video_id, ie=EllenTubeIE.ie_key(),
|
|
video_id=video_id)
|
|
|
|
|
|
class EllenTubePlaylistIE(EllenTubeBaseIE):
|
|
_VALID_URL = r'https?://(?:www\.)?ellentube\.com/(?:episode|studios)/(?P<id>.+?)\.html'
|
|
_TESTS = [{
|
|
'url': 'https://www.ellentube.com/episode/dax-shepard-jordan-fisher-haim.html',
|
|
'info_dict': {
|
|
'id': 'dax-shepard-jordan-fisher-haim',
|
|
'title': "Dax Shepard, 'DWTS' Team Jordan Fisher & Lindsay Arnold, HAIM",
|
|
'description': 'md5:bfc982194dabb3f4e325e43aa6b2e21c',
|
|
},
|
|
'playlist_count': 6,
|
|
}, {
|
|
'url': 'https://www.ellentube.com/studios/macey-goes-rving0.html',
|
|
'only_matching': True,
|
|
}]
|
|
|
|
def _real_extract(self, url):
|
|
display_id = self._match_id(url)
|
|
webpage = self._download_webpage(url, display_id)
|
|
data = self._extract_data_config(webpage, display_id)['data']
|
|
feed = self._download_json(
|
|
'https://api-prod.ellentube.com/ellenapi/api/feed/?%s'
|
|
% data['filter'], display_id)
|
|
entries = [
|
|
self._extract_video(elem, elem['id'])
|
|
for elem in feed if elem.get('type') == 'VIDEO' and elem.get('id')]
|
|
return self.playlist_result(
|
|
entries, display_id, data.get('title'),
|
|
clean_html(data.get('description')))
|