Retry and timeout can be change by setting a class attribute

rytilahti · Dec 6, 2020 · c7ec21c · c7ec21c
1 parent f481a9e
commit c7ec21c
Showing 1 changed file with 4 additions and 0 deletions.
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
         )