From 64487d6b6d19e2758f453e35cf667dfd752b0de3 Mon Sep 17 00:00:00 2001 From: Jan Baier <7996094+baierjan@users.noreply.github.com> Date: Fri, 16 May 2025 21:52:12 +0200 Subject: [PATCH] Apply suggestions from code review Co-authored-by: bashonly <88596187+bashonly@users.noreply.github.com> --- yt_dlp/extractor/iprima.py | 24 +++++++++--------------- 1 file changed, 9 insertions(+), 15 deletions(-) diff --git a/yt_dlp/extractor/iprima.py b/yt_dlp/extractor/iprima.py index da1dcec055..ffb8184202 100644 --- a/yt_dlp/extractor/iprima.py +++ b/yt_dlp/extractor/iprima.py @@ -15,8 +15,6 @@ class IPrimaIE(InfoExtractor): _VALID_URL = r'https?://(?!cnn)(?:[^/]+)\.iprima\.cz/(?:[^/]+/)*(?P[^/?#&]+)' _GEO_BYPASS = False _NETRC_MACHINE = 'iprima' - _AUTH_ROOT = 'https://ucet.iprima.cz' - _DEVICE_ID = 'Windows Chrome' access_token = None _TESTS = [{ @@ -86,22 +84,18 @@ class IPrimaIE(InfoExtractor): if self.access_token: return - token_request_data = json.dumps({ - 'email': username, - 'password': password, - 'deviceName': self._DEVICE_ID, - }).encode() - - token_request_headers = {'content-type': 'application/json'} - token_data = self._download_json( - f'{self._AUTH_ROOT}/api/session/create', None, + 'https://ucet.iprima.cz/api/session/create', None, note='Downloading token', errnote='Downloading token failed', - data=token_request_data, headers=token_request_headers) + data=json.dumps({ + 'email': username, + 'password': password, + 'deviceName': 'Windows Chrome', + }).encode(), headers={'content-type': 'application/json'}) - self.access_token = traverse_obj(token_data, ('accessToken', 'value')) - if self.access_token is None: - raise ExtractorError('Getting token failed', expected=True) + self.access_token = token_data['accessToken']['value'] + if not self.access_token: + raise ExtractorError('Failed to fetch access token') def _real_initialize(self): if not self.access_token: