diff options
Diffstat (limited to 'youtube_dl/extractor/iprima.py')
| -rw-r--r-- | youtube_dl/extractor/iprima.py | 17 | 
1 files changed, 9 insertions, 8 deletions
| diff --git a/youtube_dl/extractor/iprima.py b/youtube_dl/extractor/iprima.py index dde482998..bf5e44d88 100644 --- a/youtube_dl/extractor/iprima.py +++ b/youtube_dl/extractor/iprima.py @@ -10,7 +10,7 @@ from ..utils import compat_urllib_request  class IPrimaIE(InfoExtractor): -    _VALID_URL = r'https?://play\.iprima\.cz/(?P<videogroup>.+)/(?P<videoid>.+)' +    _VALID_URL = r'https?://play\.iprima\.cz/[^?#]+/(?P<id>[^?#]+)'      _TESTS = [{          'url': 'http://play.iprima.cz/particka/particka-92', @@ -29,7 +29,7 @@ class IPrimaIE(InfoExtractor):      def _real_extract(self, url):          mobj = re.match(self._VALID_URL, url) -        video_id = mobj.group('videoid') +        video_id = mobj.group('id')          webpage = self._download_webpage(url, video_id) @@ -44,18 +44,19 @@ class IPrimaIE(InfoExtractor):          base_url = ''.join(re.findall(r"embed\['stream'\] = '(.+?)'.+'(\?auth=)'.+'(.+?)';", playerpage)[1])          zoneGEO = self._html_search_regex(r'"zoneGEO":(.+?),', webpage, 'zoneGEO') -          if zoneGEO != '0': -            base_url = base_url.replace('token', 'token_'+zoneGEO) +            base_url = base_url.replace('token', 'token_' + zoneGEO)          formats = []          for format_id in ['lq', 'hq', 'hd']: -            filename = self._html_search_regex(r'"%s_id":(.+?),' % format_id, webpage, 'filename') +            filename = self._html_search_regex( +                r'"%s_id":(.+?),' % format_id, webpage, 'filename')              if filename == 'null':                  continue -            real_id = self._search_regex(r'Prima-[0-9]{10}-([0-9]+)_', filename, 'real video id') +            real_id = self._search_regex( +                r'Prima-[0-9]{10}-([0-9]+)_', filename, 'real video id')              if format_id == 'lq':                  quality = 0 @@ -63,13 +64,13 @@ class IPrimaIE(InfoExtractor):                  quality = 1              elif format_id == 'hd':                  quality = 2 -                filename = 'hq/'+filename +                filename = 'hq/' + filename              formats.append({                  'format_id': format_id,                  'url': base_url,                  'quality': quality, -                'play_path': 'mp4:'+filename.replace('"', '')[:-4], +                'play_path': 'mp4:' + filename.replace('"', '')[:-4],                  'rtmp_live': True,                  'ext': 'flv',              }) | 
