From 83bc75c9cc82ecc4a3b5da44677cebbe9afcc748 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Thu, 6 Feb 2020 16:52:09 +0000
Subject: add a setting to only track words from registered users

---
 modules/words.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'modules/words.py')

diff --git a/modules/words.py b/modules/words.py
index b094a8ff..58022e2d 100644
--- a/modules/words.py
+++ b/modules/words.py
@@ -11,9 +11,13 @@ WORD_STOP = WORD_DELIM+"”)}>;:.,!?"
 
 SETTING = utils.BoolSetting("word-tracking",
     "Disable/enable tracking your wordcounts")
+REGISTERED_SETTING = utils.BoolSetting("word-tracking-registered",
+    "Whether or not word tracking is registered-users-only")
 
 @utils.export("set", SETTING)
 @utils.export("channelset", SETTING)
+@utils.export("serverset", REGISTERED_SETTING)
+@utils.export("channelset", REGISTERED_SETTING)
 @utils.export("channelset", utils.BoolSetting("words-prevent-highlight",
     "Whether or not to prevent highlights in wordiest lists"))
 class Module(ModuleManager.BaseModule):
@@ -60,6 +64,11 @@ class Module(ModuleManager.BaseModule):
                 ) or not user.get_setting("word-tracking", True):
             return
 
+        if event["channel"].get_setting("word-tracking-registered",
+                event["server"].get_setting("word-tracking-registered", False)):
+            if not self.exports.get_one("is-identified")(event["user"]):
+                return
+
         if user.get_setting("first-words", None) == None:
             user.set_setting("first-words", time.time())
 
-- 
cgit v1.3.1-10-gc9f91