diff options
Diffstat (limited to 'youtube_dl/extractor/screenjunkies.py')
-rw-r--r-- | youtube_dl/extractor/screenjunkies.py | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/youtube_dl/extractor/screenjunkies.py b/youtube_dl/extractor/screenjunkies.py new file mode 100644 index 000000000..5da3f9bcd --- /dev/null +++ b/youtube_dl/extractor/screenjunkies.py @@ -0,0 +1,74 @@ +from __future__ import unicode_literals + +import re +import json + +from .common import ( + InfoExtractor, + ExtractorError, +) + +from ..utils import ( + int_or_none, +) + +class ScreenJunkiesIE(InfoExtractor): + _VALID_URL = r'http://www.screenjunkies.com/video/(.+-(?P<id>\d+)|.+)' + + _TESTS = [{ + 'url': 'http://www.screenjunkies.com/video/best-quentin-tarantino-movie-2841915', + 'info_dict': { + 'id': '2841915', + 'ext': 'mp4', + 'title': 'Best Quentin Tarantino Movie', + }, + }, { + 'url': 'http://www.screenjunkies.com/video/honest-trailers-the-dark-knight', + 'info_dict': { + 'id': '2348808', + 'ext': 'mp4', + 'title': "Honest Trailers - 'The Dark Knight'", + }, + }, { + 'url': 'http://www.screenjunkies.com/video/knocking-dead-ep-1-the-show-so-far-3003285', + 'only_matching': True, + }] + + def _real_extract(self, url): + video_id = self._match_id(url) + + if not video_id: # Older urls didn't have the id in them, but we can grab it manually + webpage = self._download_webpage(url, url) + video_id = self._html_search_regex(r'src="/embed/(\d+)"', webpage, 'video id') + + webpage = self._download_webpage('http://www.screenjunkies.com/embed/%s' %video_id, video_id) + video_vars_str = self._html_search_regex(r'embedVars = (\{.+\})\s*</script>', webpage, 'video variables', flags=re.S) + video_vars = self._parse_json(video_vars_str, video_id) + + # TODO: Figure out required cookies + if video_vars['subscriptionLevel'] > 0: + raise ExtractorError('This video requires ScreenJunkiesPlus', expected=True) + + formats = [] + for f in video_vars['media']: + if not f['mediaPurpose'] == 'play': + continue + + formats.append({ + 'url': f['uri'], + 'width': int_or_none(f.get('width')), + 'height': int_or_none(f.get('height')), + 'tbr': int_or_none(f.get('bitRate')), + 'format': 'mp4', + }) + self._sort_formats(formats) + + return { + 'id': video_id, + 'title': video_vars['contentName'], + 'formats': formats, + 'duration': int_or_none(video_vars.get('videoLengthInSeconds')), + 'thumbnail': video_vars.get('thumbUri'), + 'tags': video_vars['tags'].split(',') if 'tags' in video_vars else [], + } + |