Simplify connection closed exception.

diff --git a/controlpi_plugins/wsserver.py b/controlpi_plugins/wsserver.py
index fc7a3fe1f08b704eaf2428fcaa3f63d7fb3d6ac1..2a79b8c3260bfff766c87d394edd204c30357df5 100644 (file)
--- a/controlpi_plugins/wsserver.py
+++ b/controlpi_plugins/wsserver.py
@@ -10,8 +10,7 @@ TODO: Let clients filter messages received over websocket
 import os
 import sys
 import json
-from websockets import (WebSocketServerProtocol, ConnectionClosedOK,
-                        ConnectionClosedError, serve)
+from websockets import WebSocketServerProtocol, ConnectionClosed, serve
 from websockets.http import Headers
 from http import HTTPStatus
 
@@ -63,7 +62,7 @@ class Connection:
                 message = {'sender': self._name}
                 message.update(original_message)
                 await self._bus.send(message)
-        except (ConnectionClosedOK, ConnectionClosedError):
+        except ConnectionClosed:
             pass
         await self._bus.send({'sender': self._name,
                               'event': 'connection closed'})