etc/yt-dlp
0
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-22 02:15:12 +00:00
Code Issues Projects Releases Wiki Activity

[youtube] Fix _extract_alerts

Browse source 
* Sometimes one warning is split between multiple runs
* Also simplified code
This commit is contained in:
pukkandan 2021-04-09 17:44:23 +05:30
parent f1823403b0
commit 3ffc7c89b0
No known key found for this signature in database
GPG key ID: 0F00D95A001F4698
1 changed files with 12 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
yt_dlp/extractor/youtube.py
View file

@ -3222,25 +3222,26 @@ def _real_extract_alerts():
alert_type = alert.get('type')
if not alert_type:
continue
message = try_get(alert, lambda x: x['text']['simpleText'], compat_str)
message = try_get(alert, lambda x: x['text']['simpleText'], compat_str) or ''
if message:
yield alert_type, message
for run in try_get(alert, lambda x: x['text']['runs'], list) or []:
message = try_get(run, lambda x: x['text'], compat_str)
if message:
yield alert_type, message
message += try_get(run, lambda x: x['text'], compat_str)
if message:
yield alert_type, message
err_msg = None
errors = []
warnings = []
for alert_type, alert_message in _real_extract_alerts():
if alert_type.lower() == 'error':
if err_msg:
self._downloader.report_warning('YouTube said: %s - %s' % ('ERROR', err_msg))
err_msg = alert_message
errors.append([alert_type, alert_message])
else:
self._downloader.report_warning('YouTube said: %s - %s' % (alert_type, alert_message))
warnings.append([alert_type, alert_message])
if err_msg:
raise ExtractorError('YouTube said: %s' % err_msg, expected=expected)
for alert_type, alert_message in (warnings + errors[:-1]):
self._downloader.report_warning('YouTube said: %s - %s' % (alert_type, alert_message))
if errors:
raise ExtractorError('YouTube said: %s' % errors[-1][1], expected=expected)
def _extract_webpage(self, url, item_id):
retries = self._downloader.params.get('extractor_retries', 3)