From 167fe7b070305dbbbb548f7fff2e9de7b971b2a3 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Fri, 3 May 2019 16:34:41 +0100
Subject: Add !friends and !enemies to ducks.py

---
 modules/ducks.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

(limited to 'modules/ducks.py')

diff --git a/modules/ducks.py b/modules/ducks.py
index c17cf5be..79894d10 100644
--- a/modules/ducks.py
+++ b/modules/ducks.py
@@ -87,3 +87,36 @@ class Module(ModuleManager.BaseModule):
             self._no_duck(event["target"], event["user"], event["stderr"],
                 "shoot")
 
+    @utils.hook("received.command.friends")
+    def friends(self, event):
+        stats = self._duck_stats(event["server"], "ducks-befriended", "friends",
+            event["args_split"][0] if event["args"] else None)
+        event["stdout"].write(stats)
+    @utils.hook("received.command.enemies")
+    def enemies(self, event):
+        stats = self._duck_stats(event["server"], "ducks-shot", "enemies",
+            event["args_split"][0] if event["args"] else None)
+        event["stdout"].write(stats)
+
+    def _duck_stats(self, server, setting, description, channel_query):
+        channel_query_str = ""
+        if not channel_query == None:
+            channel_query = server.irc_lower(channel_query)
+            channel_query_str = " in %s" % channel_query
+
+        stats = server.find_all_user_channel_settings(setting)
+
+        user_stats = {}
+        for channel, nickname, value in stats:
+            if not channel_query or channel_query == channel:
+                if not nickname in user_stats:
+                    user_stats[nickname] = 0
+                user_stats[nickname] += value
+
+        nick_func = lambda nickname: utils.prevent_highlight(
+            server.get_user(nickname).nickname)
+
+        top_10 = utils.top_10(user_stats, convert_key=nick_func)
+        return "Top duck %s%s: %s" % (description, channel_query_str,
+            ", ".join(top_10))
+
-- 
cgit v1.3.1-10-gc9f91