From a3e6c36926c7adc0ce032385d4179aecd0e59375 Mon Sep 17 00:00:00 2001
From: b3yond <b3yond@riseup.net>
Date: Sat, 17 Jun 2017 20:16:03 +0200
Subject: [PATCH] everything is in small beautiful functions now

---
 retweetbot.py | 108 ++++++++++++++++++++++++++++----------------------
 1 file changed, 60 insertions(+), 48 deletions(-)

diff --git a/retweetbot.py b/retweetbot.py
index 82ad724..d37ec86 100644
--- a/retweetbot.py
+++ b/retweetbot.py
@@ -16,7 +16,9 @@ class Retweetbot(object):
 
     def __init__(self, keypath="appkeys/ticketfrei@twitter.com",
                  historypath="last_mention",
-                 triggerpath="triggerwords"):
+                 triggerpath="triggerwords",
+                 user_id="801098086005243904",
+                 screen_name="links_tech"):
         """
         Initializes the bot and loads all the necessary data.
 
@@ -31,6 +33,8 @@ class Retweetbot(object):
                                access_token_secret=keys[3].strip())
         self.historypath = historypath
         self.triggerpath = triggerpath
+        self.user_id = user_id
+        self.screen_name = screen_name
         self.last_mention = bot.get_history(self.historypath)
         self.triggers = bot.get_trigger(self.triggerpath)
 
@@ -71,61 +75,69 @@ class Retweetbot(object):
         toot = status.user.name + ": " + status.text
         return toot
 
+    def crawl_mentions(self):
+        done = False
+        mentions = []
+        while not done:
+            try:
+                mentions = self.api.GetMentions(since_id=self.last_mention)
+                done = True
+            except requests.exceptions.ConnectionError:
+                print("[ERROR] Bad Connection.")
+                sleep(10)
+        return mentions
+
+    def trigger_rt(self, status):
+        for triggerword in self.triggers:
+            if status.text.lower().find(triggerword):
+                return True
+        return False
+
+    def retweet(self, status):
+        done = False
+        while not done:
+            try:
+                self.api.PostRetweet(status.id)
+                self.bridge_mastodon(status)
+                done = True
+
+            # Hopefully we got rid of this error. If not, try to uncomment these lines.
+            # except twitter.error.TwitterError:
+            #    print("[ERROR] probably you already retweeted this tweet.")
+            #    done = True
+            except requests.exceptions.ConnectionError:
+                print("[ERROR] Bad Connection.")
+                sleep(10)
+
     def flow(self):
-        """
+        """ The flow of crawling mentions and retweeting them."""
 
-        """
-        try:
-            while 1:
-                sleep(1)
+        # Store all mentions in a list of Status Objects
+        mentions = self.crawl_mentions()
 
-                # Store all mentions in a list of Status Objects
-                done = False
-                while not done:
-                    try:
-                        mentions = self.api.GetMentions(since_id=self.last_mention)
-                        done = True
-                    except requests.exceptions.ConnectionError:
-                        print("[ERROR] Bad Connection.")
-                        sleep(10)
+        for status in mentions:
+            # Is the Text of the Tweet in the triggerlist?
+            should_retweet = self.trigger_rt(status)
 
-                print mentions  # debug
-                for status in mentions:
+            # Retweet status
+            if should_retweet:
+                self.retweet(status)
 
-                    # Is the Text of the Tweet in the triggerlist?
-                    should_retweet = False
-                    for triggerword in self.triggers:
-                        if status.text.lower().find(triggerword):
-                            should_retweet = True
-                            break
+            # save the id so it doesn't get crawled again
+            self.last_mention = status.id
+            print self.last_mention
 
-                    # Retweet status
-                    if should_retweet:
-                        done = False
-                        while not done:
-                            try:
-                                self.api.PostRetweet(status.id)
-                                self.bridge_mastodon(status)
-                                done = True
-
-                            # Hopefully we got rid of this error. If not, try to uncomment these lines.
-                            # except twitter.error.TwitterError:
-                            #    print("[ERROR] probably you already retweeted this tweet.")
-                            #    done = True
-                            except requests.exceptions.ConnectionError:
-                                print("[ERROR] Bad Connection.")
-                                sleep(10)
-
-                    # save the id so it doesn't get crawled again
-                    self.last_mention = status.id
-                    print self.last_mention
-        except:
-            print "[ERROR] Shit went wrong, closing down."
-            with open(self.historypath, "w") as f:
-                f.write(str(self.last_mention))
-            self.api.PostDirectMessage("Help! I broke down. restart me pls :$", "801098086005243904", "links_tech")
+    def shutdown(self):
+        print "[ERROR] Shit went wrong, closing down."
+        with open(self.historypath, "w") as f:
+            f.write(str(self.last_mention))
+        self.api.PostDirectMessage("Help! I broke down. restart me pls :$", self.user_id, self.screen_name)
 
 
 if __name__ == "main":
     # create an Api object
     bot = Retweetbot()
+    try:
+        bot.flow()
+    except:
+        bot.shutdown()