Add websocket mode option in flow

custom_components/heatzy/__init__.py

@@ -34,3 +34,8 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
             coordinator.unsub()
 
     return unload_ok
+
+
+async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry):
+    """Reload if change option."""
+    await hass.config_entries.async_reload(entry.entry_id)

custom_components/heatzy/config_flow.py

@@ -13,9 +13,10 @@
 
 from homeassistant import config_entries
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.core import callback
 from homeassistant.helpers.aiohttp_client import async_create_clientsession
 
-from .const import DOMAIN
+from .const import CONF_WEBSOCKET, DOMAIN
 
 DATA_SCHEMA = vol.Schema(
     {vol.Required(CONF_USERNAME): str, vol.Required(CONF_PASSWORD): str}
@@ -29,6 +30,12 @@ class HeatzyFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
 
     VERSION = 1
 
+    @staticmethod
+    @callback
+    def async_get_options_flow(config_entry):
+        """Get option flow."""
+        return OptionsFlowHandler(config_entry)
+
     async def async_step_user(self, user_input=None):
         """Handle a flow initialized by the user."""
         errors = {}
@@ -56,3 +63,26 @@ async def async_step_user(self, user_input=None):
         return self.async_show_form(
             step_id="user", data_schema=DATA_SCHEMA, errors=errors
         )
+
+
+class OptionsFlowHandler(config_entries.OptionsFlow):
+    """Handle option."""
+
+    def __init__(self, config_entry):
+        """Initialize the options flow."""
+        self.config_entry = config_entry
+
+    async def async_step_init(self, user_input=None):
+        """Handle a flow initialized by the user."""
+        options_schema = vol.Schema(
+            {
+                vol.Required(
+                    CONF_WEBSOCKET,
+                    default=self.config_entry.options.get(CONF_WEBSOCKET, True),
+                ): bool
+            },
+        )
+        if user_input is not None:
+            return self.async_create_entry(title="", data=user_input)
+
+        return self.async_show_form(step_id="init", data_schema=options_schema)

custom_components/heatzy/const.py

@@ -23,6 +23,7 @@
 CONF_VERSION = "wifi_soft_version"
 CUR_TEMP_H = "cur_tempH"
 CUR_TEMP_L = "cur_tempL"
+CONF_WEBSOCKET = "websocket_feature"
 DEBOUNCE_COOLDOWN = 10
 DOMAIN = "heatzy"
 ECO_TEMP_H = "eco_tempH"

custom_components/heatzy/strings.json

@@ -18,5 +18,14 @@
     "abort": {
       "already_configured": "[%key:common::config_flow::abort::already_configured_service%]"
     }
+  },
+  "options": {
+    "step": {
+      "init": {
+        "data": {
+          "websocket_feature": "Enable websocket feature"
+        }
+      }
+    }
   }
 }

custom_components/heatzy/translations/en.json

@@ -18,5 +18,14 @@
     "abort": {
       "already_configured": "Your account is already configured."
     }
+  },
+  "options": {
+    "step": {
+      "init": {
+        "data": {
+          "websocket_feature": "Enable websocket feature"
+        }
+      }
+    }
   }
 }

custom_components/heatzy/translations/fr.json

@@ -18,5 +18,14 @@
     "abort": {
       "already_configured": "Votre compte est déjà enregistré."
     }
+  },
+  "options": {
+    "step": {
+      "init": {
+        "data": {
+          "websocket_feature": "Activer le mode websocket"
+        }
+      }
+    }
   }
 }