# coding: utf-8 from __future__ import unicode_literals import re from .common import InfoExtractor from ..compat import compat_urlparse from ..utils import ( extract_attributes, get_element_by_class, urlencode_postdata, ) class NJPWWorldIE(InfoExtractor): _VALID_URL = r'https?://njpwworld\.com/p/(?P<id>[a-z0-9_]+)' IE_DESC = '新日本プロレスワールド' _NETRC_MACHINE = 'njpwworld' _TEST = { 'url': 'http://njpwworld.com/p/s_series_00155_1_9/', 'info_dict': { 'id': 's_series_00155_1_9', 'ext': 'mp4', 'title': '第9試合 ランディ・サベージ vs リック・スタイナー', 'tags': list, }, 'params': { 'skip_download': True, # AES-encrypted m3u8 }, 'skip': 'Requires login', } _LOGIN_URL = 'https://front.njpwworld.com/auth/login' def _real_initialize(self): self._login() def _login(self): username, password = self._get_login_info() # No authentication to be performed if not username: return True # Setup session (will set necessary cookies) self._request_webpage( 'https://njpwworld.com/', None, note='Setting up session') webpage, urlh = self._download_webpage_handle( self._LOGIN_URL, None, note='Logging in', errnote='Unable to login', data=urlencode_postdata({'login_id': username, 'pw': password}), headers={'Referer': 'https://front.njpwworld.com/auth'}) # /auth/login will return 302 for successful logins if urlh.geturl() == self._LOGIN_URL: self.report_warning('unable to login') return False return True def _real_extract(self, url): video_id = self._match_id(url) webpage = self._download_webpage(url, video_id) formats = [] for mobj in re.finditer(r'<a[^>]+\bhref=(["\'])/player.+?[^>]*>', webpage): player = extract_attributes(mobj.group(0)) player_path = player.get('href') if not player_path: continue kind = self._search_regex( r'(low|high)$', player.get('class') or '', 'kind', default='low') player_url = compat_urlparse.urljoin(url, player_path) player_page = self._download_webpage( player_url, video_id, note='Downloading player page') entries = self._parse_html5_media_entries( player_url, player_page, video_id, m3u8_id='hls-%s' % kind, m3u8_entry_protocol='m3u8_native') kind_formats = entries[0]['formats'] for f in kind_formats: f['quality'] = 2 if kind == 'high' else 1 formats.extend(kind_formats) self._sort_formats(formats) post_content = get_element_by_class('post-content', webpage) tags = re.findall( r'<li[^>]+class="tag-[^"]+"><a[^>]*>([^<]+)</a></li>', post_content ) if post_content else None return { 'id': video_id, 'title': self._og_search_title(webpage), 'formats': formats, 'tags': tags, }