From 89e378ffd9610a3caa12924ed40f01023e391f89 Mon Sep 17 00:00:00 2001 From: Simon Date: Wed, 8 Aug 2018 22:41:25 -0400 Subject: [PATCH] Reddit comment callback is not an edit instead of a new comment --- callbacks.py | 4 ++-- reddit_bot.py | 15 +++++++++++++-- 2 files changed, 15 insertions(+), 4 deletions(-) diff --git a/callbacks.py b/callbacks.py index fec098a..a238cdb 100644 --- a/callbacks.py +++ b/callbacks.py @@ -52,12 +52,12 @@ class RedditCommentCallback(RedditCallback): def run(self, task_result: TaskResult, search: SearchEngine): comment_id = self.task.callback_args["comment_id"] - print("Replying to comment " + comment_id) + print("Editing comment comment " + comment_id) stats = search.get_stats(self.task.website_id) message = self.reddit_bot.get_comment(stats, self.task.website_id) print(message) - self.reddit_bot.reply(self.reddit_bot.reddit.comment(comment_id), message) + self.reddit_bot.edit(self.reddit_bot.reddit.comment(comment_id), message) class DiscordCallback(PostCrawlCallback): diff --git a/reddit_bot.py b/reddit_bot.py index bff336f..26defae 100644 --- a/reddit_bot.py +++ b/reddit_bot.py @@ -41,17 +41,28 @@ class RedditBot: while True: try: - # Double check has_crawled if not self.has_crawled(reddit_obj.id): - reddit_obj.reply(comment) + reply = reddit_obj.reply(comment) self.log_crawl(reddit_obj.id) print("Reply to " + reddit_obj.id) + return reply break except Exception as e: print("Waiting 5 minutes: " + str(e)) time.sleep(300) continue + def edit(self, reddit_comment, new_message): + + while True: + try: + reddit_comment.edit(new_message) + print("Edit comment " + reddit_comment.id) + except Exception as e: + print("Waiting 5 minutes: " + str(e)) + time.sleep(300) + continue + @staticmethod def get_comment(stats: dict, website_id, message: str = ""): comment = message + " \n" if message else ""