diff options
author | marieell <marieell@tuta.io> | 2022-02-15 10:16:49 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-02-15 01:16:49 -0800 |
commit | 5c6dfc1f79f00e65e487b2e3fcebab9d89d16e86 (patch) | |
tree | 652bc6b8d06fd1713ffa819e6c363d4ee99b2197 | |
parent | c2a8547fdcc8b9c3c2da43a6cb18281bbf429870 (diff) | |
download | hypervideo-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.py | 6 |
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] |