From bad108d7ce7d59ba7c1ad8cc30b2e7c5b8d49022 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Tue, 10 Sep 2019 10:34:44 +0100
Subject: support disabling word tracking per channel, not just per user

---
 modules/words.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'modules')

diff --git a/modules/words.py b/modules/words.py
index 6fe8f6f4..1e07971c 100644
--- a/modules/words.py
+++ b/modules/words.py
@@ -9,11 +9,15 @@ WORD_DELIM = "\"'…~*`"
 WORD_START = WORD_DELIM+"“({<"
 WORD_STOP = WORD_DELIM+"”)}>;:.,!?"
 
-@utils.export("set", utils.BoolSetting(
-    "word-tracking", "Disable/enable tracking your wordcounts"))
+SETTING = utils.BoolSetting("word-tracking",
+    "Disable/enable tracking your wordcounts")
+
+@utils.export("set", SETTING)
+@utils.export("channelset", SETTING)
 class Module(ModuleManager.BaseModule):
     def _channel_message(self, user, event):
-        if not user.get_setting("word-tracking", True):
+        if not event["channel"].get_setting("word-tracking", True
+                ) or not user.get_setting("word-tracking", True):
             return
 
         words = list(filter(None, event["message_split"]))
-- 
cgit v1.3.1-10-gc9f91