()
| 24 | |
| 25 | |
| 26 | async def main(): |
| 27 | uri = "ws://localhost:8000/chat" |
| 28 | async with websockets.connect(uri) as websocket: |
| 29 | print("connected successfully") |
| 30 | while True: |
| 31 | stdin = asyncio.create_task(stdin_to_ws(websocket)) |
| 32 | ws = asyncio.create_task(ws_next(websocket)) |
| 33 | done, _ = await asyncio.wait([stdin, ws], return_when=asyncio.FIRST_COMPLETED) |
| 34 | msg = done.pop() |
| 35 | if msg.result(): |
| 36 | print(msg.result()) |
| 37 | else: |
| 38 | break |
| 39 | |
| 40 | |
| 41 | if __name__ == "__main__": |
no test coverage detected