BAPSicle/websocket_server.py

import asyncio
import websockets
import json

baps_clients = set()
channel_to_q = None
server_name = None


async def websocket_handler(websocket, path):
    baps_clients.add(websocket)
    await websocket.send(json.dumps({"message": "Hello", "serverName": server_name}))
    print("New Client: {}".format(websocket))

    try:
        async for message in websocket:
            data = json.loads(message)
            channel = int(data["channel"])
            if "command" in data.keys():
                if data["command"] == "PLAY":
                    channel_to_q[channel].put("PLAY")
                elif data["command"] == "PAUSE":
                    channel_to_q[channel].put("PAUSE")
                elif data["command"] == "UNPAUSE":
                    channel_to_q[channel].put("UNPAUSE")
                elif data["command"] == "STOP":
                    channel_to_q[channel].put("STOP")
                elif data["command"] == "SEEK":
                    channel_to_q[channel].put("SEEK:" + str(data["time"]))
                elif data["command"] == "LOAD":
                    channel_to_q[channel].put("LOAD:" + str(data["weight"]))
                elif data["command"] == "ADD":
                    print(data)
                    new_item: Dict[str, any] = {
                        "channel_weight": int(data["newItem"]["weight"]),
                        "filename": "dev\\test.mp3",
                        "title":  data["newItem"]["title"],
                        "artist":  data["newItem"]["artist"] if "artist" in data["newItem"].keys() else None
                    }
                    channel_to_q[channel].put("ADD:" + json.dumps(new_item))
                elif data["command"] == "REMOVE":
                    channel_to_q[channel].put("REMOVE:" + str(data["weight"]))

            await asyncio.wait([conn.send(message) for conn in baps_clients])

    except websockets.exceptions.ConnectionClosedError:
        print("RIP {}".format(websocket))

    except Exception as e:
        print("Exception", e)

    finally:
        baps_clients.remove(websocket)


class WebsocketServer:

    def __init__(self, in_q, state):
        global channel_to_q
        channel_to_q = in_q

        global server_name
        server_name = state.state["server_name"]

        websocket_server = websockets.serve(websocket_handler, state.state["host"], state.state["ws_port"])

        asyncio.get_event_loop().run_until_complete(websocket_server)
        asyncio.get_event_loop().run_forever()


if __name__ == "__main__":
    print("Don't do this")