From babdf8ff6466a132c1d41dabba4531f8320fafee Mon Sep 17 00:00:00 2001 From: Felix Stupp Date: Thu, 18 Aug 2022 11:39:22 +0000 Subject: [PATCH] extr/media/youtube: Reraise exceptions from extraction as ExtractionErrors --- .../entertainment_decider/extractors/media/youtube.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/server/entertainment_decider/extractors/media/youtube.py b/server/entertainment_decider/extractors/media/youtube.py index 131dd58..b1a6e64 100644 --- a/server/entertainment_decider/extractors/media/youtube.py +++ b/server/entertainment_decider/extractors/media/youtube.py @@ -97,10 +97,13 @@ class YoutubeMediaExtractor(MediaExtractor[YoutubeVideoData]): if not uri_match: raise Exception(f"URI not suitable: {uri!r}") id = uri_match.group("id") - vid_data: YoutubeVideoData = Video.getInfo( - videoLink=f"https://www.youtube.com/watch?v={id}", - mode=ResultMode.dict, - ) + try: + vid_data: YoutubeVideoData = Video.getInfo( + videoLink=f"https://www.youtube.com/watch?v={id}", + mode=ResultMode.dict, + ) + except Exception as e: + raise ExtractionError() from e if vid_data["isLiveNow"]: raise ExtractionError("Video is live, so pass extraction") return ExtractedData[YoutubeVideoData](