Browse Source

Refactor code to use url_result

Elan Ruusamäe 6 years ago
parent
commit
d226c560a6

+ 1 - 1
youtube_dl/extractor/audiomack.py

@@ -62,7 +62,7 @@ class AudiomackIE(InfoExtractor):
         # Audiomack wraps a lot of soundcloud tracks in their branded wrapper
         # Audiomack wraps a lot of soundcloud tracks in their branded wrapper
         # if so, pass the work off to the soundcloud extractor
         # if so, pass the work off to the soundcloud extractor
         if SoundcloudIE.suitable(api_response['url']):
         if SoundcloudIE.suitable(api_response['url']):
-            return {'_type': 'url', 'url': api_response['url'], 'ie_key': 'Soundcloud'}
+            return self.url_result(api_response['url'], SoundcloudIE.ie_key())
 
 
         return {
         return {
             'id': compat_str(api_response.get('id', album_url_tag)),
             'id': compat_str(api_response.get('id', album_url_tag)),

+ 2 - 10
youtube_dl/extractor/cnn.py

@@ -119,11 +119,7 @@ class CNNBlogsIE(InfoExtractor):
     def _real_extract(self, url):
     def _real_extract(self, url):
         webpage = self._download_webpage(url, url_basename(url))
         webpage = self._download_webpage(url, url_basename(url))
         cnn_url = self._html_search_regex(r'data-url="(.+?)"', webpage, 'cnn url')
         cnn_url = self._html_search_regex(r'data-url="(.+?)"', webpage, 'cnn url')
-        return {
-            '_type': 'url',
-            'url': cnn_url,
-            'ie_key': CNNIE.ie_key(),
-        }
+        return self.url_result(cnn_url, CNNIE.ie_key())
 
 
 
 
 class CNNArticleIE(InfoExtractor):
 class CNNArticleIE(InfoExtractor):
@@ -145,8 +141,4 @@ class CNNArticleIE(InfoExtractor):
     def _real_extract(self, url):
     def _real_extract(self, url):
         webpage = self._download_webpage(url, url_basename(url))
         webpage = self._download_webpage(url, url_basename(url))
         cnn_url = self._html_search_regex(r"video:\s*'([^']+)'", webpage, 'cnn url')
         cnn_url = self._html_search_regex(r"video:\s*'([^']+)'", webpage, 'cnn url')
-        return {
-            '_type': 'url',
-            'url': 'http://cnn.com/video/?/video/' + cnn_url,
-            'ie_key': CNNIE.ie_key(),
-        }
+        return self.url_result('http://cnn.com/video/?/video/' + cnn_url, CNNIE.ie_key())

+ 2 - 5
youtube_dl/extractor/freespeech.py

@@ -1,6 +1,7 @@
 from __future__ import unicode_literals
 from __future__ import unicode_literals
 
 
 from .common import InfoExtractor
 from .common import InfoExtractor
+from .youtube import YoutubeIE
 
 
 
 
 class FreespeechIE(InfoExtractor):
 class FreespeechIE(InfoExtractor):
@@ -27,8 +28,4 @@ class FreespeechIE(InfoExtractor):
             r'data-video-url="([^"]+)"',
             r'data-video-url="([^"]+)"',
             webpage, 'youtube url')
             webpage, 'youtube url')
 
 
-        return {
-            '_type': 'url',
-            'url': youtube_url,
-            'ie_key': 'Youtube',
-        }
+        return self.url_result(youtube_url, YoutubeIE.ie_key())

+ 1 - 4
youtube_dl/extractor/generic.py

@@ -2197,10 +2197,7 @@ class GenericIE(InfoExtractor):
 
 
     def _real_extract(self, url):
     def _real_extract(self, url):
         if url.startswith('//'):
         if url.startswith('//'):
-            return {
-                '_type': 'url',
-                'url': self.http_scheme() + url,
-            }
+            return self.url_result(self.http_scheme() + url)
 
 
         parsed_url = compat_urlparse.urlparse(url)
         parsed_url = compat_urlparse.urlparse(url)
         if not parsed_url.scheme:
         if not parsed_url.scheme:

+ 1 - 4
youtube_dl/extractor/livestream.py

@@ -363,7 +363,4 @@ class LivestreamShortenerIE(InfoExtractor):
         id = mobj.group('id')
         id = mobj.group('id')
         webpage = self._download_webpage(url, id)
         webpage = self._download_webpage(url, id)
 
 
-        return {
-            '_type': 'url',
-            'url': self._og_search_url(webpage),
-        }
+        return self.url_result(self._og_search_url(webpage))

+ 2 - 5
youtube_dl/extractor/savefrom.py

@@ -30,8 +30,5 @@ class SaveFromIE(InfoExtractor):
     def _real_extract(self, url):
     def _real_extract(self, url):
         mobj = re.match(self._VALID_URL, url)
         mobj = re.match(self._VALID_URL, url)
         video_id = os.path.splitext(url.split('/')[-1])[0]
         video_id = os.path.splitext(url.split('/')[-1])[0]
-        return {
-            '_type': 'url',
-            'id': video_id,
-            'url': mobj.group('url'),
-        }
+
+        return self.url_result(mobj.group('url'), video_id=video_id)

+ 2 - 4
youtube_dl/extractor/ted.py

@@ -203,10 +203,8 @@ class TEDIE(InfoExtractor):
                 ext_url = None
                 ext_url = None
                 if service.lower() == 'youtube':
                 if service.lower() == 'youtube':
                     ext_url = external.get('code')
                     ext_url = external.get('code')
-                return {
-                    '_type': 'url',
-                    'url': ext_url or external['uri'],
-                }
+
+                return self.url_result(ext_url or external['uri'])
 
 
         resources_ = player_talk.get('resources') or talk_info.get('resources')
         resources_ = player_talk.get('resources') or talk_info.get('resources')
 
 

+ 1 - 5
youtube_dl/extractor/testurl.py

@@ -61,8 +61,4 @@ class TestURLIE(InfoExtractor):
 
 
         self.to_screen('Test URL: %s' % tc['url'])
         self.to_screen('Test URL: %s' % tc['url'])
 
 
-        return {
-            '_type': 'url',
-            'url': tc['url'],
-            'id': video_id,
-        }
+        return self.url_result(tc['url'], video_id=video_id)

+ 1 - 5
youtube_dl/extractor/wimp.py

@@ -40,11 +40,7 @@ class WimpIE(InfoExtractor):
              r'data-id=["\']([0-9A-Za-z_-]{11})'),
              r'data-id=["\']([0-9A-Za-z_-]{11})'),
             webpage, 'video URL', default=None)
             webpage, 'video URL', default=None)
         if youtube_id:
         if youtube_id:
-            return {
-                '_type': 'url',
-                'url': youtube_id,
-                'ie_key': YoutubeIE.ie_key(),
-            }
+            return self.url_result(youtube_id, YoutubeIE.ie_key())
 
 
         info_dict = self._extract_jwplayer_data(
         info_dict = self._extract_jwplayer_data(
             webpage, video_id, require_title=False)
             webpage, video_id, require_title=False)