diff options
| -rw-r--r-- | youtube_dl/extractor/macgamestore.py | 41 |
1 files changed, 22 insertions, 19 deletions
diff --git a/youtube_dl/extractor/macgamestore.py b/youtube_dl/extractor/macgamestore.py index 4541f6d66..b818cf50c 100644 --- a/youtube_dl/extractor/macgamestore.py +++ b/youtube_dl/extractor/macgamestore.py @@ -1,3 +1,5 @@ +from __future__ import unicode_literals + import re from .common import InfoExtractor @@ -5,36 +7,37 @@ from ..utils import ExtractorError class MacGameStoreIE(InfoExtractor): - IE_NAME = u'macgamestore' - IE_DESC = u'MacGameStore trailers' + IE_NAME = 'macgamestore' + IE_DESC = 'MacGameStore trailers' _VALID_URL = r'https?://www\.macgamestore\.com/mediaviewer\.php\?trailer=(?P<id>\d+)' _TEST = { - u'url': u'http://www.macgamestore.com/mediaviewer.php?trailer=2450', - u'file': u'2450.m4v', - u'md5': u'8649b8ea684b6666b4c5be736ecddc61', - u'info_dict': { - u'title': u'Crow', + 'url': 'http://www.macgamestore.com/mediaviewer.php?trailer=2450', + 'file': '2450.m4v', + 'md5': '8649b8ea684b6666b4c5be736ecddc61', + 'info_dict': { + 'title': 'Crow', } } def _real_extract(self, url): mobj = re.match(self._VALID_URL, url) video_id = mobj.group('id') - - webpage = self._download_webpage(url, video_id, u'Downloading trailer page') - + + webpage = self._download_webpage(url, video_id, 'Downloading trailer page') + if re.search(r'>Missing Media<', webpage) is not None: - raise ExtractorError(u'Trailer %s does not exist' % video_id, expected=True) - - mobj = re.search(r'<title>MacGameStore: (?P<title>.*?) Trailer</title>', webpage) - video_title = mobj.group('title') - - mobj = re.search(r'(?s)<div\s+id="video-player".*?href="(?P<video>[^"]+)"\s*>', webpage) - video_url = mobj.group('video') - + raise ExtractorError('Trailer %s does not exist' % video_id, expected=True) + + video_title = self._html_search_regex( + r'<title>MacGameStore: (.*?) Trailer</title>', webpage, 'title') + + video_url = self._html_search_regex( + r'(?s)<div\s+id="video-player".*?href="([^"]+)"\s*>', + webpage, 'video URL') + return { 'id': video_id, 'url': video_url, 'title': video_title - }
\ No newline at end of file + } |