Retry and timeout can be change by setting a class attribute

titilambert · titilambert · commit c7ec21c1bb08 · 2020-12-05T20:23:23.000-05:00
diff --git a/miio/device.py b/miio/device.py
@@ -117,6 +117,8 @@ def __init__(
     ) -> None:
         self.ip = ip
         self.token = token
+        if hasattr(self, "timeout"):
+            timeout = self.timeout
         self._protocol = MiIOProtocol(
             ip, token, start_id, debug, lazy_discover, timeout
         )
@@ -143,6 +145,8 @@ def send(
         :param int retry_count: How many times to retry on error
         :param dict extra_parameters: Extra top-level parameters
         """
+        if hasattr(self, "retry_count"):
+            retry_count = self.retry_count
         return self._protocol.send(
             command, parameters, retry_count, extra_parameters=extra_parameters
         )
