Myzel394/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-07-01 14:00:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

[ie/hotstar] Fix formats extraction (#13585)

Browse Source 
Fix b5bd057fe86550f3aa67f2fc8790d1c6a251c57b

Authored by: bashonly
This commit is contained in:
bashonly 2025-06-30 14:19:43 -05:00 committed by GitHub
parent 958153a226
commit e9f157669e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
yt_dlp/extractor/hotstar.py
View File

@ -64,12 +64,16 @@ class HotStarBaseIE(InfoExtractor):
'container': ['fmp4br', 'fmp4'], 'container': ['fmp4br', 'fmp4'],
'ads': ['non_ssai', 'ssai'], 'ads': ['non_ssai', 'ssai'],
'audio_channel': ['atmos', 'dolby51', 'stereo'], 'audio_channel': ['atmos', 'dolby51', 'stereo'],
'encryption': ['plain'], 'encryption': ['plain', 'widevine'], # wv only so we can raise appropriate error
'video_codec': ['h265'], # or ['h264'] 'video_codec': ['h265', 'h264'],
'ladder': ['tv', 'full'], 'ladder': ['tv', 'full'],
'resolution': ['4k'], # or ['hd'] 'resolution': ['4k', 'hd'],
'true_resolution': ['4k'], # or ['hd'] 'true_resolution': ['4k', 'hd'],
'dynamic_range': ['hdr'], # or ['sdr'] 'dynamic_range': ['hdr', 'sdr'],
}, separators=(',', ':')),
'drm_parameters': json.dumps({
'widevine_security_level': ['SW_SECURE_DECODE', 'SW_SECURE_CRYPTO'],
'hdcp_version': ['HDCP_V2_2', 'HDCP_V2_1', 'HDCP_V2', 'HDCP_V1'],
}, separators=(',', ':')), }, separators=(',', ':')),
}, st=st, cookies=cookies) }, st=st, cookies=cookies)
@ -281,7 +285,7 @@ class HotStarIE(HotStarBaseIE):
self.report_drm(video_id) self.report_drm(video_id)
geo_restricted = False geo_restricted = False
formats, subs = [], {} formats, subs, has_drm = [], {}, False
headers = {'Referer': f'{self._BASE_URL}/in'} headers = {'Referer': f'{self._BASE_URL}/in'}
content_type = traverse_obj(video_data, ('contentType', {str})) or self._CONTENT_TYPE[video_type] content_type = traverse_obj(video_data, ('contentType', {str})) or self._CONTENT_TYPE[video_type]
@ -304,6 +308,11 @@ class HotStarIE(HotStarBaseIE):
for ignore in self._configuration_arg(key)): for ignore in self._configuration_arg(key)):
continue continue
tag_dict = dict((*t.split(':', 1), None)[:2] for t in tags.split(';'))
if tag_dict.get('encryption') not in ('plain', None):
has_drm = True
continue
format_url = re.sub(r'(?<=//staragvod)(\d)', r'web\1', playback_set['content_url']) format_url = re.sub(r'(?<=//staragvod)(\d)', r'web\1', playback_set['content_url'])
ext = determine_ext(format_url) ext = determine_ext(format_url)
@ -330,10 +339,6 @@ class HotStarIE(HotStarBaseIE):
self.write_debug(e) self.write_debug(e)
continue continue
tag_dict = dict((*t.split(':', 1), None)[:2] for t in tags.split(';'))
if tag_dict.get('encryption') not in ('plain', None):
for f in current_formats:
f['has_drm'] = True
for f in current_formats: for f in current_formats:
for k, v in self._TAG_FIELDS.items(): for k, v in self._TAG_FIELDS.items():
if not f.get(k): if not f.get(k):
@ -361,6 +366,8 @@ class HotStarIE(HotStarBaseIE):
if not formats and geo_restricted: if not formats and geo_restricted:
self.raise_geo_restricted(countries=['IN'], metadata_available=True) self.raise_geo_restricted(countries=['IN'], metadata_available=True)
elif not formats and has_drm:
self.report_drm(video_id)
self._remove_duplicate_formats(formats) self._remove_duplicate_formats(formats)
for f in formats: for f in formats:
f.setdefault('http_headers', {}).update(headers) f.setdefault('http_headers', {}).update(headers)