websocket_handshake.py

def handshake(frame, client):
    headers = http_message.decode('utf8').split('\n')
    for header in headers:
        if header.startswith('Sec-WebSocket-Key'):
            key = header.replace('Sec-WebSocket-Key:', '').strip()
            accept_key = generate_accept_key(key)
            response = (
                "HTTP/1.1 101 Switching Protocols\r\n"
                "Upgrade: websocket\r\n"
                "Connection: Upgrade\r\n"
                f"Sec-WebSocket-Accept: {accept_key}\r\n\r\n"
            )
            client.send(response.encode('utf8'))