python websocket server ws

python-websocket-server.py

# -*- coding: utf-8 -*-
from websocket_server import WebsocketServer
import logging

logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')

log = logging.getLogger('main')


def new_client(client, server):
    log.info('new client connected: %s, %s', client['id'], client['address'])


def client_left(client, server):
    log.info('client left: %s, %s', client['id'], client['address'])


def receive_message(client, server: WebsocketServer, message):
    log.info('receive client message: %s', message)
    server.send_message(client, 'MSG: xxx')


if __name__ == '__main__':
    server = WebsocketServer(29091, host='0.0.0.0')
    server.set_fn_new_client(new_client)
    server.set_fn_client_left(client_left)
    server.set_fn_message_received(receive_message)
    server.run_forever()