From 6464bfa78cd14cace52f6f1978b2cb90d47eb8da Mon Sep 17 00:00:00 2001 From: bashonly Date: Wed, 30 Apr 2025 01:26:36 -0500 Subject: [PATCH] OK to use `assert` for developer errors Authored by: bashonly --- yt_dlp/extractor/zdf.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/yt_dlp/extractor/zdf.py b/yt_dlp/extractor/zdf.py index 36d4e5c2c7..a89b909d4b 100644 --- a/yt_dlp/extractor/zdf.py +++ b/yt_dlp/extractor/zdf.py @@ -5,7 +5,6 @@ import time from .common import InfoExtractor from ..utils import ( - ExtractorError, determine_ext, filter_dict, float_or_none, @@ -141,19 +140,17 @@ class ZDFBaseIE(InfoExtractor): } def _download_graphql(self, item_id, data_desc, query=None, body=None): - if not query and not body: - raise ExtractorError( - 'GraphQL API requires either query parameters or a body', - video_id=item_id) + assert query or body, 'One of query or body is required' return self._download_json( - 'https://api.zdf.de/graphql', item_id, note=f'Downloading {data_desc}', - errnote=f'Failed to download {data_desc}', query=query, - data=json.dumps(body).encode() if body else None, headers={ + 'https://api.zdf.de/graphql', item_id, + f'Downloading {data_desc}', f'Failed to download {data_desc}', + query=query, data=json.dumps(body).encode() if body else None, + headers=filter_dict({ 'Api-Auth': self._get_api_token(), 'Apollo-Require-Preflight': True, 'Content-Type': 'application/json' if body else None, - }) + })) @staticmethod def _extract_thumbnails(source):