Catch InvalidMessage exception.

diff --git a/controlpi_plugins/wsclient.py b/controlpi_plugins/wsclient.py
index 73e9f23ff5ff5cee471cf3c5a0f454c252d94526..f0a2263ba2b797e3dc6ce1badf9ce180c7d5e53b 100644 (file)
--- a/controlpi_plugins/wsclient.py
+++ b/controlpi_plugins/wsclient.py
@@ -3,7 +3,7 @@ import fcntl
 import json
 import socket
 import struct
-from websockets.exceptions import ConnectionClosed
+from websockets.exceptions import ConnectionClosed, InvalidMessage
 from websockets.legacy.client import connect, WebSocketClientProtocol
 
 from controlpi import BasePlugin, Message, MessageTemplate
@@ -196,6 +196,6 @@ class WSClient(BasePlugin):
                     await self.bus.send(Message(self.name,
                                                 {'event':
                                                  'connection closed'}))
-            except OSError:
+            except (OSError, InvalidMessage):
                 pass
             await asyncio.sleep(1)