diff options
| author | Remita Amine <remitamine@gmail.com> | 2017-02-21 17:47:14 +0100 | 
|---|---|---|
| committer | Remita Amine <remitamine@gmail.com> | 2017-02-21 17:52:50 +0100 | 
| commit | 3444844b04ae482edc5a353d9125b45ba47cd8d8 (patch) | |
| tree | 1c8d35df8ad6e318ca98b826178f07a3379ca75c | |
| parent | 8c6c88c7dae595d5cb7d5926eb00fbaf40103f8c (diff) | |
[limelight] extract PlaylistService errors
| -rw-r--r-- | youtube_dl/extractor/limelight.py | 17 | 
1 files changed, 14 insertions, 3 deletions
| diff --git a/youtube_dl/extractor/limelight.py b/youtube_dl/extractor/limelight.py index a3712665b..422be2528 100644 --- a/youtube_dl/extractor/limelight.py +++ b/youtube_dl/extractor/limelight.py @@ -4,11 +4,13 @@ from __future__ import unicode_literals  import re  from .common import InfoExtractor +from ..compat import compat_HTTPError  from ..utils import (      determine_ext,      float_or_none,      int_or_none,      unsmuggle_url, +    ExtractorError,  ) @@ -20,9 +22,17 @@ class LimelightBaseIE(InfoExtractor):          headers = {}          if referer:              headers['Referer'] = referer -        return self._download_json( -            self._PLAYLIST_SERVICE_URL % (self._PLAYLIST_SERVICE_PATH, item_id, method), -            item_id, 'Downloading PlaylistService %s JSON' % method, fatal=fatal, headers=headers) +        try: +            return self._download_json( +                self._PLAYLIST_SERVICE_URL % (self._PLAYLIST_SERVICE_PATH, item_id, method), +                item_id, 'Downloading PlaylistService %s JSON' % method, fatal=fatal, headers=headers) +        except ExtractorError as e: +            if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403: +                error = self._parse_json(e.cause.read().decode(), item_id)['detail']['contentAccessPermission'] +                if error == 'CountryDisabled': +                    self.raise_geo_restricted() +                raise ExtractorError(error, expected=True) +            raise      def _call_api(self, organization_id, item_id, method):          return self._download_json( @@ -213,6 +223,7 @@ class LimelightMediaIE(LimelightBaseIE):      def _real_extract(self, url):          url, smuggled_data = unsmuggle_url(url, {})          video_id = self._match_id(url) +        self._initialize_geo_bypass(smuggled_data.get('geo_countries'))          pc, mobile, metadata = self._extract(              video_id, 'getPlaylistByMediaId', | 
