Handle broken received data

main.py

@@ -11,7 +11,7 @@
 
 logging.basicConfig(level=logging.DEBUG)
 
-CLIENT_READY_TIMEOUT = 60.0
+CLIENT_READY_TIMEOUT = 10.0
 HOST = ('192.168.1.64', 8124)
 
 
@@ -36,7 +36,7 @@ def client_disconnected():
     client.on_connected = client_connected
     client.on_disconnected = client_disconnected
 
-    asyncio.create_task(client.async_connect())
+    task = asyncio.create_task(client.async_connect())
 
     try:
         async with async_timeout.timeout(CLIENT_READY_TIMEOUT):

pyflichub/button.py

@@ -1,7 +1,8 @@
-class FlicButton:
+class FlicButton(dict):
     def __init__(self, bdaddr: str, serial_number: str, color: str, name: str, active_disconnect: bool, connected: bool,
                  ready: bool, battery_status: int, uuid: str, flic_version: int, firmware_version: int, key: str,
                  passive_mode: bool) -> None:
+        super().__init__()
         self.bdaddr = bdaddr
         self.serial_number = serial_number
         self.color = color

pyflichub/client.py

@@ -45,11 +45,13 @@ def __init__(self, ip, port, loop, timeout=1.0, reconnect_timeout=10.0, event_ca
         self._data = None
         self.on_connected = None
         self.on_disconnected = None
+        self._connecting = False
 
     async def async_connect(self):
+        self._connecting = True
         """Connect to the socket."""
         try:
-            while True:
+            while self._connecting:
                 _LOGGER.info("Trying to connect to %s", self._server_address)
                 try:
                     await asyncio.wait_for(
@@ -72,6 +74,9 @@ async def async_connect(self):
             _LOGGER.debug("Connect attempt to %s cancelled", self._server_address)
 
     def disconnect(self):
+        _LOGGER.info("Disconnected")
+        self._connecting = False
+
         if self._transport is not None:
             self._transport.close()
 
@@ -112,11 +117,15 @@ def data_received(self, data):
         if decoded_data == 'pong':
             return
 
-        msg = json.loads(decoded_data)
-        if 'event' in msg:
-            self._handle_event(Event(**msg))
-        if 'command' in msg:
-            self._handle_command(Command(**msg))
+        try:
+            msg = json.loads(decoded_data)
+            if 'event' in msg:
+                self._handle_event(Event(**msg))
+            if 'command' in msg:
+                self._handle_command(Command(**msg))
+        except Exception:
+            _LOGGER.warning('Unable to decode received data')
+
 
     def connection_lost(self, exc):
         _LOGGER.info("Connection lost")