From 5b1e3c6fb15674248ecaddff24e2e485823a2549 Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Sat, 10 Mar 2018 14:01:43 +0000
Subject: [PATCH] Implement counting nickname changes and remove check on
 listener

---
 core/bot.py    | 4 ++++
 core/server.py | 9 +++------
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/core/bot.py b/core/bot.py
index b5f6acb..a704ae7 100644
--- a/core/bot.py
+++ b/core/bot.py
@@ -101,6 +101,10 @@ class IRCBot(IRCClient):
         newnick = nickname + "_"
         return newnick
 
+    def nickChanged(self, nick):
+        self.nickname = nick
+        count.event(self.name, "selfnick")
+
     def irc_ERR_NICKNAMEINUSE(self, prefix, params):
         self._attemptedNick = self.alterCollidedNick(self._attemptedNick)
         self.setNick(self._attemptedNick)
diff --git a/core/server.py b/core/server.py
index 71e9056..4aa81cf 100644
--- a/core/server.py
+++ b/core/server.py
@@ -34,13 +34,10 @@ class Server(Protocol):
     def connectionLost(self, reason):
         self.authed = False
         log("Connection lost from %s:%s -- %s" % (self.addr.host, self.addr.port, reason.getErrorMessage()))
-        if not listener == None:
-            if self.addr in connections.keys():
-                del connections[self.addr]
-            else:
-                warn("Tried to remove a non-existant connection.")
+        if self.addr in connections.keys():
+            del connections[self.addr]
         else:
-            warn("Tried to remove a connection from a listener that wasn't running.")
+            warn("Tried to remove a non-existant connection.")
         if self.addr in MonitorPool:
             MonitorPool.remove(self.addr)