[nrk] Improve geo restriction detection

2016-11-13 22:29:36 +07:00 · 2016-11-13 22:29:36 +07:00 · 50913b8241
parent 7e08e2cab0
commit 50913b8241
1 changed files with 3 additions and 2 deletions
--- a/youtube_dl/extractor/nrk.py
+++ b/youtube_dl/extractor/nrk.py
@ -93,8 +93,9 @@ class NRKBaseIE(InfoExtractor):
                }]

        if not entries:
-            message_type = data.get('messageType')
-            if message_type == 'ProgramIsGeoBlocked' and not self._faked_ip:
+            message_type = data.get('messageType', '')
+            # Can be ProgramIsGeoBlocked or ChannelIsGeoBlocked*
+            if 'IsGeoBlocked' in message_type and not self._faked_ip:
                self.report_warning(
                    'Video is geo restricted, trying to fake IP')
                self._fake_ip()