1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-03-09 12:50:23 -05:00

Adhere coding styles

This commit is contained in:
Michaël De Boey 2025-03-06 01:05:01 +01:00
parent 5386daa073
commit d7de58b405
No known key found for this signature in database

View file

@ -92,34 +92,22 @@ def _call_api(self, video_id, client='null', id_token=None, version='v2'):
player_info = {'exp': (round(time.time(), 3) + 900), **self._PLAYER_INFO} player_info = {'exp': (round(time.time(), 3) + 900), **self._PLAYER_INFO}
vrt_player_token = self._download_json( vrt_player_token = self._download_json(
f'https://media-services-public.vrt.be/vualto-video-aggregator-web/rest/external/{version}/tokens', f'https://media-services-public.vrt.be/vualto-video-aggregator-web/rest/external/{version}/tokens',
video_id, video_id, 'Downloading player token', 'Failed to download player token', headers={
'Downloading player token',
'Failed to download player token',
headers={
**self.geo_verification_headers(), **self.geo_verification_headers(),
'Content-Type': 'application/json', 'Content-Type': 'application/json',
}, }, data=json.dumps({
data=json.dumps({
'identityToken': id_token or '', 'identityToken': id_token or '',
'playerInfo': jwt_encode_hs256( 'playerInfo': jwt_encode_hs256(player_info, self._JWT_SIGNING_KEY, headers={
player_info, 'kid': self._JWT_KEY_ID,
self._JWT_SIGNING_KEY, }).decode(),
headers={'kid': self._JWT_KEY_ID}, }, separators=(',', ':')).encode())['vrtPlayerToken']
).decode(),
}, separators=(',', ':')).encode(),
)['vrtPlayerToken']
return self._download_json( return self._download_json(
f'https://media-services-public.vrt.be/media-aggregator/{version}/media-items/{video_id}', f'https://media-services-public.vrt.be/media-aggregator/{version}/media-items/{video_id}',
video_id, video_id, 'Downloading API JSON', 'Failed to download API JSON', query={
'Downloading API JSON',
'Failed to download API JSON',
query={
'client': client, 'client': client,
'vrtPlayerToken': vrt_player_token, 'vrtPlayerToken': vrt_player_token,
}, }, expected_status=400)
expected_status=400,
)
class VRTIE(VRTBaseIE): class VRTIE(VRTBaseIE):
@ -307,11 +295,8 @@ def _fetch_video_token(self):
return None return None
self._download_webpage( self._download_webpage(
'https://www.vrt.be/vrtmax/sso/login', 'https://www.vrt.be/vrtmax/sso/login', None,
None, 'Refreshing video token', query={'scope': 'openid,mid'})
'Refreshing video token',
query={'scope': 'openid,mid'},
)
video_token = self._get_video_token_from_cookie() video_token = self._get_video_token_from_cookie()
if not video_token: if not video_token:
@ -334,37 +319,22 @@ def _perform_login(self, username, password):
return return
self._request_webpage( self._request_webpage(
'https://www.vrt.be/vrtmax/sso/login', 'https://www.vrt.be/vrtmax/sso/login', None,
None, note='Getting session cookies', errnote='Failed to get session cookies')
note='Getting session cookies',
errnote='Failed to get session cookies',
)
login_data = self._download_json( login_data = self._download_json(
'https://login.vrt.be/perform_login', 'https://login.vrt.be/perform_login', None, data=json.dumps({
None,
data=json.dumps({
'clientId': 'vrtnu-site', 'clientId': 'vrtnu-site',
'loginID': username, 'loginID': username,
'password': password, 'password': password,
}).encode(), }).encode(), headers={
headers={
'Content-Type': 'application/json', 'Content-Type': 'application/json',
'Oidcxsrf': self._get_cookies('https://login.vrt.be').get('OIDCXSRF').value, 'Oidcxsrf': self._get_cookies('https://login.vrt.be').get('OIDCXSRF').value,
}, }, note='Logging in', errnote='Login failed', expected_status=403)
note='Logging in',
errnote='Login failed',
expected_status=403,
)
if login_data.get('errorCode'): if login_data.get('errorCode'):
raise ExtractorError(f'Login failed: {login_data.get("errorMessage")}', expected=True) raise ExtractorError(f'Login failed: {login_data.get("errorMessage")}', expected=True)
self._download_webpage( self._download_webpage(login_data['redirectUrl'], None, note='Getting access token', errnote='Failed to get access token')
login_data['redirectUrl'],
None,
note='Getting access token',
errnote='Failed to get access token',
)
self.cache.store(self._NETRC_MACHINE, 'video_token', self._get_video_token_from_cookie()) self.cache.store(self._NETRC_MACHINE, 'video_token', self._get_video_token_from_cookie())
self.cache.store(self._NETRC_MACHINE, 'refresh_token', self._get_refresh_token_from_cookie()) self.cache.store(self._NETRC_MACHINE, 'refresh_token', self._get_refresh_token_from_cookie())
@ -374,23 +344,18 @@ def _real_extract(self, url):
metadata = self._download_json( metadata = self._download_json(
'https://www.vrt.be/vrtnu-api/graphql/public/v1', 'https://www.vrt.be/vrtnu-api/graphql/public/v1',
display_id, display_id, 'Downloading asset JSON', 'Unable to download asset JSON',
'Downloading asset JSON',
'Unable to download asset JSON',
data=json.dumps({ data=json.dumps({
'operationName': 'VideoPage', 'operationName': 'VideoPage',
'query': self._VIDEO_PAGE_QUERY, 'query': self._VIDEO_PAGE_QUERY,
'variables': { 'variables': {'pageId': urllib.parse.urlparse(url).path},
'pageId': urllib.parse.urlparse(url).path,
},
}).encode(), }).encode(),
headers={ headers={
'content-type': 'application/json', 'content-type': 'application/json',
'x-vrt-client-name': 'WEB', 'x-vrt-client-name': 'WEB',
'x-vrt-client-version': '1.5.9', 'x-vrt-client-version': '1.5.9',
'x-vrt-zone': 'default', 'x-vrt-zone': 'default',
}, })['data']['page']
)['data']['page']
video_id = metadata['player']['modes'][0]['streamId'] video_id = metadata['player']['modes'][0]['streamId']
video_token = self._fetch_video_token() video_token = self._fetch_video_token()
@ -414,11 +379,8 @@ def _real_extract(self, url):
raise ExtractorError(f'Unable to extract formats: {code}') raise ExtractorError(f'Unable to extract formats: {code}')
return { return {
**self._json_ld( 'id': video_id,
traverse_obj(metadata, ('ldjson', ..., {json.loads})), **self._json_ld(traverse_obj(metadata, ('ldjson', ..., {json.loads})), video_id, fatal=False),
video_id,
fatal=False,
),
**traverse_obj(metadata, ('episode', { **traverse_obj(metadata, ('episode', {
'age_limit': ('ageRaw', {parse_age_limit}), 'age_limit': ('ageRaw', {parse_age_limit}),
'description': ('description', {str}), 'description': ('description', {str}),
@ -437,7 +399,6 @@ def _real_extract(self, url):
'display_id': display_id, 'display_id': display_id,
'duration': float_or_none(streaming_info.get('duration'), 1000), 'duration': float_or_none(streaming_info.get('duration'), 1000),
'formats': formats, 'formats': formats,
'id': video_id,
'subtitles': subtitles, 'subtitles': subtitles,
'thumbnail': url_or_none(streaming_info.get('posterImageUrl')), 'thumbnail': url_or_none(streaming_info.get('posterImageUrl')),
'_old_archive_ids': [make_archive_id('Canvas', video_id)], '_old_archive_ids': [make_archive_id('Canvas', video_id)],