[metacafe] move tests
[youtube-dl] / youtube_dl / extractor / metacafe.py
1 import re
2 import socket
3
4 from .common import InfoExtractor
5 from ..utils import (
6     compat_http_client,
7     compat_parse_qs,
8     compat_urllib_error,
9     compat_urllib_parse,
10     compat_urllib_request,
11     compat_str,
12
13     ExtractorError,
14 )
15
16 class MetacafeIE(InfoExtractor):
17     """Information Extractor for metacafe.com."""
18
19     _VALID_URL = r'(?:http://)?(?:www\.)?metacafe\.com/watch/([^/]+)/([^/]+)/.*'
20     _DISCLAIMER = 'http://www.metacafe.com/family_filter/'
21     _FILTER_POST = 'http://www.metacafe.com/f/index.php?inputType=filter&controllerGroup=user'
22     IE_NAME = u'metacafe'
23     _TEST = {
24         u"name": u"Metacafe",
25         u"add_ie": ["Youtube"],
26         u"url":  u"http://metacafe.com/watch/yt-_aUehQsCQtM/the_electric_company_short_i_pbs_kids_go/",
27         u"file":  u"_aUehQsCQtM.flv",
28         u"info_dict": {
29             u"upload_date": u"20090102",
30             u"title": u"The Electric Company | \"Short I\" | PBS KIDS GO!",
31             u"description": u"md5:2439a8ef6d5a70e380c22f5ad323e5a8",
32             u"uploader": u"PBS",
33             u"uploader_id": u"PBS"
34         }
35     }
36
37
38     def report_disclaimer(self):
39         """Report disclaimer retrieval."""
40         self.to_screen(u'Retrieving disclaimer')
41
42     def _real_initialize(self):
43         # Retrieve disclaimer
44         request = compat_urllib_request.Request(self._DISCLAIMER)
45         try:
46             self.report_disclaimer()
47             compat_urllib_request.urlopen(request).read()
48         except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
49             raise ExtractorError(u'Unable to retrieve disclaimer: %s' % compat_str(err))
50
51         # Confirm age
52         disclaimer_form = {
53             'filters': '0',
54             'submit': "Continue - I'm over 18",
55             }
56         request = compat_urllib_request.Request(self._FILTER_POST, compat_urllib_parse.urlencode(disclaimer_form))
57         try:
58             self.report_age_confirmation()
59             compat_urllib_request.urlopen(request).read()
60         except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
61             raise ExtractorError(u'Unable to confirm age: %s' % compat_str(err))
62
63     def _real_extract(self, url):
64         # Extract id and simplified title from URL
65         mobj = re.match(self._VALID_URL, url)
66         if mobj is None:
67             raise ExtractorError(u'Invalid URL: %s' % url)
68
69         video_id = mobj.group(1)
70
71         # Check if video comes from YouTube
72         mobj2 = re.match(r'^yt-(.*)$', video_id)
73         if mobj2 is not None:
74             return [self.url_result('http://www.youtube.com/watch?v=%s' % mobj2.group(1), 'Youtube')]
75
76         # Retrieve video webpage to extract further information
77         webpage = self._download_webpage('http://www.metacafe.com/watch/%s/' % video_id, video_id)
78
79         # Extract URL, uploader and title from webpage
80         self.report_extraction(video_id)
81         mobj = re.search(r'(?m)&mediaURL=([^&]+)', webpage)
82         if mobj is not None:
83             mediaURL = compat_urllib_parse.unquote(mobj.group(1))
84             video_extension = mediaURL[-3:]
85
86             # Extract gdaKey if available
87             mobj = re.search(r'(?m)&gdaKey=(.*?)&', webpage)
88             if mobj is None:
89                 video_url = mediaURL
90             else:
91                 gdaKey = mobj.group(1)
92                 video_url = '%s?__gda__=%s' % (mediaURL, gdaKey)
93         else:
94             mobj = re.search(r' name="flashvars" value="(.*?)"', webpage)
95             if mobj is None:
96                 raise ExtractorError(u'Unable to extract media URL')
97             vardict = compat_parse_qs(mobj.group(1))
98             if 'mediaData' not in vardict:
99                 raise ExtractorError(u'Unable to extract media URL')
100             mobj = re.search(r'"mediaURL":"(?P<mediaURL>http.*?)",(.*?)"key":"(?P<key>.*?)"', vardict['mediaData'][0])
101             if mobj is None:
102                 raise ExtractorError(u'Unable to extract media URL')
103             mediaURL = mobj.group('mediaURL').replace('\\/', '/')
104             video_extension = mediaURL[-3:]
105             video_url = '%s?__gda__=%s' % (mediaURL, mobj.group('key'))
106
107         mobj = re.search(r'(?im)<title>(.*) - Video</title>', webpage)
108         if mobj is None:
109             raise ExtractorError(u'Unable to extract title')
110         video_title = mobj.group(1).decode('utf-8')
111
112         mobj = re.search(r'submitter=(.*?);', webpage)
113         if mobj is None:
114             raise ExtractorError(u'Unable to extract uploader nickname')
115         video_uploader = mobj.group(1)
116
117         return [{
118             'id':       video_id.decode('utf-8'),
119             'url':      video_url.decode('utf-8'),
120             'uploader': video_uploader.decode('utf-8'),
121             'upload_date':  None,
122             'title':    video_title,
123             'ext':      video_extension.decode('utf-8'),
124         }]