[9gag] Make display_id optional
authorSergey M․ <dstftw@gmail.com>
Tue, 22 Sep 2015 19:40:06 +0000 (01:40 +0600)
committerSergey M․ <dstftw@gmail.com>
Tue, 22 Sep 2015 19:40:06 +0000 (01:40 +0600)
youtube_dl/extractor/ninegag.py

index 0a2725c65dc99d9c00bd7235b563480eac8ea9f4..6103c7517a9b347b0a896858ee8623b85a369e09 100644 (file)
@@ -9,7 +9,7 @@ from ..utils import str_to_int
 
 class NineGagIE(InfoExtractor):
     IE_NAME = '9gag'
-    _VALID_URL = r'https?://(?:www\.)?9gag\.com/tv/p/(?P<id>[a-zA-Z0-9]+)/(?P<display_id>[^?#/]+)'
+    _VALID_URL = r'https?://(?:www\.)?9gag\.com/tv/p/(?P<id>[a-zA-Z0-9]+)(?:/(?P<display_id>[^?#/]+))?'
 
     _TESTS = [{
         "url": "http://9gag.com/tv/p/Kk2X5/people-are-awesome-2013-is-absolutely-awesome",
@@ -61,7 +61,7 @@ class NineGagIE(InfoExtractor):
     def _real_extract(self, url):
         mobj = re.match(self._VALID_URL, url)
         video_id = mobj.group('id')
-        display_id = mobj.group('display_id')
+        display_id = mobj.group('display_id') or video_id
 
         webpage = self._download_webpage(url, display_id)