aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authormarieell <marieell@tuta.io>2022-02-15 10:16:49 +0100
committerGitHub <noreply@github.com>2022-02-15 01:16:49 -0800
commit5c6dfc1f79f00e65e487b2e3fcebab9d89d16e86 (patch)
tree652bc6b8d06fd1713ffa819e6c363d4ee99b2197
parentc2a8547fdcc8b9c3c2da43a6cb18281bbf429870 (diff)
downloadhypervideo-pre-5c6dfc1f79f00e65e487b2e3fcebab9d89d16e86.tar.lz
hypervideo-pre-5c6dfc1f79f00e65e487b2e3fcebab9d89d16e86.tar.xz
hypervideo-pre-5c6dfc1f79f00e65e487b2e3fcebab9d89d16e86.zip
[ATVAt] Detect geo-restriction (#2777)
Authored by: marieell
-rw-r--r--yt_dlp/extractor/atvat.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/yt_dlp/extractor/atvat.py b/yt_dlp/extractor/atvat.py
index 7c30cfcbb..481a09737 100644
--- a/yt_dlp/extractor/atvat.py
+++ b/yt_dlp/extractor/atvat.py
@@ -8,6 +8,7 @@ from ..utils import (
float_or_none,
jwt_encode_hs256,
try_get,
+ ExtractorError,
)
@@ -94,6 +95,11 @@ class ATVAtIE(InfoExtractor):
})
video_id, videos_data = list(videos['data'].items())[0]
+ error_msg = try_get(videos_data, lambda x: x['error']['title'])
+ if error_msg == 'Geo check failed':
+ self.raise_geo_restricted(error_msg)
+ elif error_msg:
+ raise ExtractorError(error_msg)
entries = [
self._extract_video_info(url, contentResource[video['id']], video)
for video in videos_data]