include tenacity retrying

bbo76 · Dec 27, 2021 · c3733af · c3733af
1 parent 262e991
commit c3733af
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/custom_components/beewi_light/light.py b/custom_components/beewi_light/light.py
@@ -15,6 +15,8 @@
 import homeassistant.util.color as color_util
 
 from bulbeewipy  import BeewiSmartLight
+import tenacity
+
 _LOGGER = logging.getLogger(__name__)
 
 # Validation of the user's configuration
@@ -72,6 +74,7 @@ def rgbw_color(self):
     def is_on(self):
         return self._isOn
 
+    @tenacity.retry(stop=(tenacity.stop_after_delay(10) | tenacity.stop_after_attempt(5)))
     def turn_on(self, **kwargs):
         try:
             brightness = kwargs.get(ATTR_BRIGHTNESS)
@@ -103,13 +106,15 @@ def turn_on(self, **kwargs):
         except Exception as e:
             _LOGGER.error(e)
 
+    @tenacity.retry(stop=(tenacity.stop_after_delay(10) | tenacity.stop_after_attempt(5)))
     def turn_off(self, **kwargs):
         try:
             self._light.turnOff()
             self._isOn = False
         except Exception as e:
             _LOGGER.error(e)
 
+    @tenacity.retry(stop=(tenacity.stop_after_attempt(5)))
     def update(self):
         try:
             _LOGGER.debug("Trying get states")

diff --git a/custom_components/beewi_light/manifest.json b/custom_components/beewi_light/manifest.json
@@ -3,7 +3,7 @@
   "name": "Beewi Light",
   "documentation": "https://github.com/bbo76/light.beewi",
   "codeowners": ["@bbo76"],
-  "requirements": ["bulbeewipy==1.0.3"],
+  "requirements": ["bulbeewipy==1.0.4","tenacity"],
   "version": "2.0.0",
   "iot_class": "local_polling"
 }