Abrupt loss of connection will now save the player data and destroy the old token.

Or at least, it will try to.

1 files changed, 7 insertions, 2 deletions

diff --git a/server.py b/server.py
index c665886..b3b1aa6 100755
--- a/server.py
+++ b/server.py
@@ -10,7 +10,7 @@ from websock import WebSocketServer, WebSocket
 ## Local Modules
 from utils import stdout, now, clients, debug
 from consts import MAX_CLIENTS, PACKET_ACK
-import protocol, security, traceback
+import protocol, security, player, traceback
 
 ###############################################################
 # Configuration
@@ -100,7 +100,12 @@ class WebSocketConn(WebSocket):
         #for client in clients:
         #    client.send_message(self.address[0] + u' - disconnected')
         stdout(self.address[0] + u' - disconnected')
-        # TODO: if self.token != "0": try: player.clear(self.token)
+        if self.token != "0":
+            try:
+                player.clear(self.token)
+            except:
+                traceback.print_exc()
+                stdout("Error at player.clear")
 
     ##########################
     # Useful functions: