wsproto_websocket_handshake_and_message_handling_quickstart.py

1from wsproto import WSConnection, ConnectionType
2from wsproto.events import Request, AcceptConnection, TextMessage
3
4# To establish a connection, we need a WSConnection object. 
5# We specify whether it's a client or server connection.
6conn = WSConnection(ConnectionType.SERVER)
7
8# To handle an incoming connection request:
9# (Assuming 'data' contains the raw bytes from a TCP connection)
10data = b"GET / HTTP/1.1\r\nHost: localhost\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nSec-WebSocket-Version: 13\r\n\r\n"
11conn.receive_data(data)
12
13for event in conn.events():
14    if isinstance(event, Request):
15        # We accept the connection
16        print("Received a connection request")
17        conn.send(AcceptConnection())
18
19# To send a message:
20conn.send(TextMessage(data="Hello, world!"))
21
22# To see the bytes that need to be sent over the network:
23network_data = conn.bytes_to_send()
24print(f"Bytes to send: {network_data}")
25
26# To handle an incoming message:
27# (Assuming 'message_data' contains raw bytes from the network)
28message_data = b"\x81\x05Hello"
29conn.receive_data(message_data)
30
31for event in conn.events():
32    if isinstance(event, TextMessage):
33        print(f"Received message: {event.data}")