class namespace in /api/game/socket

1 files changed, 6 insertions, 4 deletions

diff --git a/api/game/socket.py b/api/game/socket.py
index 405d9cd..7931ada 100644
--- a/api/game/socket.py
+++ b/api/game/socket.py
@@ -1,10 +1,12 @@
 from flask import Blueprint, request, make_response
-from flask_socketio import SocketIO, emit, disconnect
+from flask_socketio import SocketIO, emit, disconnect, Namespace
 from main import socketio
 import time
 import json
 
-@socketio.on("connect", namespace="/game/socket")
-def connect():
-    print("connected")
+class GameSocketNamespace(Namespace):
+    def connect(self):
+        print("connection")
+
+socketio.on_namespace(GameSocketNamespace("/game/socket"))