From 5c6dfc1f79f00e65e487b2e3fcebab9d89d16e86 Mon Sep 17 00:00:00 2001 From: marieell Date: Tue, 15 Feb 2022 10:16:49 +0100 Subject: [PATCH] [ATVAt] Detect geo-restriction (#2777) Authored by: marieell --- yt_dlp/extractor/atvat.py | 6 ++++++ 1 file changed, 6 insertions(+) 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]