diff options
Diffstat (limited to 'youtube/comments.py')
| -rw-r--r-- | youtube/comments.py | 57 |
1 files changed, 42 insertions, 15 deletions
diff --git a/youtube/comments.py b/youtube/comments.py index 8dc9d86..2fb1fa2 100644 --- a/youtube/comments.py +++ b/youtube/comments.py @@ -7,6 +7,7 @@ import json import base64 import urllib import re +import traceback import flask from flask import request @@ -137,23 +138,49 @@ def post_process_comments_info(comments_info): def video_comments(video_id, sort=0, offset=0, lc='', secret_key=''): - if settings.comments_mode: - comments_info = yt_data_extract.extract_comments_info(request_comments(make_comment_ctoken(video_id, sort, offset, lc, secret_key))) - post_process_comments_info(comments_info) - - other_sort_url = util.URL_ORIGIN + '/comments?ctoken=' + make_comment_ctoken(video_id, sort=1 - sort, lc=lc) - other_sort_text = 'Sort by ' + ('newest' if sort == 0 else 'top') - - this_sort_url = (util.URL_ORIGIN - + '/comments?ctoken=' - + make_comment_ctoken(video_id, sort=sort, lc=lc)) - - comments_info['comment_links'] = [(other_sort_text, other_sort_url), - ('Direct link', this_sort_url)] + try: + if settings.comments_mode: + comments_info = {'error': None} + other_sort_url = ( + util.URL_ORIGIN + '/comments?ctoken=' + + make_comment_ctoken(video_id, sort=1 - sort, lc=lc) + ) + other_sort_text = 'Sort by ' + ('newest' if sort == 0 else 'top') + + this_sort_url = (util.URL_ORIGIN + + '/comments?ctoken=' + + make_comment_ctoken(video_id, sort=sort, lc=lc)) + + comments_info['comment_links'] = [ + (other_sort_text, other_sort_url), + ('Direct link', this_sort_url) + ] + + comments_info.update(yt_data_extract.extract_comments_info( + request_comments( + make_comment_ctoken(video_id, sort, offset, lc, secret_key) + ) + )) + post_process_comments_info(comments_info) + + return comments_info + else: + return {} + except util.FetchError as e: + print('Error retrieving comments for ' + str(video_id)) + if e.code == '429' and settings.route_tor: + comments_info['error'] = 'Error: Youtube blocked the request because the Tor exit node is overutilized.' + if e.error_message: + comments_info['error'] += '\n\n' + e.error_message + comments_info['error'] += '\n\nExit node IP address: %s' % e.ip + else: + comments_info['error'] = traceback.format_exc() - return comments_info + except Exception as e: + print('Error retrieving comments for ' + str(video_id)) + comments_info['error'] = traceback.format_exc() - return {} + return comments_info |