artem/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2026-04-22 08:39:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply suggestions from code review

Browse Source 
Co-authored-by: bashonly <88596187+bashonly@users.noreply.github.com>
This commit is contained in:
Jan Baier 2025-05-16 21:52:12 +02:00 committed by GitHub
parent f7b8fb8a1c
commit 64487d6b6d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
yt_dlp/extractor/iprima.py
View File

@ -15,8 +15,6 @@ class IPrimaIE(InfoExtractor):
_VALID_URL = r'https?://(?!cnn)(?:[^/]+)\.iprima\.cz/(?:[^/]+/)*(?P<id>[^/?#&]+)' _VALID_URL = r'https?://(?!cnn)(?:[^/]+)\.iprima\.cz/(?:[^/]+/)*(?P<id>[^/?#&]+)'
_GEO_BYPASS = False _GEO_BYPASS = False
_NETRC_MACHINE = 'iprima' _NETRC_MACHINE = 'iprima'
_AUTH_ROOT = 'https://ucet.iprima.cz'
_DEVICE_ID = 'Windows Chrome'
access_token = None access_token = None
_TESTS = [{ _TESTS = [{
@ -86,22 +84,18 @@ class IPrimaIE(InfoExtractor):
if self.access_token: if self.access_token:
return 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( 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', 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')) self.access_token = token_data['accessToken']['value']
if self.access_token is None: if not self.access_token:
raise ExtractorError('Getting token failed', expected=True) raise ExtractorError('Failed to fetch access token')
def _real_initialize(self): def _real_initialize(self):
if not self.access_token: if not self.access_token: