Pārlūkot izejas kodu

[slutload] Add support for mobile URLs

Windom 7 gadi atpakaļ
vecāks
revīzija
4f5cf31977
1 mainītis faili ar 11 papildinājumiem un 3 dzēšanām
  1. 11 3
      youtube_dl/extractor/slutload.py

+ 11 - 3
youtube_dl/extractor/slutload.py

@@ -1,11 +1,13 @@
 from __future__ import unicode_literals
 from __future__ import unicode_literals
 
 
+import re
+
 from .common import InfoExtractor
 from .common import InfoExtractor
 
 
 
 
 class SlutloadIE(InfoExtractor):
 class SlutloadIE(InfoExtractor):
     _VALID_URL = r'^https?://(?:\w+\.)?slutload\.com/video/[^/]+/(?P<id>[^/]+)/?$'
     _VALID_URL = r'^https?://(?:\w+\.)?slutload\.com/video/[^/]+/(?P<id>[^/]+)/?$'
-    _TEST = {
+    _TESTS = [{
         'url': 'http://www.slutload.com/video/virginie-baisee-en-cam/TD73btpBqSxc/',
         'url': 'http://www.slutload.com/video/virginie-baisee-en-cam/TD73btpBqSxc/',
         'md5': '868309628ba00fd488cf516a113fd717',
         'md5': '868309628ba00fd488cf516a113fd717',
         'info_dict': {
         'info_dict': {
@@ -15,11 +17,17 @@ class SlutloadIE(InfoExtractor):
             'age_limit': 18,
             'age_limit': 18,
             'thumbnail': r're:https?://.*?\.jpg'
             'thumbnail': r're:https?://.*?\.jpg'
         }
         }
-    }
+    }, {
+        # mobile site
+        'url': 'http://mobile.slutload.com/video/masturbation-solo/fviFLmc6kzJ/',
+        'only_matching': True,
+    }]
 
 
     def _real_extract(self, url):
     def _real_extract(self, url):
         video_id = self._match_id(url)
         video_id = self._match_id(url)
-        webpage = self._download_webpage(url, video_id)
+
+        desktop_url = re.sub(r'^(https?://)mobile\.', r'\1', url)
+        webpage = self._download_webpage(desktop_url, video_id)
 
 
         video_title = self._html_search_regex(r'<h1><strong>([^<]+)</strong>',
         video_title = self._html_search_regex(r'<h1><strong>([^<]+)</strong>',
                                               webpage, 'title').strip()
                                               webpage, 'title').strip()