浏览代码

Changes suggested by @phihag

Pierre Rudloff 12 年之前
父节点
当前提交
5c1d63b737
共有 1 个文件被更改,包括 9 次插入9 次删除
  1. 9 9
      youtube_dl/extractor/websurg.py

+ 9 - 9
youtube_dl/extractor/websurg.py

@@ -36,21 +36,21 @@ class WeBSurgIE(InfoExtractor):
         request.add_header(
         request.add_header(
             'Content-Type', 'application/x-www-form-urlencoded;charset=utf-8')
             'Content-Type', 'application/x-www-form-urlencoded;charset=utf-8')
         compat_urllib_request.urlopen(request).info()
         compat_urllib_request.urlopen(request).info()
+        request = compat_urllib_request.Request(self._LOGIN_URL)
+        webpage = compat_urllib_request.urlopen(request).read()
         
         
-    def _real_extract(self, url):
-
-        request = compat_urllib_request.Request(url)
-        webpage = unicode(
-            compat_urllib_request.urlopen(request).read(), 'utf-8')
+        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)
         video_id = re.match(self._VALID_URL, url).group(1)
         
         
+        request = compat_urllib_request.Request(url)
+        webpage = self._download_webpage(url, video_id)
+        
         url_info = re.search(r'streamer="(.*?)" src="(.*?)"', webpage)
         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,
         return {'id': video_id,
                 'title': self._og_search_title(webpage),
                 'title': self._og_search_title(webpage),
                 'description': self._og_search_description(webpage),
                 'description': self._og_search_description(webpage),