feat: add ability to ping/pong with client

This commit is contained in:
2021-07-14 18:09:08 -04:00
parent 56b7f5f9cd
commit 87862931d2

15
main.py
View File

@@ -83,13 +83,16 @@ def create_game(game_id, client_id=None, player=None):
async def init(websocket, path): async def init(websocket, path):
try: try:
async for message in websocket: async for message in websocket:
data = json.loads(message) if message == "ping":
if data["action"] == "enter_game": await websocket.send("pong");
await enter_game(data["gameId"], data["clientId"], websocket)
elif data["action"] == "send_piece":
await send_piece(data["gameId"], data["clientId"], websocket, data["piece"])
else: else:
print("Unsupported action...") data = json.loads(message)
if data["action"] == "enter_game":
await enter_game(data["gameId"], data["clientId"], websocket)
elif data["action"] == "send_piece":
await send_piece(data["gameId"], data["clientId"], websocket, data["piece"])
else:
print("Unsupported action...")
finally: finally:
print("Exiting game...") print("Exiting game...")
await exit_game(websocket.remote_address[0]) await exit_game(websocket.remote_address[0])