From 3444e91ec7c4f218720fb4d09195400ab23ccd10 Mon Sep 17 00:00:00 2001
From: Alberto Xamin <alberto@xamin.it>
Date: Tue, 24 Nov 2020 14:55:24 +0100
Subject: [PATCH] fix player count

---
 backend/__init__.py | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/backend/__init__.py b/backend/__init__.py
index a26b998..390e2f3 100644
--- a/backend/__init__.py
+++ b/backend/__init__.py
@@ -21,14 +21,14 @@ def advertise_lobbies():
 
 @sio.event
 def connect(sid, environ):
-    global online_players
     print('connect ', sid)
-    online_players += 1
     sio.enter_room(sid, 'lobby')
     sio.emit('players', room='lobby', data=online_players)
 
 @sio.event
 def set_username(sid, username):
+    global online_players
+    online_players += 1
     sio.save_session(sid, Player(username, sid, sio))
     print(f'{sid} is now {username}')
     advertise_lobbies()
@@ -40,13 +40,14 @@ def my_message(sid, data):
 @sio.event
 def disconnect(sid):
     global online_players
-    online_players -= 1
-    sio.emit('players', room='lobby', data=online_players)
-    if sio.get_session(sid).game and sio.get_session(sid).disconnect():
-        sio.close_room(sio.get_session(sid).game.name)
-        games.pop(games.index(sio.get_session(sid).game))
-    print('disconnect ', sid)
-    advertise_lobbies()
+    if sio.get_session(sid):
+        online_players -= 1
+        sio.emit('players', room='lobby', data=online_players)
+        if sio.get_session(sid).game and sio.get_session(sid).disconnect():
+            sio.close_room(sio.get_session(sid).game.name)
+            games.pop(games.index(sio.get_session(sid).game))
+        print('disconnect ', sid)
+        advertise_lobbies()
 
 @sio.event
 def create_room(sid, room_name):