python websocket服务器与ssl

Posted 2021-05-09

#!/usr/bin/env python3


import asyncio
import pathlib
import ssl

import websockets

async def hello(websocket, path):
    print(websocket, path)
    while True:
        name = await websocket.recv()
        print(type(name))
        print(f"< {name}")

        greeting = f"Hello {name}!"

        await websocket.send(greeting)
        print(f"> {greeting}")

ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
ssl_context.load_cert_chain('.ssl/tyio.net.crt', ".ssl/tyio.net.key")

start_server = websockets.serve(hello, '0.0.0.0', 4443, ssl=ssl_context)
start_server2 = websockets.serve(hello, '0.0.0.0', 4442, ssl=ssl_context)

loop = asyncio.get_event_loop()
loop.run_until_complete(start_server)
loop.run_until_complete(start_server2)
loop.run_forever()