From 89cb7f7d98d18c2ed53a2ce0a0e048c7e02d36ff Mon Sep 17 00:00:00 2001 From: HobbyistDev <105957301+HobbyistDev@users.noreply.github.com> Date: Thu, 11 Apr 2024 07:22:56 +0800 Subject: [PATCH] Simplify extraction process Co-authored-by: pukkandan --- yt_dlp/extractor/godresource.py | 20 +++++++------------- 1 file changed, 7 insertions(+), 13 deletions(-) diff --git a/yt_dlp/extractor/godresource.py b/yt_dlp/extractor/godresource.py index 54f2e55c4..723657270 100644 --- a/yt_dlp/extractor/godresource.py +++ b/yt_dlp/extractor/godresource.py @@ -48,22 +48,16 @@ class GodResourceIE(InfoExtractor): f'https://api.godresource.com/api/Streams/{display_id}', display_id) video_url = api_data['streamUrl'] - - # TODO: better name? - extraction_result = {} - if determine_ext(video_url) == 'm3u8': + if (ext := determine_ext(video_url)) == 'm3u8': formats, subtitles = self._extract_m3u8_formats_and_subtitles( api_data['streamUrl'], display_id) - - extraction_result = { - 'formats': formats, - 'subtitles': subtitles - } - elif determine_ext(video_url) == 'mp4': - extraction_result = { + elif ext == 'mp4': + formats, subtitles = [{ 'url': video_url, - 'ext': 'mp4' - } + 'ext': ext + }], {} + else: + raise ExtractorError(f'Unexpected video format {ext}') return { 'id': display_id,