Ver código fonte

[imdb] Improve _VALID_URL (Closes #9788)

Sergey M․ 9 anos atrás
pai
commit
4c93ee8d14
1 arquivos alterados com 7 adições e 1 exclusões
  1. 7 1
      youtube_dl/extractor/imdb.py

+ 7 - 1
youtube_dl/extractor/imdb.py

@@ -12,7 +12,7 @@ from ..utils import (
 class ImdbIE(InfoExtractor):
 class ImdbIE(InfoExtractor):
     IE_NAME = 'imdb'
     IE_NAME = 'imdb'
     IE_DESC = 'Internet Movie Database trailers'
     IE_DESC = 'Internet Movie Database trailers'
-    _VALID_URL = r'https?://(?:www|m)\.imdb\.com/video/[^/]+/vi(?P<id>\d+)'
+    _VALID_URL = r'https?://(?:www|m)\.imdb\.com/(?:video/[^/]+/|title/tt\d+.*?#lb-)vi(?P<id>\d+)'
 
 
     _TESTS = [{
     _TESTS = [{
         'url': 'http://www.imdb.com/video/imdb/vi2524815897',
         'url': 'http://www.imdb.com/video/imdb/vi2524815897',
@@ -25,6 +25,12 @@ class ImdbIE(InfoExtractor):
     }, {
     }, {
         'url': 'http://www.imdb.com/video/_/vi2524815897',
         'url': 'http://www.imdb.com/video/_/vi2524815897',
         'only_matching': True,
         'only_matching': True,
+    }, {
+        'url': 'http://www.imdb.com/title/tt1667889/?ref_=ext_shr_eml_vi#lb-vi2524815897',
+        'only_matching': True,
+    }, {
+        'url': 'http://www.imdb.com/title/tt1667889/#lb-vi2524815897',
+        'only_matching': True,
     }]
     }]
 
 
     def _real_extract(self, url):
     def _real_extract(self, url):