Reconnect on connection failure.

syssi · Jul 17, 2017 · 399f8a7 · 399f8a7
1 parent b3bc762
commit 399f8a7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/xiaomiplug/device.py b/xiaomiplug/device.py
@@ -138,9 +138,10 @@ def send(self, command: str, parameters: Any=None) -> Any:
                                                             m.data.value["id"],
                                                             m.data.value))
             return m.data.value["result"]
-        except OSError as ex:
+        except Exception as ex:
             _LOGGER.error("got error when receiving: %s", ex)
-            raise DeviceException from ex
+            self.__enter__()
+            raise
 
     @property
     def _id(self) -> int: