处理WebSocket模块异常和错误

Question

我正在尝试使用Python中的WebSocket模块创建一个简单的WebSocket服务器

我的任务是创建一个WebSocket服务器，将消息发送给所有正在侦听它的客户端。问题是，每次客户端断开连接时，WebSocket都会启动一个异常（ConnectionClosedOK或ConnectionClosedError，具体取决于发生的情况），我需要捕获该异常，因为我必须注销已断开连接的客户端，以便WebSocket停止在以太网上发送消息（并停止WebSocket服务器在每次尝试向关闭的客户端发送消息时抛出这些异常）

以下是服务器的代码：

async def foo_call():
    await asyncio.sleep(3)
    print('Hi!')


async def register(websocket):
    users.add(websocket)
    await notify_users()


async def unregister(websocket):
    users.remove(websocket)


async def start(websocket, path):
    await register(websocket)

    try:
        await foo_call()
    except websockets.ConnectionClosedError:
        await unregister(websocket)


# WebSocket clients that are listening this WebSocket server
users = set()

# Open WebSocket server
start_server = websockets.serve(start, "localhost", 8765)

# Run asyncio event loop
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

这是回溯：

Task exception was never retrieved
future: <Task finished coro=<WebSocketCommonProtocol.send() done, defined at /home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/protocol.py:521> exception=ConnectionClosedError('code = 1006 (connection closed abnormally [internal]), no reason',)>
Traceback (most recent call last):
  File "/home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/protocol.py", line 827, in transfer_data
    message = await self.read_message()
  File "/home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/protocol.py", line 895, in read_message
    frame = await self.read_data_frame(max_size=self.max_size)
  File "/home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/protocol.py", line 971, in read_data_frame
    frame = await self.read_frame(max_size)
  File "/home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/protocol.py", line 1051, in read_frame
    extensions=self.extensions,
  File "/home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/framing.py", line 105, in read
    data = await reader(2)
  File "/usr/lib/python3.6/asyncio/streams.py", line 672, in readexactly
    raise IncompleteReadError(incomplete, n)
asyncio.streams.IncompleteReadError: 0 bytes read on a total of 2 expected bytes

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/protocol.py", line 555, in send
    await self.ensure_open()
  File "/home/vlado/Projects/dahua/venv/lib/python3.6/site-packages/websockets/protocol.py", line 803, in ensure_open
    raise self.connection_closed_exc()
websockets.exceptions.ConnectionClosedError: code = 1006 (connection closed abnormally [internal]), no reason

我真的不明白如何捕捉websockets模块抛出的异常