diff options
Diffstat (limited to 'youtube')
| -rw-r--r-- | youtube/post_comment.py | 11 | ||||
| -rw-r--r-- | youtube/youtube.py | 9 |
2 files changed, 10 insertions, 10 deletions
diff --git a/youtube/post_comment.py b/youtube/post_comment.py index fc4d4b1..5166b6c 100644 --- a/youtube/post_comment.py +++ b/youtube/post_comment.py @@ -109,11 +109,18 @@ def get_session_token(video_id, cookiejar): else: raise Exception("Couldn't find xsrf_token") -def delete_comment(parameters, fields): +def delete_comment(env, start_response): + fields = env['fields'] video_id = fields['video_id'][0] cookiejar = accounts.account_cookiejar(fields['channel_id'][0]) token = get_session_token(video_id, cookiejar) - return _delete_comment(video_id, fields['comment_id'][0], fields['author_id'][0], token, cookiejar) + + code = _delete_comment(video_id, fields['comment_id'][0], fields['author_id'][0], token, cookiejar) + + if code == "SUCCESS": + start_response('303 See Other', [('Location', common.URL_ORIGIN + '/comment_delete_success'),] ) + else: + start_response('303 See Other', [('Location', common.URL_ORIGIN + '/comment_delete_fail'),] ) def post_comment(parameters, fields): channel_id = fields['channel_id'][0] diff --git a/youtube/youtube.py b/youtube/youtube.py index b9632c6..d5a64ab 100644 --- a/youtube/youtube.py +++ b/youtube/youtube.py @@ -21,6 +21,7 @@ get_handlers = { post_handlers = { 'edit_playlist': local_playlist.edit_playlist, 'login': accounts.add_account, + 'delete_comment': post_comment.delete_comment, } def youtube(env, start_response): @@ -123,14 +124,6 @@ def youtube(env, start_response): start_response('303 See Other', (('Location', common.URL_ORIGIN + '/comments?ctoken=' + comments.make_comment_ctoken(video_id, sort=1)),) ) return '' - elif path == "/delete_comment": - parameters = urllib.parse.parse_qs(query_string) - code = post_comment.delete_comment(parameters, fields) - if code == "SUCCESS": - start_response('303 See Other', (('Location', common.URL_ORIGIN + '/comment_delete_success'),) ) - else: - start_response('303 See Other', (('Location', common.URL_ORIGIN + '/comment_delete_fail'),) ) - else: start_response('404 Not Found', ()) return b'404 Not Found' |