From 81ff5ab99ca05c4559c604ba7517d17f4bc79ea8 Mon Sep 17 00:00:00 2001 From: James Taylor Date: Tue, 11 Aug 2020 19:47:37 -0700 Subject: extract_channel_info: Improve error extraction Use extract_str function since it's not always 'simpleText' Make sure we don't output an empty error message if we don't know what it is. channel.py: Don't check if error message is empty, check if it's None --- youtube/yt_data_extract/everything_else.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'youtube/yt_data_extract/everything_else.py') diff --git a/youtube/yt_data_extract/everything_else.py b/youtube/yt_data_extract/everything_else.py index ac1ab7d..d1389c6 100644 --- a/youtube/yt_data_extract/everything_else.py +++ b/youtube/yt_data_extract/everything_else.py @@ -21,10 +21,13 @@ def extract_channel_info(polymer_json, tab): # example terminated channel: https://www.youtube.com/channel/UCnKJeK_r90jDdIuzHXC0Org except KeyError: if response.get('alerts'): - return {'error': ' '.join( - deep_get(alert, 'alertRenderer', 'text', 'simpleText', default='') + error_string = ' '.join( + extract_str(deep_get(alert, 'alertRenderer', 'text'), default='') for alert in response['alerts'] - )} + ) + if not error_string: + error_string = 'Failed to extract error' + return {'error': error_string} elif deep_get(response, 'responseContext', 'errors'): for error in response['responseContext']['errors'].get('error', []): if error.get('code') == 'INVALID_VALUE' and error.get('location') == 'browse_id': -- cgit v1.2.3