heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Extraction: Add fallback playability error extraction from renderers

Browse Source
This commit is contained in:
James Taylor 2019-12-14 22:08:36 -08:00
parent a04aa63efe
commit e870eea057
1 changed files with 24 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
youtube/yt_data_extract.py
View File

@ -1091,6 +1091,25 @@ def extract_formats(info, player_response):
info['formats'].append(fmt) info['formats'].append(fmt)
def extract_playability_error(info, player_response, error_prefix=''):
if info['formats']:
info['playability_error'] = None
return
playability_status = default_multi_get(player_response, 'playabilityStatus', 'status', default=None)
info['playability_status'] = playability_status
playability_reason = extract_plain_text(multi_default_multi_get(player_response,
['playabilityStatus', 'reason'],
['playabilityStatus', 'errorScreen', 'playerErrorMessageRenderer', 'reason'],
default='Could not find playability error')
)
if playability_status not in (None, 'OK'):
info['playability_error'] = error_prefix + playability_reason
else:
info['playability_error'] = error_prefix + 'Unknown playability error'
SUBTITLE_FORMATS = ('srv1', 'srv2', 'srv3', 'ttml', 'vtt') SUBTITLE_FORMATS = ('srv1', 'srv2', 'srv3', 'ttml', 'vtt')
def extract_watch_info(polymer_json): def extract_watch_info(polymer_json):
info = {'playability_error': None, 'error': None} info = {'playability_error': None, 'error': None}
@ -1146,16 +1165,12 @@ def extract_watch_info(polymer_json):
# formats # formats
extract_formats(info, player_response) extract_formats(info, player_response)
playability_status = default_multi_get(player_response, 'playabilityStatus', 'status', default=None)
playability_reason = default_multi_get(player_response, 'playabilityStatus', 'reason', default='Could not find playability error') # playability errors
if not info['formats']: extract_playability_error(info, player_response)
if playability_status not in (None, 'OK'):
info['playability_error'] = playability_reason
else:
info['playability_error'] = 'Unknown playability error'
# check age-restriction # check age-restriction
info['age_restricted'] = (playability_status == 'LOGIN_REQUIRED' and playability_reason and ' age' in playability_reason) info['age_restricted'] = (info['playability_status'] == 'LOGIN_REQUIRED' and info['playability_reason'] and ' age' in info['playability_reason'])
# base_js (for decryption of signatures) # base_js (for decryption of signatures)
info['base_js'] = default_multi_get(top_level, 'player', 'assets', 'js') info['base_js'] = default_multi_get(top_level, 'player', 'assets', 'js')
@ -1202,12 +1217,4 @@ def update_with_age_restricted_info(info, video_info_page):
return return
extract_formats(info, player_response) extract_formats(info, player_response)
if info['formats']: extract_playability_error(info, player_response, error_prefix=ERROR_PREFIX)
info['playability_error'] = None
else:
playability_status = default_multi_get(player_response, 'playabilityStatus', 'status', default=None)
playability_reason = default_multi_get(player_response, 'playabilityStatus', 'reason', default=ERROR_PREFIX + 'Could not find playability error')
if playability_status not in (None, 'OK'):
info['playability_error'] = ERROR_PREFIX + playability_reason
else:
info['playability_error'] = ERROR_PREFIX + 'Unknown playability error'