added tracking certain words in words.py and fixed/changed a couple of things to allow it to work.

1 files changed, 18 insertions, 2 deletions

diff --git a/modules/words.py b/modules/words.py
index c0c8169b..42c0b19a 100644
--- a/modules/words.py
+++ b/modules/words.py
@@ -3,19 +3,35 @@
 class Module(object):
     def __init__(self, bot):
         self.bot = bot
+        bot.events.on("new").on("server").hook(self.new_server)
         bot.events.on("received").on("message").on("channel"
             ).hook(self.channel_message)
         bot.events.on("received").on("command").on("words"
             ).hook(self.words, channel_only=True)
 
+    def new_server(self, event):
+        event["server"].tracked_words = set([])
+        settings = event["server"].find_settings("word-%")
+        for word, _ in settings:
+            word = word.split("word-", 1)[1]
+            event["server"].tracked_words.add(word)
+
     def channel_message(self, event):
-        word_count = len(list(filter(None, event["message_split"
-            ])))
+        words = list(filter(None, event["message_split"]))
+        word_count = len(words)
         words = event["user"].get_setting("words", {})
         if not event["channel"].name in words:
             words[event["channel"].name] = 0
         words[event["channel"].name] += word_count
         event["user"].set_setting("words", words)
+        for word in words:
+            if word.lower() in event["server"].tracked_words:
+                setting = "word-%s" % word
+                tracked_word = event["server"].get_setting(setting, {})
+                if not event["user"].name in tracked_word:
+                    tracked_word[event["user"].name] = 0
+                tracked_word[event["user"].name] += 1
+                event["server"].set_setting(setting, tracked_word)
 
     def words(self, event):
         if event["args_split"]: