22 lines
535 B
Python
22 lines
535 B
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import asyncio
|
|
import websockets
|
|
|
|
async def echo(websocket, path):
|
|
async for message in websocket:
|
|
print(message)
|
|
await websocket.send(message)
|
|
|
|
if os.getenv('TEST_CLOSE'):
|
|
print('Closing')
|
|
# breakpoint()
|
|
await websocket.close(1001, 'close message')
|
|
# await websocket.close()
|
|
break
|
|
|
|
asyncio.get_event_loop().run_until_complete(
|
|
websockets.serve(echo, 'localhost', 5678))
|
|
asyncio.get_event_loop().run_forever()
|