Merge pull request #220 from smilechaser/feature-login_disconnect

Fix for #219
SpockBotMC · Mar 23, 2016 · 2cbebd1 · 2cbebd1
2 parents 2787cca + 92149bb
commit 2cbebd1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/spockbot/plugins/core/net.py b/spockbot/plugins/core/net.py
@@ -170,6 +170,7 @@ class NetPlugin(PluginBase):
         'SOCKET_ERR': 'handle_err',
         'SOCKET_HUP': 'handle_hup',
         'PLAY<Disconnect': 'handle_disconnect',
+        'LOGIN<Disconnect': 'handle_login_disconnect',
         'HANDSHAKE>Handshake': 'handle_handshake',
         'LOGIN<Login Success': 'handle_login_success',
         'LOGIN<Set Compression': 'handle_comp',
@@ -259,6 +260,13 @@ def handle_disconnect(self, name, packet):
         logger.debug("NETPLUGIN: Disconnected: %s", packet.data['reason'])
         self.event.emit('net_disconnect', packet.data['reason'])
 
+    def handle_login_disconnect(self, name, packet):
+
+        reason = packet.data.get('json_data', {}).get('text', '???')
+
+        logger.debug("NETPLUGIN: Disconnected: %s", reason)
+        self.event.emit('net_disconnect', reason)
+
     # Kill event - Try to shutdown the socket politely
     def handle_kill(self, name, data):
         if self.net.connected: