From 2c524303aa747e36e309a78234b81472e0932780 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Fri, 10 Jan 2020 11:51:46 +0000
Subject: by default, prevent highlighting from !wordiest

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

(limited to 'modules')

diff --git a/modules/words.py b/modules/words.py
index 85e9ad99..716b7437 100644
--- a/modules/words.py
+++ b/modules/words.py
@@ -129,6 +129,12 @@ class Module(ModuleManager.BaseModule):
         else:
             event["stderr"].write("That word is not being tracked")
 
+    def _get_nickname(self, server, target, nickname):
+        nickname = server.get_user(nickname).nickname
+        if target.get_setting("wordiest-prevent-highlight", True):
+            nickname = utils.prevent_highlight(nickname)
+        return nickname
+
     @utils.hook("received.command.wordiest")
     def wordiest(self, event):
         """
@@ -155,7 +161,7 @@ class Module(ModuleManager.BaseModule):
                 user_words[nickname] += word_count
 
         top_10 = utils.top_10(user_words,
-            convert_key=lambda nickname:
-            event["server"].get_user(nickname).nickname)
+            convert_key=lambda nickname: self._get_nickname(
+            event["server"], event["target"], nickname))
         event["stdout"].write("wordiest%s: %s" % (
             word_prefix, ", ".join(top_10)))
-- 
cgit v1.3.1-10-gc9f91