from __future__ import unicode_literals import re from .common import InfoExtractor from ..utils import ( compat_urllib_parse, compat_urllib_request, ) class GDCVaultIE(InfoExtractor): _VALID_URL = r'https?://(?:www\.)?gdcvault\.com/play/(?P\d+)/(?P(\w|-)+)' _TESTS = [ { 'url': 'http://www.gdcvault.com/play/1019721/Doki-Doki-Universe-Sweet-Simple', 'md5': '7ce8388f544c88b7ac11c7ab1b593704', 'info_dict': { 'id': '1019721', 'ext': 'mp4', 'title': 'Doki-Doki Universe: Sweet, Simple and Genuine (GDC Next 10)' } }, { 'url': 'http://www.gdcvault.com/play/1015683/Embracing-the-Dark-Art-of', 'info_dict': { 'id': '1015683', 'ext': 'flv', 'title': 'Embracing the Dark Art of Mathematical Modeling in AI' }, 'params': { 'skip_download': True, # Requires rtmpdump } }, ] def _parse_mp4(self, xml_description): video_formats = [] mp4_video = xml_description.find('./metadata/mp4video') if mp4_video is None: return None mobj = re.match(r'(?Phttps?://.*?/).*', mp4_video.text) video_root = mobj.group('root') formats = xml_description.findall('./metadata/MBRVideos/MBRVideo') for format in formats: mobj = re.match(r'mp4\:(?P.*)', format.find('streamName').text) url = video_root + mobj.group('path') vbr = format.find('bitrate').text video_formats.append({ 'url': url, 'vbr': int(vbr), }) return video_formats def _parse_flv(self, xml_description): video_formats = [] akami_url = xml_description.find('./metadata/akamaiHost').text slide_video_path = xml_description.find('./metadata/slideVideo').text video_formats.append({ 'url': 'rtmp://' + akami_url + '/' + slide_video_path, 'format_note': 'slide deck video', 'quality': -2, 'preference': -2, 'format_id': 'slides', }) speaker_video_path = xml_description.find('./metadata/speakerVideo').text video_formats.append({ 'url': 'rtmp://' + akami_url + '/' + speaker_video_path, 'format_note': 'speaker video', 'quality': -1, 'preference': -1, 'format_id': 'speaker', }) return video_formats def _login(self, webpage_url, video_id): (username, password) = self._get_login_info() if username is None or password is None: self.report_warning('It looks like ' + webpage_url + ' requires a login. Try specifying a username and password and try again.') return None mobj = re.match(r'(?Phttps?://.*?/).*', webpage_url) login_url = mobj.group('root_url') + 'api/login.php' logout_url = mobj.group('root_url') + 'logout' login_form = { 'email': username, 'password': password, } request = compat_urllib_request.Request(login_url, compat_urllib_parse.urlencode(login_form)) request.add_header('Content-Type', 'application/x-www-form-urlencoded') self._download_webpage(request, video_id, 'Logging in') start_page = self._download_webpage(webpage_url, video_id, 'Getting authenticated video page') self._download_webpage(logout_url, video_id, 'Logging out') return start_page def _real_extract(self, url): mobj = re.match(self._VALID_URL, url) video_id = mobj.group('id') webpage_url = 'http://www.gdcvault.com/play/' + video_id start_page = self._download_webpage(webpage_url, video_id) xml_root = self._html_search_regex(r'', start_page, 'xml root') xml_name = self._html_search_regex(r'', start_page, 'xml filename') xml_decription_url = xml_root + 'xml/' + xml_name xml_description = self._download_xml(xml_decription_url, video_id) video_title = xml_description.find('./metadata/title').text video_formats = self._parse_mp4(xml_description) if video_formats is None: video_formats = self._parse_flv(xml_description) return { 'id': video_id, 'title': video_title, 'formats': video_formats, }