Use self._download_webpage everywhere

[youtube-dl] / youtube_dl / extractor / websurg.py

diff --git a/youtube_dl/extractor/websurg.py b/youtube_dl/extractor/websurg.py
index 953bc983131e15c317782cf85e8115cd75464605..7d335d444e5231df69e1282490dd3d2212411f93 100644 (file)
--- a/youtube_dl/extractor/websurg.py
+++ b/youtube_dl/extractor/websurg.py
@@ -23,7 +23,7 @@ class WeBSurgIE(InfoExtractor):
     
     _LOGIN_URL = 'http://www.websurg.com/inc/login/login_div.ajax.php?login=1'
 
-    def _real_extract(self, url):
+    def _real_initialize(self):
 
         login_form = {
             'username': self._downloader.params['username'],
@@ -35,33 +35,24 @@ class WeBSurgIE(InfoExtractor):
             self._LOGIN_URL, compat_urllib_parse.urlencode(login_form))
         request.add_header(
             'Content-Type', 'application/x-www-form-urlencoded;charset=utf-8')
-        login_results = compat_urllib_request.urlopen(request).info()
+        compat_urllib_request.urlopen(request).info()
+        webpage = self._download_webpage(self._LOGIN_URL, '', 'Logging in')
         
-        sessid = re.match(r'PHPSESSID=(.*);',
-            login_results['Set-Cookie']).group(1)
-        request = compat_urllib_request.Request(
-            url, compat_urllib_parse.urlencode(login_form),
-            {'Cookie': 'PHPSESSID=' + sessid + ';'})
-        webpage = compat_urllib_request.urlopen(request).read()
+        if webpage != 'OK':
+            self._downloader.report_error(
+                u'Unable to log in: bad username/password')
         
+    def _real_extract(self, url):
         video_id = re.match(self._VALID_URL, url).group(1)
         
+        webpage = self._download_webpage(url, video_id)
+        
         url_info = re.search(r'streamer="(.*?)" src="(.*?)"', webpage)
         
-        if url_info is None:
-            self._downloader.report_warning(
-                u'Unable to log in: bad username/password')
-            return
-            
         return {'id': video_id,
-                'title' : re.search(
-                    r'property="og:title" content="(.*?)" />'
-                    , webpage).group(1),
-                'description': re.search(
-                    r'name="description" content="(.*?)" />', webpage).group(1),
+                'title': self._og_search_title(webpage),
+                'description': self._og_search_description(webpage),
                 'ext' : 'mp4',
                 'url' : url_info.group(1) + '/' + url_info.group(2),
-                'thumbnail': re.search(
-                    r'property="og:image" content="(.*?)" />', webpage
-                ).group(1)
+                'thumbnail': self._og_search_thumbnail(webpage)
                 }