Add ability to create custom entities

custom_components/plum_ecomax/__init__.py

@@ -47,7 +47,6 @@
 
 PLATFORMS: list[Platform] = [
     Platform.BINARY_SENSOR,
-    Platform.BUTTON,
     Platform.CLIMATE,
     Platform.NUMBER,
     Platform.SELECT,
@@ -102,6 +101,16 @@ async def _async_close_connection(event: Event | None = None) -> None:
     return True
 
 
+async def async_reload_config(
+    hass: HomeAssistant, entry: ConfigEntry, connection: EcomaxConnection
+) -> None:
+    """Reload config on update."""
+    data = dict(entry.data)
+    data[CONF_SUB_DEVICES] = await async_get_sub_devices(connection.device)
+    hass.config_entries.async_update_entry(entry, data=data)
+    await hass.config_entries.async_reload(entry.entry_id)
+
+
 @callback
 def async_setup_events(hass: HomeAssistant, connection: EcomaxConnection) -> bool:
     """Set up the ecoMAX events."""

custom_components/plum_ecomax/climate.py

@@ -145,7 +145,7 @@ async def async_update_preset_mode(self, mode: int) -> None: ...
     @overload
     async def async_update_preset_mode(self, mode: ThermostatNumber) -> None: ...
 
-    async def async_update_preset_mode(self, mode: ThermostatNumber | int) -> None:
+    async def async_update_preset_mode(self, mode: Any) -> None:
         """Update preset mode."""
         if isinstance(mode, ThermostatNumber):
             mode = int(mode.value)