diff options
Diffstat (limited to 'youtube/account_functions.py')
| -rw-r--r-- | youtube/account_functions.py | 35 |
1 files changed, 31 insertions, 4 deletions
diff --git a/youtube/account_functions.py b/youtube/account_functions.py index 985f7d1..f34e5f2 100644 --- a/youtube/account_functions.py +++ b/youtube/account_functions.py @@ -1,8 +1,11 @@ # Contains functions having to do with logging in or requiring that one is logged in + import urllib import json from youtube import common, proto, comments import re +import traceback + def _post_comment(text, video_id, session_token, cookie): headers = { 'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 10_3_1 like Mac OS X) AppleWebKit/603.1.30 (KHTML, like Gecko) Version/10.0 Mobile/14E304 Safari/602.1', @@ -30,6 +33,10 @@ def _post_comment(text, video_id, session_token, cookie): req = urllib.request.Request("https://m.youtube.com/service_ajax?name=createCommentEndpoint", headers=headers, data=data) response = urllib.request.urlopen(req, timeout = 5) content = response.read() + content = common.decode_content(content, response.getheader('Content-Encoding', default='identity')) + code = json.loads(content)['code'] + print("Comment posting code: " + code) + return code '''with open('debug/post_comment_response', 'wb') as f: f.write(content)''' @@ -61,6 +68,10 @@ def _post_comment_reply(text, video_id, parent_comment_id, session_token, cookie req = urllib.request.Request("https://m.youtube.com/service_ajax?name=createCommentReplyEndpoint", headers=headers, data=data) response = urllib.request.urlopen(req, timeout = 5) content = response.read() + content = common.decode_content(content, response.getheader('Content-Encoding', default='identity')) + code = json.loads(content)['code'] + print("Comment posting code: " + code) + return code '''with open('debug/post_comment_response', 'wb') as f: f.write(content)''' @@ -115,11 +126,27 @@ def post_comment(query_string, fields): raise Exception("Couldn't find xsrf_token") if 'parent_id' in parameters: - _post_comment_reply(fields['comment_text'][0], parameters['video_id'][0], parameters['parent_id'][0], token, cookie_data) - return comments.get_comments_page(query_string) + code = _post_comment_reply(fields['comment_text'][0], parameters['video_id'][0], parameters['parent_id'][0], token, cookie_data) + try: + response = comments.get_comments_page(query_string) + except socket.error as e: + traceback.print_tb(e.__traceback__) + return b'Refreshing comment page yielded error 502 Bad Gateway.\nPost comment status code: ' + code.encode('ascii') + except Exception as e: + traceback.print_tb(e.__traceback__) + return b'Refreshing comment page yielded error 500 Internal Server Error.\nPost comment status code: ' + code.encode('ascii') + return response else: - _post_comment(fields['comment_text'][0], fields['video_id'][0], token, cookie_data) - return comments.get_comments_page('ctoken=' + comments.make_comment_ctoken(video_id, sort=1)) + code = _post_comment(fields['comment_text'][0], fields['video_id'][0], token, cookie_data) + try: + response = comments.get_comments_page('ctoken=' + comments.make_comment_ctoken(video_id, sort=1)) + except socket.error as e: + traceback.print_tb(e.__traceback__) + return b'Refreshing comment page yielded error 502 Bad Gateway.\nPost comment status code: ' + code.encode('ascii') + except Exception as e: + traceback.print_tb(e.__traceback__) + return b'Refreshing comment page yielded error 500 Internal Server Error.\nPost comment status code: ' + code.encode('ascii') + return response |