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

Merge 291ea778c3 into 05c8023a27

Browse source
This commit is contained in:
v3DJG6GL 2025-03-07 22:15:21 +01:00 committed by GitHub
commit 5db640a5be
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 25 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
yt_dlp/extractor/playsuisse.py
View file

@ -120,6 +120,12 @@ class PlaySuisseIE(InfoExtractor):
id id
name name
description description
descriptionLong
year
contentTypes
directors
mainCast
productionCountries
duration duration
episodeNumber episodeNumber
seasonNumber seasonNumber
@ -215,9 +221,7 @@ def _perform_login(self, username, password):
if not self._ID_TOKEN: if not self._ID_TOKEN:
raise ExtractorError('Login failed') raise ExtractorError('Login failed')
def _get_media_data(self, media_id): def _get_media_data(self, media_id, locale):
# NOTE In the web app, the "locale" header is used to switch between languages,
# However this doesn't seem to take effect when passing the header here.
response = self._download_json( response = self._download_json(
'https://www.playsuisse.ch/api/graphql', 'https://www.playsuisse.ch/api/graphql',
media_id, data=json.dumps({ media_id, data=json.dumps({
@ -225,8 +229,7 @@ def _get_media_data(self, media_id):
'query': self._GRAPHQL_QUERY, 'query': self._GRAPHQL_QUERY,
'variables': {'assetId': media_id}, 'variables': {'assetId': media_id},
}).encode(), }).encode(),
headers={'Content-Type': 'application/json', 'locale': 'de'}) headers={'Content-Type': 'application/json', 'locale': locale})
return response['data']['assetV2'] return response['data']['assetV2']
def _real_extract(self, url): def _real_extract(self, url):
@ -234,7 +237,10 @@ def _real_extract(self, url):
self.raise_login_required(method='password') self.raise_login_required(method='password')
media_id = self._match_id(url) media_id = self._match_id(url)
media_data = self._get_media_data(media_id) query = parse_qs(url)
locale_param = (query.get('locale') or ['de'])[0].lower()
locale = locale_param if locale_param in {'de', 'fr', 'it', 'rm'} else 'de'
media_data = self._get_media_data(media_id, locale)
info = self._extract_single(media_data) info = self._extract_single(media_data)
if media_data.get('episodes'): if media_data.get('episodes'):
info.update({ info.update({
@ -257,15 +263,20 @@ def _extract_single(self, media_data):
self._merge_subtitles(subs, target=subtitles) self._merge_subtitles(subs, target=subtitles)
return { return {
'id': media_data['id'], 'id': traverse_obj(media_data, 'id'),
'title': media_data.get('name'), 'title': traverse_obj(media_data, 'name'),
'description': media_data.get('description'), 'description': traverse_obj(media_data, 'descriptionLong') or traverse_obj(media_data, 'description'),
'genres': traverse_obj(media_data, 'contentTypes'),
'creators': traverse_obj(media_data, 'directors'),
'cast': traverse_obj(media_data, 'mainCast'),
'location': traverse_obj(media_data, 'productionCountries'),
'release_year': int_or_none(traverse_obj(media_data, 'year', lambda _, x: x[:4])),
'thumbnails': thumbnails, 'thumbnails': thumbnails,
'duration': int_or_none(media_data.get('duration')), 'duration': int_or_none(traverse_obj(media_data, 'duration')),
'formats': formats, 'formats': formats,
'subtitles': subtitles, 'subtitles': subtitles,
'series': media_data.get('seriesName'), 'series': traverse_obj(media_data, 'seriesName'),
'season_number': int_or_none(media_data.get('seasonNumber')), 'season_number': int_or_none(traverse_obj(media_data, 'seasonNumber')),
'episode': media_data.get('name') if media_data.get('episodeNumber') else None, 'episode': traverse_obj(media_data, 'name') if traverse_obj(media_data, 'episodeNumber') else None,
'episode_number': int_or_none(media_data.get('episodeNumber')), 'episode_number': int_or_none(traverse_obj(media_data, 'episodeNumber')),
} }