import asyncio
import websockets

async def send_wow_periodically(websocket):
    try:
        while True:
            await websocket.send("wow")
            await asyncio.sleep(0.5)  # Wait for 500ms before sending the next message
    except websockets.exceptions.ConnectionClosed:
        print("Connection closed, stopping periodic messages.")

async def websocket_handler(websocket, path):
    task = asyncio.create_task(send_wow_periodically(websocket))

    try:
        async for message in websocket:
            if message == "hello":
                # If the message is "hello", send "wrong message" and cancel the periodic task
                await websocket.send("wrong message")
                print("Sent 'wrong message', closing connection.")
                task.cancel()  # Cancel the periodic "wow" messages
                try:
                    await task  # Optionally wait for the task cancellation to complete
                except asyncio.CancelledError:
                    pass
                await websocket.close(reason="Wrong message received")
                break
    except websockets.exceptions.ConnectionClosed as e:
        print(f"Connection closed with error: {e}")
        task.cancel()  # Ensure the task is cancelled if the connection is closed unexpectedly

async def main():
    # Define the server to listen on localhost:9001
    async with websockets.serve(websocket_handler, "0.0.0.0", 9001):
        print("WebSocket Server Started on ws://0.0.0.0:9001")
        await asyncio.Future()  # This keeps the server running indefinitely

# Run the server
asyncio.run(main())

++Sent decoded: fin=1 opcode=1 data=b'hello'
++Rcv raw: b'\x88\x18\x03\xe8Wrong message received'
++Rcv decoded: fin=1 opcode=8 data=b'\x03\xe8Wrong message received'
++Sent raw: b'\x88\x82\xd4#\xeb\xc8\xd7\xcb'
++Sent decoded: fin=1 opcode=8 data=b'\x03\xe8'
code 1000: Wrong message received

  if op_code == ABNF.OPCODE_CLOSE and not reconnect:
      return teardown(frame)

websocket._exceptions.WebSocketConnectionClosedException: Connection to remote host was lost.
Connection to remote host was lost. - reconnect
Calling dispatcher reconnect [5 frames in stack]
reconnect() - retrying in 5 seconds [6 frames in stack]

if op_code == ABNF.OPCODE_CLOSE and not reconnect:
    return teardown(frame)