merged

goodlists/nbg_goodlist

@@ -1,6 +1,8 @@
-kontrolle
+kontroll?e
 konti
 db
+vgn
+vag
 zivil
 sicherheit
 uniform
@@ -11,12 +13,5 @@ tram
 linie
 nuernberg
 nürnberg
-s1
+s\d
-s2
+u\d\d?
-s3
-u1
-u2
-u3
-s4
-u21
-u11

retweetbot.py

@@ -191,10 +191,9 @@ if __name__ == "__main__":
     trigger = trigger.Trigger(config)
 
     bot = RetweetBot(trigger, config)
+    try:
         while True:
             bot.flow()
-        try:
+            sleep(3)
-            pass
     except:
         bot.shutdown()
-        sleep(3)

ticketfrei.cfg.example

@@ -10,13 +10,17 @@ server = 'yourmastodoninstance'
 consumer_key = "yourconsumerkey"
 consumer_secret = yourconsumersecret"
 
+# shutdown_contact_userid = 012345
+# shutdown_contact_screen_name = 'yourscreenname'
+
 [tuser]
 access_token_key = "youraccesstokenkey"
 access_token_secret = "youraccesstokensecret"
 
-# shutdown_contact_userid = 012345
-# shutdown_contact_screen_name = 'yourscreenname'
-
 # [trigger]
+# goodlists are one regex per line.
+# badlists are one badword per line.
+# a message musst match at least one regex in goodlist and contain none of the badwords.
+
 # goodlist_path = 'goodlists'
 # blacklist_path = 'blacklists'

ticketfrei.py

@@ -20,7 +20,7 @@ if __name__ == '__main__':
         statuses = []
         while True:
             statuses = mbot.retoot(statuses)
-            statuses = tbot.flow(statuses)  # XXX not implemented in RetweetBot
+            statuses = tbot.flow(statuses)
             time.sleep(1)
     except:
         tbot.shutdown()

trigger.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
-import os
 import pytoml as toml
+import os
+import re
 
 
 class Trigger(object):
@@ -11,33 +12,32 @@ class Trigger(object):
         self.config = config
 
         try:
-            self.goodlistpath = config['trigger']['goodlist_path']
+            goodlistpath = config['trigger']['goodlist_path']
         except KeyError:
-            self.goodlistpath = 'goodlists'
+            goodlistpath = 'goodlists'
-        self.goodlist = self.get_lists(self.goodlistpath)
+
+        # load goodlists
+        self.goodlist = []
+        for filename in os.listdir(goodlistpath):
+            with open(os.path.join(goodlistpath, filename), "r+") as listfile:
+                for pattern in listfile:
+                    pattern = pattern.strip()
+                    if pattern:
+                        self.goodlist.append(re.compile(pattern))
 
         try:
-            self.blacklistpath = config['trigger']['blacklist_path']
+            blacklistpath = config['trigger']['blacklist_path']
         except KeyError:
-            self.blacklistpath = 'blacklists'
+            blacklistpath = 'blacklists'
-        self.blacklist = self.get_lists(self.blacklistpath)
 
-    def get_lists(self, path):
+        # load blacklists
-        """
+        self.blacklist = set()
-        pass a folder with text files in it. each line in the files becomes a
+        for filename in os.listdir(blacklistpath):
-        filter word.
+            with open(os.path.join(blacklistpath, filename), "r+") as listfile:
-
-        :param path: path to folder whose files shall be added to the set
-        :return: set of trigger words.
-        """
-        trigger_words = set()
-        for filename in os.listdir(path):
-            with open(os.path.join(path, filename), "r+") as listfile:
                 for word in listfile:
                     word = word.strip()
                     if word:
-                        trigger_words.add(word)
+                        self.blacklist.add(word)
-        return trigger_words
 
     def is_ok(self, message):
         """
@@ -46,13 +46,17 @@ class Trigger(object):
         :param message: A given string. Tweet or Toot, cleaned from html.
         :return: If the string passes the test
         """
-        ret = False
+        for pattern in self.goodlist:
+            if pattern.match(message):
+                break
+        else:
+            # no pattern matched
+            return False
+
         for word in message.lower().split():
-            if word in self.goodlist:
-                ret = True
             if word in self.blacklist:
                 return False
-        return ret
+        return True
 
 
 if __name__ == "__main__":