Pārlūkot izejas kodu

[go] Improve geo restriction detection and use geo bypass mechanism

Sergey M․ 8 gadi atpakaļ
vecāks
revīzija
e633f21a96
1 mainītis faili ar 4 papildinājumiem un 0 dzēšanām
  1. 4 0
      youtube_dl/extractor/go.py

+ 4 - 0
youtube_dl/extractor/go.py

@@ -101,6 +101,10 @@ class GoIE(AdobePassIE):
                         video_id, data=urlencode_postdata(data), headers=self.geo_verification_headers())
                     errors = entitlement.get('errors', {}).get('errors', [])
                     if errors:
+                        for error in errors:
+                            if error.get('code') == 1002:
+                                self.raise_geo_restricted(
+                                    error['message'], countries=['US'])
                         error_message = ', '.join([error['message'] for error in errors])
                         raise ExtractorError('%s said: %s' % (self.IE_NAME, error_message), expected=True)
                     asset_url += '?' + entitlement['uplynkData']['sessionKey']