Explorar o código

[nrk] Improve geo restriction detection

Sergey M․ %!s(int64=9) %!d(string=hai) anos
pai
achega
50913b8241
Modificáronse 1 ficheiros con 3 adicións e 2 borrados
  1. 3 2
      youtube_dl/extractor/nrk.py

+ 3 - 2
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()