|
|
- from __future__ import unicode_literals
-
- import re
-
- from .common import InfoExtractor
- from ..compat import compat_str
- from ..utils import (
- try_get,
- unescapeHTML,
- url_or_none,
- urljoin,
- )
-
-
- class WWEBaseIE(InfoExtractor):
- _SUBTITLE_LANGS = {
- 'English': 'en',
- 'Deutsch': 'de',
- }
-
- def _extract_entry(self, data, url, video_id=None):
- video_id = compat_str(video_id or data['nid'])
- title = data['title']
-
- formats = self._extract_m3u8_formats(
- data['file'], video_id, 'mp4', entry_protocol='m3u8_native',
- m3u8_id='hls')
-
- description = data.get('description')
- thumbnail = urljoin(url, data.get('image'))
- series = data.get('show_name')
- episode = data.get('episode_name')
-
- subtitles = {}
- tracks = data.get('tracks')
- if isinstance(tracks, list):
- for track in tracks:
- if not isinstance(track, dict):
- continue
- if track.get('kind') != 'captions':
- continue
- track_file = url_or_none(track.get('file'))
- if not track_file:
- continue
- label = track.get('label')
- lang = self._SUBTITLE_LANGS.get(label, label) or 'en'
- subtitles.setdefault(lang, []).append({
- 'url': track_file,
- })
-
- return {
- 'id': video_id,
- 'title': title,
- 'description': description,
- 'thumbnail': thumbnail,
- 'series': series,
- 'episode': episode,
- 'formats': formats,
- 'subtitles': subtitles,
- }
-
-
- class WWEIE(WWEBaseIE):
- _VALID_URL = r'https?://(?:[^/]+\.)?wwe\.com/(?:[^/]+/)*videos/(?P<id>[^/?#&]+)'
- _TESTS = [{
- 'url': 'https://www.wwe.com/videos/daniel-bryan-vs-andrade-cien-almas-smackdown-live-sept-4-2018',
- 'md5': '92811c6a14bfc206f7a6a9c5d9140184',
- 'info_dict': {
- 'id': '40048199',
- 'ext': 'mp4',
- 'title': 'Daniel Bryan vs. Andrade "Cien" Almas: SmackDown LIVE, Sept. 4, 2018',
- 'description': 'md5:2d7424dbc6755c61a0e649d2a8677f67',
- 'thumbnail': r're:^https?://.*\.jpg$',
- }
- }, {
- 'url': 'https://de.wwe.com/videos/gran-metalik-vs-tony-nese-wwe-205-live-sept-4-2018',
- 'only_matching': True,
- }]
-
- def _real_extract(self, url):
- display_id = self._match_id(url)
- webpage = self._download_webpage(url, display_id)
-
- landing = self._parse_json(
- self._html_search_regex(
- r'(?s)Drupal\.settings\s*,\s*({.+?})\s*\)\s*;',
- webpage, 'drupal settings'),
- display_id)['WWEVideoLanding']
-
- data = landing['initialVideo']['playlist'][0]
- video_id = landing.get('initialVideoId')
-
- info = self._extract_entry(data, url, video_id)
- info['display_id'] = display_id
- return info
-
-
- class WWEPlaylistIE(WWEBaseIE):
- _VALID_URL = r'https?://(?:[^/]+\.)?wwe\.com/(?:[^/]+/)*(?P<id>[^/?#&]+)'
- _TESTS = [{
- 'url': 'https://www.wwe.com/shows/raw/2018-11-12',
- 'info_dict': {
- 'id': '2018-11-12',
- },
- 'playlist_mincount': 11,
- }, {
- 'url': 'http://www.wwe.com/article/walk-the-prank-wwe-edition',
- 'only_matching': True,
- }, {
- 'url': 'https://www.wwe.com/shows/wwenxt/article/matt-riddle-interview',
- 'only_matching': True,
- }]
-
- @classmethod
- def suitable(cls, url):
- return False if WWEIE.suitable(url) else super(WWEPlaylistIE, cls).suitable(url)
-
- def _real_extract(self, url):
- display_id = self._match_id(url)
- webpage = self._download_webpage(url, display_id)
-
- entries = []
- for mobj in re.finditer(
- r'data-video\s*=\s*(["\'])(?P<data>{.+?})\1', webpage):
- video = self._parse_json(
- mobj.group('data'), display_id, transform_source=unescapeHTML,
- fatal=False)
- if not video:
- continue
- data = try_get(video, lambda x: x['playlist'][0], dict)
- if not data:
- continue
- try:
- entry = self._extract_entry(data, url)
- except Exception:
- continue
- entry['extractor_key'] = WWEIE.ie_key()
- entries.append(entry)
-
- return self.playlist_result(entries, display_id)
|
