Diff to HTML by rtfpessoa

Files changed (17) hide show

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/__init__.py +22 -61
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/binary_sensor.py +49 -81
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/button.py +13 -55
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/climate.py +62 -122
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/config_flow.py +62 -182
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/const.py +29 -156
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/coordinator.py +38 -72
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/diagnostics.py +1 -0
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/entity.py +14 -25
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/icons.json +0 -3
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/manifest.json +5 -6
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/number.py +23 -40
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/select.py +23 -52
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/sensor.py +101 -167
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/strings.json +49 -63
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/switch.py +24 -41
/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/util.py +3 -8

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/__init__.py RENAMED Viewed

@@ -1,47 +1,29 @@
 """Plugwise platform for Home Assistant Core."""
 from __future__ import annotations
 from typing import Any
-from plugwise.exceptions import PlugwiseError
-import voluptuous as vol
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
-from homeassistant.core import HomeAssistant, ServiceCall, callback
 from homeassistant.helpers import device_registry as dr, entity_registry as er
-from .const import (
-    CONF_REFRESH_INTERVAL,  # pw-beta options
-    DOMAIN,
-    LOGGER,
-    PLATFORMS,
-    SERVICE_DELETE,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
 type PlugwiseConfigEntry = ConfigEntry[PlugwiseDataUpdateCoordinator]
 async def async_setup_entry(hass: HomeAssistant, entry: PlugwiseConfigEntry) -> bool:
-    """Set up the Plugwise Device from a config entry."""
     await er.async_migrate_entries(hass, entry.entry_id, async_migrate_entity_entry)
-    cooldown = 1.5  # pw-beta frontend refresh-interval
-    if (
-        custom_refresh := entry.options.get(CONF_REFRESH_INTERVAL)
-    ) is not None:  # pragma: no cover
-        cooldown = custom_refresh
-    LOGGER.debug("DUC cooldown interval: %s", cooldown)
-    coordinator = PlugwiseDataUpdateCoordinator(
-        hass, cooldown
-    )  # pw-beta - cooldown, update_interval as extra
     await coordinator.async_config_entry_first_refresh()
-    # Migrate a changed sensor unique_id
     migrate_sensor_entities(hass, coordinator)
-    entry.runtime_data = coordinator  # pw-beta
     device_registry = dr.async_get(hass)
     device_registry.async_get_or_create(
@@ -49,48 +31,21 @@
         identifiers={(DOMAIN, str(coordinator.api.gateway_id))},
         manufacturer="Plugwise",
         model=coordinator.api.smile_model,
         name=coordinator.api.smile_name,
-        sw_version=coordinator.api.smile_version,
-    )
-    async def delete_notification(
-        call: ServiceCall,
-    ) -> None:  # pragma: no cover  # pw-beta: HA service - delete_notification
-        """Service: delete the Plugwise Notification."""
-        LOGGER.debug(
-            "Service delete PW Notification called for %s",
-            coordinator.api.smile_name,
-        )
-        try:
-            await coordinator.api.delete_notification()
-            LOGGER.debug("PW Notification deleted")
-        except PlugwiseError:
-            LOGGER.debug(
-                "Failed to delete the Plugwise Notification for %s",
-                coordinator.api.smile_name,
-            )
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
-    entry.async_on_unload(entry.add_update_listener(update_listener))
-    for component in PLATFORMS:  # pw-beta
-        if component == Platform.BINARY_SENSOR:
-            hass.services.async_register(
-                DOMAIN, SERVICE_DELETE, delete_notification, schema=vol.Schema({})
-            )
     return True
-async def update_listener(
-    hass: HomeAssistant, entry: PlugwiseConfigEntry
-) -> None:  # pragma: no cover  # pw-beta
-    """Handle options update."""
-    await hass.config_entries.async_reload(entry.entry_id)
 async def async_unload_entry(hass: HomeAssistant, entry: PlugwiseConfigEntry) -> bool:
-    """Unload a config entry."""
     return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
 @callback
 def async_migrate_entity_entry(entry: er.RegistryEntry) -> dict[str, Any] | None:
     """Migrate Plugwise entity entries.
@@ -109,15 +64,15 @@
         "-relative_humidity"
     ):
         return {
-            "new_unique_id": entry.unique_id.replace(
-                "-relative_humidity", "-humidity"
-            )
         }
     if entry.domain == Platform.SWITCH and entry.unique_id.endswith("-plug"):
         return {"new_unique_id": entry.unique_id.replace("-plug", "-relay")}
     # No migration needed
     return None
 def migrate_sensor_entities(
     hass: HomeAssistant,
     coordinator: PlugwiseDataUpdateCoordinator,
@@ -125,10 +80,10 @@
     """Migrate Sensors if needed."""
     ent_reg = er.async_get(hass)
-    # Migrate opentherm_outdoor_temperature  # pw-beta add to Core
     # to opentherm_outdoor_air_temperature sensor
     for device_id, device in coordinator.data.devices.items():
-        if device["dev_class"] != "heater_central":  # pw-beta add to Core
             continue
         old_unique_id = f"{device_id}-outdoor_temperature"
@@ -136,4 +91,10 @@
             Platform.SENSOR, DOMAIN, old_unique_id
         ):
             new_unique_id = f"{device_id}-outdoor_air_temperature"
             ent_reg.async_update_entity(entity_id, new_unique_id=new_unique_id)

 """Plugwise platform for Home Assistant Core."""
 from __future__ import annotations
 from typing import Any
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers import device_registry as dr, entity_registry as er
+from .const import DOMAIN, LOGGER, PLATFORMS
 from .coordinator import PlugwiseDataUpdateCoordinator
 type PlugwiseConfigEntry = ConfigEntry[PlugwiseDataUpdateCoordinator]
 async def async_setup_entry(hass: HomeAssistant, entry: PlugwiseConfigEntry) -> bool:
+    """Set up Plugwise components from a config entry."""
     await er.async_migrate_entries(hass, entry.entry_id, async_migrate_entity_entry)
+    coordinator = PlugwiseDataUpdateCoordinator(hass)
     await coordinator.async_config_entry_first_refresh()
     migrate_sensor_entities(hass, coordinator)
+    entry.runtime_data = coordinator
     device_registry = dr.async_get(hass)
     device_registry.async_get_or_create(
         identifiers={(DOMAIN, str(coordinator.api.gateway_id))},
         manufacturer="Plugwise",
         model=coordinator.api.smile_model,
+        model_id=coordinator.api.smile_model_id,
         name=coordinator.api.smile_name,
+        sw_version=str(coordinator.api.smile_version),
+    )  # required for adding the entity-less P1 Gateway
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
     return True
 async def async_unload_entry(hass: HomeAssistant, entry: PlugwiseConfigEntry) -> bool:
+    """Unload the Plugwise components."""
     return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
 @callback
 def async_migrate_entity_entry(entry: er.RegistryEntry) -> dict[str, Any] | None:
     """Migrate Plugwise entity entries.
         "-relative_humidity"
     ):
         return {
+            "new_unique_id": entry.unique_id.replace("-relative_humidity", "-humidity")
         }
     if entry.domain == Platform.SWITCH and entry.unique_id.endswith("-plug"):
         return {"new_unique_id": entry.unique_id.replace("-plug", "-relay")}
     # No migration needed
     return None
 def migrate_sensor_entities(
     hass: HomeAssistant,
     coordinator: PlugwiseDataUpdateCoordinator,
     """Migrate Sensors if needed."""
     ent_reg = er.async_get(hass)
+    # Migrating opentherm_outdoor_temperature
     # to opentherm_outdoor_air_temperature sensor
     for device_id, device in coordinator.data.devices.items():
+        if device.get("dev_class") != "heater_central":
             continue
         old_unique_id = f"{device_id}-outdoor_temperature"
             Platform.SENSOR, DOMAIN, old_unique_id
         ):
             new_unique_id = f"{device_id}-outdoor_air_temperature"
+            LOGGER.debug(
+                "Migrating entity %s from old unique ID '%s' to new unique ID '%s'",
+                entity_id,
+                old_unique_id,
+                new_unique_id,
+            )
             ent_reg.async_update_entity(entity_id, new_unique_id=new_unique_id)

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/binary_sensor.py RENAMED Viewed

@@ -1,4 +1,5 @@
 """Plugwise Binary Sensor component for Home Assistant."""
 from __future__ import annotations
 from collections.abc import Mapping
@@ -8,33 +9,19 @@
 from plugwise.constants import BinarySensorType
 from homeassistant.components.binary_sensor import (
     BinarySensorEntity,
     BinarySensorEntityDescription,
 )
-from homeassistant.const import ATTR_NAME, EntityCategory
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
-from .const import (
-    BINARY_SENSORS,
-    COMPRESSOR_STATE,
-    COOLING_ENABLED,
-    COOLING_STATE,
-    DHW_STATE,
-    DOMAIN,
-    FLAME_STATE,
-    HEATING_STATE,
-    LOGGER,
-    NOTIFICATIONS,
-    PLUGWISE_NOTIFICATION,
-    SECONDARY_BOILER_STATE,
-    SEVERITIES,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
-PARALLEL_UPDATES = 0
 @dataclass(frozen=True)
@@ -44,45 +31,52 @@
     key: BinarySensorType
-PLUGWISE_BINARY_SENSORS: tuple[PlugwiseBinarySensorEntityDescription, ...] = (
     PlugwiseBinarySensorEntityDescription(
-        key=COMPRESSOR_STATE,
-        translation_key=COMPRESSOR_STATE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
-        key=COOLING_ENABLED,
-        translation_key=COOLING_ENABLED,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
-        key=DHW_STATE,
-        translation_key=DHW_STATE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
-        key=FLAME_STATE,
-        translation_key=FLAME_STATE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
-        key=HEATING_STATE,
-        translation_key=HEATING_STATE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
-        key=COOLING_STATE,
-        translation_key=COOLING_STATE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
-        key=SECONDARY_BOILER_STATE,
-        translation_key=SECONDARY_BOILER_STATE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
-        key=PLUGWISE_NOTIFICATION,
-        translation_key=PLUGWISE_NOTIFICATION,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
 )
@@ -93,34 +87,29 @@
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
-    """Set up the Plugwise binary_sensors from a ConfigEntry."""
     coordinator = entry.runtime_data
     @callback
-    def _async_add_new_device(device_id: str) -> None:
-        """Add new detected device during runtime."""
-        device = coordinator.data.devices[device_id]
-        if not (binary_sensors := device.get(BINARY_SENSORS)):
             return
-        entities: list[PlugwiseBinarySensorEntity] = []
-        for description in PLUGWISE_BINARY_SENSORS:
-            if description.key not in binary_sensors:
-                continue
-            entities.append(
-                PlugwiseBinarySensorEntity(coordinator, device_id, description)
-            )
-            LOGGER.debug(
-                "Add %s %s binary sensor", device[ATTR_NAME], description.translation_key
             )
-        async_add_entities(entities)
-    if coordinator.new_devices:
-        for device_id in coordinator.new_devices:
-            _async_add_new_device(device_id)
-    else:
-        coordinator.new_devices_callbacks.append(_async_add_new_device)
 class PlugwiseBinarySensorEntity(PlugwiseEntity, BinarySensorEntity):
@@ -138,46 +127,25 @@
         super().__init__(coordinator, device_id)
         self.entity_description = description
         self._attr_unique_id = f"{device_id}-{description.key}"
-        self._notification: dict[str, str] = {}  # pw-beta
     @property
     def is_on(self) -> bool:
         """Return true if the binary sensor is on."""
-        # pw-beta: show Plugwise notifications as HA persistent notifications
-        if self._notification:
-            for notify_id, message in self._notification.items():
-                self.hass.components.persistent_notification.async_create(
-                    message, "Plugwise Notification:", f"{DOMAIN}.{notify_id}"
-                )
-        # return self.device["binary_sensors"][self.entity_description.key]  # type: ignore [literal-required]
         return self.device["binary_sensors"][self.entity_description.key]
     @property
     def extra_state_attributes(self) -> Mapping[str, Any] | None:
         """Return entity specific state attributes."""
-        if self.entity_description.key != PLUGWISE_NOTIFICATION:
             return None
-        # pw-beta adjustment with attrs is to only represent severities *with* content
-        # not all severities including those without content as empty lists
-        attrs: dict[str, list[str]] = {}  # pw-beta Re-evaluate against Core
-        self._notification = {}  # pw-beta
-        if notify := self.coordinator.data.gateway[NOTIFICATIONS]:
-            for notify_id, details in notify.items():  # pw-beta uses notify_id
                 for msg_type, msg in details.items():
                     msg_type = msg_type.lower()
                     if msg_type not in SEVERITIES:
-                        msg_type = "other"  # pragma: no cover
-                    if (
-                        f"{msg_type}_msg" not in attrs
-                    ):  # pw-beta Re-evaluate against Core
-                        attrs[f"{msg_type}_msg"] = []
                     attrs[f"{msg_type}_msg"].append(msg)
-                    self._notification[
-                        notify_id
-                    ] = f"{msg_type.title()}: {msg}"  # pw-beta
         return attrs

 """Plugwise Binary Sensor component for Home Assistant."""
 from __future__ import annotations
 from collections.abc import Mapping
 from plugwise.constants import BinarySensorType
 from homeassistant.components.binary_sensor import (
+    BinarySensorDeviceClass,
     BinarySensorEntity,
     BinarySensorEntityDescription,
 )
+from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
+SEVERITIES = ["other", "info", "warning", "error"]
 @dataclass(frozen=True)
     key: BinarySensorType
+BINARY_SENSORS: tuple[PlugwiseBinarySensorEntityDescription, ...] = (
+    PlugwiseBinarySensorEntityDescription(
+        key="low_battery",
+        translation_key="low_battery",
+        device_class=BinarySensorDeviceClass.BATTERY,
+        entity_category=EntityCategory.DIAGNOSTIC,
+    ),
     PlugwiseBinarySensorEntityDescription(
+        key="compressor_state",
+        translation_key="compressor_state",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
+        key="cooling_enabled",
+        translation_key="cooling_enabled",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
+        key="dhw_state",
+        translation_key="dhw_state",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
+        key="flame_state",
+        translation_key="flame_state",
+        name="Flame state",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
+        key="heating_state",
+        translation_key="heating_state",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
+        key="cooling_state",
+        translation_key="cooling_state",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
+        key="secondary_boiler_state",
+        translation_key="secondary_boiler_state",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseBinarySensorEntityDescription(
+        key="plugwise_notification",
+        translation_key="plugwise_notification",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
 )
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
+    """Set up the Smile binary_sensors from a config entry."""
     coordinator = entry.runtime_data
     @callback
+    def _add_entities() -> None:
+        """Add Entities."""
+        if not coordinator.new_devices:
             return
+        async_add_entities(
+            PlugwiseBinarySensorEntity(coordinator, device_id, description)
+            for device_id in coordinator.new_devices
+            if (
+                binary_sensors := coordinator.data.devices[device_id].get(
+                    "binary_sensors"
+                )
             )
+            for description in BINARY_SENSORS
+            if description.key in binary_sensors
+        )
+    _add_entities()
+    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
 class PlugwiseBinarySensorEntity(PlugwiseEntity, BinarySensorEntity):
         super().__init__(coordinator, device_id)
         self.entity_description = description
         self._attr_unique_id = f"{device_id}-{description.key}"
     @property
     def is_on(self) -> bool:
         """Return true if the binary sensor is on."""
         return self.device["binary_sensors"][self.entity_description.key]
     @property
     def extra_state_attributes(self) -> Mapping[str, Any] | None:
         """Return entity specific state attributes."""
+        if self.entity_description.key != "plugwise_notification":
             return None
+        attrs: dict[str, list[str]] = {f"{severity}_msg": [] for severity in SEVERITIES}
+        if notify := self.coordinator.data.gateway["notifications"]:
+            for details in notify.values():
                 for msg_type, msg in details.items():
                     msg_type = msg_type.lower()
                     if msg_type not in SEVERITIES:
+                        msg_type = "other"
                     attrs[f"{msg_type}_msg"].append(msg)
         return attrs

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/button.py RENAMED Viewed

@@ -1,30 +1,18 @@
 """Plugwise Button component for Home Assistant."""
 from __future__ import annotations
-from homeassistant.components.button import (
-    ButtonDeviceClass,
-    ButtonEntity,
-    ButtonEntityDescription,
-)
-from homeassistant.const import ATTR_NAME, EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
-from .const import GATEWAY_ID, LOGGER, REBOOT
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
-BUTTON_TYPES: tuple[ButtonEntityDescription, ...] = (
-    ButtonEntityDescription(
-        key=REBOOT,
-        translation_key=REBOOT,
-        device_class=ButtonDeviceClass.RESTART,
-        entity_category=EntityCategory.CONFIG,
-    ),
-)
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -34,59 +22,29 @@
     """Set up the Plugwise buttons from a ConfigEntry."""
     coordinator = entry.runtime_data
-    entities: list[PlugwiseButtonEntity] = []
     gateway = coordinator.data.gateway
-    for device_id, device in coordinator.data.devices.items():
-        if device_id == gateway[GATEWAY_ID] and REBOOT in gateway:
-            for description in BUTTON_TYPES:
-                entities.append(
-                    PlugwiseButtonEntity(
-                        coordinator,
-                        device_id,
-                        description,
-                    )
-                )
-                LOGGER.debug(
-                    "Add %s %s button", device[ATTR_NAME], description.key
-            )
-    async_add_entities(entities)
-    def _async_add_new_device(device_id: str) -> None:
-        """Add new detected device during runtime."""
-        device = coordinator.data.devices[device_id]
-        entities: list[PlugwiseButtonEntity] = []
-        gateway = coordinator.data.gateway
-        if device_id == gateway[GATEWAY_ID] and REBOOT in gateway:
-            for description in BUTTON_TYPES:
-                entities.append(
-                    PlugwiseButtonEntity(coordinator, device_id, description)
-                )
-                LOGGER.debug(
-                    "Add %s %s button", device[ATTR_NAME], description.translation_key
-                )
-        async_add_entities(entities)
-    coordinator.new_devices_callbacks.append(_async_add_new_device)
 class PlugwiseButtonEntity(PlugwiseEntity, ButtonEntity):
     """Defines a Plugwise button."""
-    entity_description: ButtonEntityDescription
     def __init__(
         self,
         coordinator: PlugwiseDataUpdateCoordinator,
         device_id: str,
-        description: ButtonEntityDescription,
     ) -> None:
         """Initialize the button."""
         super().__init__(coordinator, device_id)
-        self.device_id = device_id
-        self.entity_description = description
-        self._attr_unique_id = f"{device_id}-{description.key}"
     @plugwise_command
     async def async_press(self) -> None:

 """Plugwise Button component for Home Assistant."""
 from __future__ import annotations
+from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
+from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
+from .const import GATEWAY_ID, REBOOT
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
 async def async_setup_entry(
     hass: HomeAssistant,
     """Set up the Plugwise buttons from a ConfigEntry."""
     coordinator = entry.runtime_data
     gateway = coordinator.data.gateway
+    async_add_entities(
+        PlugwiseButtonEntity(coordinator, device_id)
+        for device_id in coordinator.data.devices
+        if device_id == gateway[GATEWAY_ID] and REBOOT in gateway
+    )
 class PlugwiseButtonEntity(PlugwiseEntity, ButtonEntity):
     """Defines a Plugwise button."""
+    _attr_device_class = ButtonDeviceClass.RESTART
+    _attr_entity_category = EntityCategory.CONFIG
     def __init__(
         self,
         coordinator: PlugwiseDataUpdateCoordinator,
         device_id: str,
     ) -> None:
         """Initialize the button."""
         super().__init__(coordinator, device_id)
+        self._attr_translation_key = REBOOT
+        self._attr_unique_id = f"{device_id}-reboot"
     @plugwise_command
     async def async_press(self) -> None:

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/climate.py RENAMED Viewed

@@ -1,52 +1,25 @@
 """Plugwise Climate component for Home Assistant."""
 from __future__ import annotations
 from typing import Any
-from homeassistant.components.climate import ClimateEntity
-from homeassistant.components.climate.const import (
     ATTR_HVAC_MODE,
     ATTR_TARGET_TEMP_HIGH,
     ATTR_TARGET_TEMP_LOW,
-    PRESET_AWAY,  # pw-beta homekit emulation
-    PRESET_HOME,  # pw-beta homekit emulation
     ClimateEntityFeature,
     HVACAction,
     HVACMode,
 )
-from homeassistant.const import ATTR_TEMPERATURE, STATE_OFF, STATE_ON, UnitOfTemperature
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.exceptions import HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
-from .const import (
-    ACTIVE_PRESET,
-    AVAILABLE_SCHEDULES,
-    BINARY_SENSORS,
-    CONF_HOMEKIT_EMULATION,  # pw-beta homekit emulation
-    CONTROL_STATE,
-    COOLING_PRESENT,
-    COOLING_STATE,
-    DEV_CLASS,
-    DOMAIN,
-    GATEWAY_ID,
-    HEATING_STATE,
-    LOCATION,
-    LOWER_BOUND,
-    MASTER_THERMOSTATS,
-    MODE,
-    REGULATION_MODES,
-    RESOLUTION,
-    SELECT_REGULATION_MODE,
-    SENSORS,
-    SMILE_NAME,
-    TARGET_TEMP,
-    TARGET_TEMP_HIGH,
-    TARGET_TEMP_LOW,
-    THERMOSTAT,
-    UPPER_BOUND,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
@@ -57,11 +30,8 @@
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
-    """Set up the Smile Thermostats from a ConfigEntry."""
     coordinator = entry.runtime_data
-    homekit_enabled: bool = entry.options.get(
-        CONF_HOMEKIT_EMULATION, False
-    )  # pw-beta homekit emulation
     @callback
     def _add_entities() -> None:
@@ -70,16 +40,13 @@
             return
         async_add_entities(
-            PlugwiseClimateEntity(
-                coordinator, device_id, homekit_enabled
-            )  # pw-beta homekit emulation
-            for device_id, device in coordinator.data.devices.items()
-            if device[DEV_CLASS] in MASTER_THERMOSTATS
         )
-    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
     _add_entities()
 class PlugwiseClimateEntity(PlugwiseEntity, ClimateEntity):
@@ -91,36 +58,25 @@
     _attr_translation_key = DOMAIN
     _enable_turn_on_off_backwards_compatibility = False
-    _homekit_mode: str | None = None  # pw-beta homekit emulation
-    _previous_mode: str = HVACAction.HEATING
     def __init__(
         self,
         coordinator: PlugwiseDataUpdateCoordinator,
         device_id: str,
-        homekit_enabled: bool,  # pw-beta homekit emulation
     ) -> None:
         """Set up the Plugwise API."""
         super().__init__(coordinator, device_id)
-        self._homekit_enabled = homekit_enabled  # pw-beta homekit emulation
-        gateway_id: str = coordinator.data.gateway[GATEWAY_ID]
-        self.gateway_data = coordinator.data.devices[gateway_id]
-        self._attr_max_temp = min(self.device[THERMOSTAT][UPPER_BOUND], 35.0)
-        self._attr_min_temp = self.device[THERMOSTAT][LOWER_BOUND]
-        # Ensure we don't drop below 0.1
-        self._attr_target_temperature_step = max(
-            self.device[THERMOSTAT][RESOLUTION], 0.1
-        )
         self._attr_unique_id = f"{device_id}-climate"
-        # Determine supported features
         self.cdr_gateway = coordinator.data.gateway
         self._attr_supported_features = ClimateEntityFeature.TARGET_TEMPERATURE
         if (
-            self.cdr_gateway[COOLING_PRESENT]
-            and self.cdr_gateway[SMILE_NAME] != "Adam"
         ):
             self._attr_supported_features = (
                 ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
@@ -129,9 +85,16 @@
             self._attr_supported_features |= (
                 ClimateEntityFeature.TURN_OFF | ClimateEntityFeature.TURN_ON
             )
-        if presets := self.device["preset_modes"]:  # can be NONE
             self._attr_supported_features |= ClimateEntityFeature.PRESET_MODE
-        self._attr_preset_modes = presets
     def _previous_action_mode(self, coordinator: PlugwiseDataUpdateCoordinator) -> None:
         """Return the previous action-mode when the regulation-mode is not heating or cooling.
@@ -140,17 +103,17 @@
         """
         # When no cooling available, _previous_mode is always heating
         if (
-            REGULATION_MODES in self.gateway_data
-            and HVACAction.COOLING in self.gateway_data[REGULATION_MODES]
         ):
-            mode = self.gateway_data[SELECT_REGULATION_MODE]
-            if mode in (HVACAction.COOLING, HVACAction.HEATING):
                 self._previous_mode = mode
     @property
     def current_temperature(self) -> float:
         """Return the current temperature."""
-        return self.device[SENSORS][ATTR_TEMPERATURE]
     @property
     def target_temperature(self) -> float:
@@ -159,7 +122,7 @@
         Connected to the HVACMode combination of AUTO-HEAT.
         """
-        return self.device[THERMOSTAT][TARGET_TEMP]
     @property
     def target_temperature_high(self) -> float:
@@ -167,7 +130,7 @@
         Connected to the HVACMode combination of AUTO-HEAT_COOL.
         """
-        return self.device[THERMOSTAT][TARGET_TEMP_HIGH]
     @property
     def target_temperature_low(self) -> float:
@@ -175,39 +138,30 @@
         Connected to the HVACMode combination AUTO-HEAT_COOL.
         """
-        return self.device[THERMOSTAT][TARGET_TEMP_LOW]
     @property
     def hvac_mode(self) -> HVACMode:
         """Return HVAC operation ie. auto, cool, heat, heat_cool, or off mode."""
-        if (
-            mode := self.device[MODE]
-        ) is None or mode not in self.hvac_modes:  # pw-beta add to Core
-            return HVACMode.HEAT  # pragma: no cover
-        # pw-beta homekit emulation
-        if self._homekit_enabled and self._homekit_mode == HVACMode.OFF:
-            mode = HVACMode.OFF  # pragma: no cover
         return HVACMode(mode)
     @property
     def hvac_modes(self) -> list[HVACMode]:
         """Return a list of available HVACModes."""
         hvac_modes: list[HVACMode] = []
-        if (
-            self._homekit_enabled  # pw-beta homekit emulation
-            or REGULATION_MODES in self.gateway_data
-        ):
             hvac_modes.append(HVACMode.OFF)
-        if AVAILABLE_SCHEDULES in self.device:
             hvac_modes.append(HVACMode.AUTO)
-        if self.cdr_gateway[COOLING_PRESENT]:
-            if REGULATION_MODES in self.gateway_data:
-                if self.gateway_data[SELECT_REGULATION_MODE] == HVACAction.COOLING:
                     hvac_modes.append(HVACMode.COOL)
-                if self.gateway_data[SELECT_REGULATION_MODE] == HVACAction.HEATING:
                     hvac_modes.append(HVACMode.HEAT)
             else:
                 hvac_modes.append(HVACMode.HEAT_COOL)
@@ -217,27 +171,26 @@
         return hvac_modes
     @property
-    def hvac_action(self) -> HVACAction:  # pw-beta add to Core
         """Return the current running hvac operation if supported."""
         # Keep track of the previous action-mode
         self._previous_action_mode(self.coordinator)
         # Adam provides the hvac_action for each thermostat
-        if (control_state := self.device.get(CONTROL_STATE)) == HVACAction.COOLING:
             return HVACAction.COOLING
-        if control_state == HVACAction.HEATING:
             return HVACAction.HEATING
-        if control_state == HVACAction.PREHEATING:
             return HVACAction.PREHEATING
-        if control_state == STATE_OFF:
             return HVACAction.IDLE
-        # Anna
         heater: str = self.coordinator.data.gateway["heater_id"]
         heater_data = self.coordinator.data.devices[heater]
-        if heater_data[BINARY_SENSORS][HEATING_STATE]:
             return HVACAction.HEATING
-        if heater_data[BINARY_SENSORS].get(COOLING_STATE, False):
             return HVACAction.COOLING
         return HVACAction.IDLE
@@ -245,18 +198,18 @@
     @property
     def preset_mode(self) -> str | None:
         """Return the current preset mode."""
-        return self.device[ACTIVE_PRESET]
     @plugwise_command
     async def async_set_temperature(self, **kwargs: Any) -> None:
         """Set new target temperature."""
         data: dict[str, Any] = {}
         if ATTR_TEMPERATURE in kwargs:
-            data[TARGET_TEMP] = kwargs.get(ATTR_TEMPERATURE)
         if ATTR_TARGET_TEMP_HIGH in kwargs:
-            data[TARGET_TEMP_HIGH] = kwargs.get(ATTR_TARGET_TEMP_HIGH)
         if ATTR_TARGET_TEMP_LOW in kwargs:
-            data[TARGET_TEMP_LOW] = kwargs.get(ATTR_TARGET_TEMP_LOW)
         for temperature in data.values():
             if temperature is None or not (
@@ -267,10 +220,10 @@
         if mode := kwargs.get(ATTR_HVAC_MODE):
             await self.async_set_hvac_mode(mode)
-        await self.coordinator.api.set_temperature(self.device[LOCATION], data)
     @plugwise_command
-    async def async_set_hvac_mode(self, hvac_mode: str) -> None:
         """Set the hvac mode."""
         if hvac_mode not in self.hvac_modes:
             raise HomeAssistantError("Unsupported hvac_mode")
@@ -278,30 +231,17 @@
         if hvac_mode == self.hvac_mode:
             return
-        if hvac_mode != HVACMode.OFF:
             await self.coordinator.api.set_schedule_state(
-                self.device[LOCATION],
-                STATE_ON if hvac_mode == HVACMode.AUTO else STATE_OFF,
             )
-        if (
-            not self._homekit_enabled
-        ):  # pw-beta: feature request - mimic HomeKit behavior
-            if hvac_mode == HVACMode.OFF:
-                await self.coordinator.api.set_regulation_mode(hvac_mode)
-            elif self.hvac_mode == HVACMode.OFF:
                 await self.coordinator.api.set_regulation_mode(self._previous_mode)
-        else:
-            self._homekit_mode = hvac_mode  # pragma: no cover
-            if self._homekit_mode == HVACMode.OFF:  # pragma: no cover
-                await self.async_set_preset_mode(PRESET_AWAY)  # pragma: no cover
-            if (
-                self._homekit_mode in [HVACMode.HEAT, HVACMode.HEAT_COOL]
-                and self.device[ACTIVE_PRESET] == PRESET_AWAY
-            ):  # pragma: no cover
-                await self.async_set_preset_mode(PRESET_HOME)  # pragma: no cover
     @plugwise_command
     async def async_set_preset_mode(self, preset_mode: str) -> None:
         """Set the preset mode."""
-        await self.coordinator.api.set_preset(self.device[LOCATION], preset_mode)

 """Plugwise Climate component for Home Assistant."""
 from __future__ import annotations
 from typing import Any
+from homeassistant.components.climate import (
     ATTR_HVAC_MODE,
     ATTR_TARGET_TEMP_HIGH,
     ATTR_TARGET_TEMP_LOW,
+    ClimateEntity,
     ClimateEntityFeature,
     HVACAction,
     HVACMode,
 )
+from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.exceptions import HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
+from .const import DOMAIN, MASTER_THERMOSTATS
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
+    """Set up the Smile Thermostats from a config entry."""
     coordinator = entry.runtime_data
     @callback
     def _add_entities() -> None:
             return
         async_add_entities(
+            PlugwiseClimateEntity(coordinator, device_id)
+            for device_id in coordinator.new_devices
+            if coordinator.data.devices[device_id]["dev_class"] in MASTER_THERMOSTATS
         )
     _add_entities()
+    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
 class PlugwiseClimateEntity(PlugwiseEntity, ClimateEntity):
     _attr_translation_key = DOMAIN
     _enable_turn_on_off_backwards_compatibility = False
+    _previous_mode: str = "heating"
     def __init__(
         self,
         coordinator: PlugwiseDataUpdateCoordinator,
         device_id: str,
     ) -> None:
         """Set up the Plugwise API."""
         super().__init__(coordinator, device_id)
+        self._attr_extra_state_attributes = {}
         self._attr_unique_id = f"{device_id}-climate"
         self.cdr_gateway = coordinator.data.gateway
+        gateway_id: str = coordinator.data.gateway["gateway_id"]
+        self.gateway_data = coordinator.data.devices[gateway_id]
+        # Determine supported features
         self._attr_supported_features = ClimateEntityFeature.TARGET_TEMPERATURE
         if (
+            self.cdr_gateway["cooling_present"]
+            and self.cdr_gateway["smile_name"] != "Adam"
         ):
             self._attr_supported_features = (
                 ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
             self._attr_supported_features |= (
                 ClimateEntityFeature.TURN_OFF | ClimateEntityFeature.TURN_ON
             )
+        if presets := self.device.get("preset_modes"):
             self._attr_supported_features |= ClimateEntityFeature.PRESET_MODE
+            self._attr_preset_modes = presets
+        self._attr_min_temp = self.device["thermostat"]["lower_bound"]
+        self._attr_max_temp = min(self.device["thermostat"]["upper_bound"], 35.0)
+        # Ensure we don't drop below 0.1
+        self._attr_target_temperature_step = max(
+            self.device["thermostat"]["resolution"], 0.1
+        )
     def _previous_action_mode(self, coordinator: PlugwiseDataUpdateCoordinator) -> None:
         """Return the previous action-mode when the regulation-mode is not heating or cooling.
         """
         # When no cooling available, _previous_mode is always heating
         if (
+            "regulation_modes" in self.gateway_data
+            and "cooling" in self.gateway_data["regulation_modes"]
         ):
+            mode = self.gateway_data["select_regulation_mode"]
+            if mode in ("cooling", "heating"):
                 self._previous_mode = mode
     @property
     def current_temperature(self) -> float:
         """Return the current temperature."""
+        return self.device["sensors"]["temperature"]
     @property
     def target_temperature(self) -> float:
         Connected to the HVACMode combination of AUTO-HEAT.
         """
+        return self.device["thermostat"]["setpoint"]
     @property
     def target_temperature_high(self) -> float:
         Connected to the HVACMode combination of AUTO-HEAT_COOL.
         """
+        return self.device["thermostat"]["setpoint_high"]
     @property
     def target_temperature_low(self) -> float:
         Connected to the HVACMode combination AUTO-HEAT_COOL.
         """
+        return self.device["thermostat"]["setpoint_low"]
     @property
     def hvac_mode(self) -> HVACMode:
         """Return HVAC operation ie. auto, cool, heat, heat_cool, or off mode."""
+        if (mode := self.device.get("mode")) is None or mode not in self.hvac_modes:
+            return HVACMode.HEAT
         return HVACMode(mode)
     @property
     def hvac_modes(self) -> list[HVACMode]:
         """Return a list of available HVACModes."""
         hvac_modes: list[HVACMode] = []
+        if "regulation_modes" in self.gateway_data:
             hvac_modes.append(HVACMode.OFF)
+        if "available_schedules" in self.device:
             hvac_modes.append(HVACMode.AUTO)
+        if self.cdr_gateway["cooling_present"]:
+            if "regulation_modes" in self.gateway_data:
+                if self.gateway_data["select_regulation_mode"] == "cooling":
                     hvac_modes.append(HVACMode.COOL)
+                if self.gateway_data["select_regulation_mode"] == "heating":
                     hvac_modes.append(HVACMode.HEAT)
             else:
                 hvac_modes.append(HVACMode.HEAT_COOL)
         return hvac_modes
     @property
+    def hvac_action(self) -> HVACAction:
         """Return the current running hvac operation if supported."""
         # Keep track of the previous action-mode
         self._previous_action_mode(self.coordinator)
         # Adam provides the hvac_action for each thermostat
+        if (control_state := self.device.get("control_state")) == "cooling":
             return HVACAction.COOLING
+        if control_state == "heating":
             return HVACAction.HEATING
+        if control_state == "preheating":
             return HVACAction.PREHEATING
+        if control_state == "off":
             return HVACAction.IDLE
         heater: str = self.coordinator.data.gateway["heater_id"]
         heater_data = self.coordinator.data.devices[heater]
+        if heater_data["binary_sensors"]["heating_state"]:
             return HVACAction.HEATING
+        if heater_data["binary_sensors"].get("cooling_state", False):
             return HVACAction.COOLING
         return HVACAction.IDLE
     @property
     def preset_mode(self) -> str | None:
         """Return the current preset mode."""
+        return self.device.get("active_preset")
     @plugwise_command
     async def async_set_temperature(self, **kwargs: Any) -> None:
         """Set new target temperature."""
         data: dict[str, Any] = {}
         if ATTR_TEMPERATURE in kwargs:
+            data["setpoint"] = kwargs.get(ATTR_TEMPERATURE)
         if ATTR_TARGET_TEMP_HIGH in kwargs:
+            data["setpoint_high"] = kwargs.get(ATTR_TARGET_TEMP_HIGH)
         if ATTR_TARGET_TEMP_LOW in kwargs:
+            data["setpoint_low"] = kwargs.get(ATTR_TARGET_TEMP_LOW)
         for temperature in data.values():
             if temperature is None or not (
         if mode := kwargs.get(ATTR_HVAC_MODE):
             await self.async_set_hvac_mode(mode)
+        await self.coordinator.api.set_temperature(self.device["location"], data)
     @plugwise_command
+    async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
         """Set the hvac mode."""
         if hvac_mode not in self.hvac_modes:
             raise HomeAssistantError("Unsupported hvac_mode")
         if hvac_mode == self.hvac_mode:
             return
+        if hvac_mode == HVACMode.OFF:
+            await self.coordinator.api.set_regulation_mode(hvac_mode)
+        else:
             await self.coordinator.api.set_schedule_state(
+                self.device["location"],
+                "on" if hvac_mode == HVACMode.AUTO else "off",
             )
+            if self.hvac_mode == HVACMode.OFF:
                 await self.coordinator.api.set_regulation_mode(self._previous_mode)
     @plugwise_command
     async def async_set_preset_mode(self, preset_mode: str) -> None:
         """Set the preset mode."""
+        await self.coordinator.api.set_preset(self.device["location"], preset_mode)

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/config_flow.py RENAMED Viewed

@@ -1,8 +1,8 @@
 """Config flow for Plugwise integration."""
 from __future__ import annotations
-import datetime as dt  # pw-beta options
-from typing import Any
 from plugwise import Smile
 from plugwise.exceptions import (
@@ -16,14 +16,7 @@
 import voluptuous as vol
 from homeassistant.components.zeroconf import ZeroconfServiceInfo
-from homeassistant.config_entries import (
-    SOURCE_USER,
-    ConfigEntry,
-    ConfigFlow,
-    ConfigFlowResult,
-    OptionsFlow,
-    OptionsFlowWithConfigEntry,
-)
 from homeassistant.const import (
     ATTR_CONFIGURATION_URL,
     CONF_BASE,
@@ -31,75 +24,46 @@
     CONF_NAME,
     CONF_PASSWORD,
     CONF_PORT,
-    CONF_SCAN_INTERVAL,
     CONF_USERNAME,
 )
-from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 from .const import (
-    ANNA_WITH_ADAM,
-    CONF_HOMEKIT_EMULATION,  # pw-beta option
-    CONF_REFRESH_INTERVAL,  # pw-beta option
-    CONTEXT,
-    COORDINATOR,
     DEFAULT_PORT,
-    DEFAULT_SCAN_INTERVAL,  # pw-beta option
     DEFAULT_USERNAME,
     DOMAIN,
-    FLOW_ID,
     FLOW_SMILE,
     FLOW_STRETCH,
-    INIT,
-    PRODUCT,
     SMILE,
-    SMILE_OPEN_THERM,
-    SMILE_THERMO,
     STRETCH,
     STRETCH_USERNAME,
-    THERMOSTAT,
-    TITLE_PLACEHOLDERS,
-    VERSION,
     ZEROCONF_MAP,
 )
-def _base_schema(
-    discovery_info: ZeroconfServiceInfo | None,
-    user_input: dict[str, Any] | None,
-) -> vol.Schema:
     """Generate base schema for gateways."""
     if not discovery_info:
-        if not user_input:
-            return vol.Schema(
-                {
-                    vol.Required(CONF_PASSWORD): str,
-                    vol.Required(CONF_HOST): str,
-                    vol.Optional(CONF_PORT, default=DEFAULT_PORT): int,
-                    vol.Required(CONF_USERNAME, default=SMILE): vol.In(
-                        {SMILE: FLOW_SMILE, STRETCH: FLOW_STRETCH}
-                    ),
-                }
-            )
-        return vol.Schema(
             {
-                vol.Required(CONF_PASSWORD, default=user_input[CONF_PASSWORD]): str,
-                vol.Required(CONF_HOST, default=user_input[CONF_HOST]): str,
-                vol.Optional(CONF_PORT, default=user_input[CONF_PORT]): int,
-                vol.Required(CONF_USERNAME, default=user_input[CONF_USERNAME]): vol.In(
                     {SMILE: FLOW_SMILE, STRETCH: FLOW_STRETCH}
                 ),
             }
         )
-    return vol.Schema({vol.Required(CONF_PASSWORD): str})
 async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> Smile:
     """Validate whether the user input allows us to connect to the gateway.
-    Data has the keys from _base_schema() with values provided by the user.
     """
     websession = async_get_clientsession(hass, verify_ssl=False)
     api = Smile(
@@ -107,7 +71,6 @@
         password=data[CONF_PASSWORD],
         port=data[CONF_PORT],
         username=data[CONF_USERNAME],
-        timeout=30,
         websession=websession,
     )
     await api.connect()
@@ -120,6 +83,7 @@
     VERSION = 1
     discovery_info: ZeroconfServiceInfo | None = None
     _username: str = DEFAULT_USERNAME
     async def async_step_zeroconf(
@@ -153,163 +117,79 @@
         if DEFAULT_USERNAME not in unique_id:
             self._username = STRETCH_USERNAME
-        _product = _properties.get(PRODUCT, None)
-        _version = _properties.get(VERSION, "n/a")
         _name = f"{ZEROCONF_MAP.get(_product, _product)} v{_version}"
         # This is an Anna, but we already have config entries.
         # Assuming that the user has already configured Adam, aborting discovery.
-        if self._async_current_entries() and _product == SMILE_THERMO:
-            return self.async_abort(reason=ANNA_WITH_ADAM)
         # If we have discovered an Adam or Anna, both might be on the network.
         # In that case, we need to cancel the Anna flow, as the Adam should
         # be added.
-        for flow in self._async_in_progress():
-            # This is an Anna, and there is already an Adam flow in progress
-            if (
-                _product == SMILE_THERMO
-                and CONTEXT in flow
-                and flow[CONTEXT].get(PRODUCT) == SMILE_OPEN_THERM
-            ):
-                return self.async_abort(reason=ANNA_WITH_ADAM)
-            # This is an Adam, and there is already an Anna flow in progress
-            if (
-                _product == SMILE_OPEN_THERM
-                and CONTEXT in flow
-                and flow[CONTEXT].get(PRODUCT) == SMILE_THERMO
-                and FLOW_ID in flow
-            ):
-                self.hass.config_entries.flow.async_abort(flow[FLOW_ID])
         self.context.update(
             {
-                TITLE_PLACEHOLDERS: {
-                    CONF_HOST: discovery_info.host,
-                    CONF_NAME: _name,
-                    CONF_PORT: discovery_info.port,
-                    CONF_USERNAME: self._username,
-                },
                 ATTR_CONFIGURATION_URL: (
                     f"http://{discovery_info.host}:{discovery_info.port}"
                 ),
-                PRODUCT: _product,
             }
         )
         return await self.async_step_user()
     async def async_step_user(
         self, user_input: dict[str, Any] | None = None
     ) -> ConfigFlowResult:
         """Handle the initial step when using network/gateway setups."""
         errors: dict[str, str] = {}
-        if not user_input:
-            return self.async_show_form(
-                step_id=SOURCE_USER,
-                data_schema=_base_schema(self.discovery_info, None),
-                errors=errors,
-            )
-        if self.discovery_info:
-            user_input[CONF_HOST] = self.discovery_info.host
-            user_input[CONF_PORT] = self.discovery_info.port
-            user_input[CONF_USERNAME] = self._username
-        try:
-            api = await validate_input(self.hass, user_input)
-        except ConnectionFailedError:
-            errors[CONF_BASE] = "cannot_connect"
-        except InvalidAuthentication:
-            errors[CONF_BASE] = "invalid_auth"
-        except InvalidSetupError:
-            errors[CONF_BASE] = "invalid_setup"
-        except (InvalidXMLError, ResponseError):
-            errors[CONF_BASE] = "response_error"
-        except UnsupportedDeviceError:
-            errors[CONF_BASE] = "unsupported"
-        except Exception:  # noqa: BLE001
-            errors[CONF_BASE] = "unknown"
-        if errors:
-            return self.async_show_form(
-                step_id=SOURCE_USER,
-                data_schema=_base_schema(None, user_input),
-                errors=errors,
-            )
-        await self.async_set_unique_id(
-            api.smile_hostname or api.gateway_id, raise_on_progress=False
-        )
-        self._abort_if_unique_id_configured()
-        return self.async_create_entry(title=api.smile_name, data=user_input)
-    @staticmethod
-    @callback
-    def async_get_options_flow(
-        config_entry: ConfigEntry,
-    ) -> OptionsFlow:  # pw-beta options
-        """Get the options flow for this handler."""
-        return PlugwiseOptionsFlowHandler(config_entry)
-# pw-beta - change the scan-interval via CONFIGURE
-# pw-beta - add homekit emulation via CONFIGURE
-# pw-beta - change the frontend refresh interval via CONFIGURE
-class PlugwiseOptionsFlowHandler(OptionsFlowWithConfigEntry):  # pw-beta options
-    """Plugwise option flow."""
-    async def async_step_none(
-        self, user_input: dict[str, Any] | None = None
-    ) -> ConfigFlowResult:  # pragma: no cover
-        """No options available."""
-        if user_input is not None:
-            # Apparently not possible to abort an options flow at the moment
-            return self.async_create_entry(title="", data=self._options)
-        return self.async_show_form(step_id="none")
-    async def async_step_init(
-        self, user_input: dict[str, Any] | None = None
-    ) -> ConfigFlowResult:  # pragma: no cover
-        """Manage the Plugwise options."""
-        if not self.config_entry.data.get(CONF_HOST):
-            return await self.async_step_none(user_input)
         if user_input is not None:
-            return self.async_create_entry(title="", data=user_input)
-        coordinator = self.hass.data[DOMAIN][self.config_entry.entry_id][COORDINATOR]
-        interval: dt.timedelta = DEFAULT_SCAN_INTERVAL[
-            coordinator.api.smile_type
-        ]  # pw-beta options
-        data = {
-            vol.Optional(
-                CONF_SCAN_INTERVAL,
-                default=self._options.get(
-                    CONF_SCAN_INTERVAL, interval.seconds
-                ),
-            ): vol.All(cv.positive_int, vol.Clamp(min=10)),
-        }  # pw-beta
-        if coordinator.api.smile_type != THERMOSTAT:
-            return self.async_show_form(step_id=INIT, data_schema=vol.Schema(data))
-        data.update(
-            {
-                vol.Optional(
-                    CONF_HOMEKIT_EMULATION,
-                    default=self._options.get(
-                        CONF_HOMEKIT_EMULATION, False
-                    ),
-                ): cv.boolean,
-                vol.Optional(
-                    CONF_REFRESH_INTERVAL,
-                    default=self._options.get(CONF_REFRESH_INTERVAL, 1.5),
-                ): vol.All(vol.Coerce(float), vol.Range(min=1.5, max=10.0)),
-            }
-        )  # pw-beta
-        return self.async_show_form(step_id=INIT, data_schema=vol.Schema(data))

 """Config flow for Plugwise integration."""
 from __future__ import annotations
+from typing import Any, Self
 from plugwise import Smile
 from plugwise.exceptions import (
 import voluptuous as vol
 from homeassistant.components.zeroconf import ZeroconfServiceInfo
+from homeassistant.config_entries import SOURCE_USER, ConfigFlow, ConfigFlowResult
 from homeassistant.const import (
     ATTR_CONFIGURATION_URL,
     CONF_BASE,
     CONF_NAME,
     CONF_PASSWORD,
     CONF_PORT,
     CONF_USERNAME,
 )
+from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 from .const import (
     DEFAULT_PORT,
     DEFAULT_USERNAME,
     DOMAIN,
     FLOW_SMILE,
     FLOW_STRETCH,
     SMILE,
     STRETCH,
     STRETCH_USERNAME,
     ZEROCONF_MAP,
 )
+def base_schema(discovery_info: ZeroconfServiceInfo | None) -> vol.Schema:
     """Generate base schema for gateways."""
+    schema = vol.Schema({vol.Required(CONF_PASSWORD): str})
     if not discovery_info:
+        schema = schema.extend(
             {
+                vol.Required(CONF_HOST): str,
+                vol.Optional(CONF_PORT, default=DEFAULT_PORT): int,
+                vol.Required(CONF_USERNAME, default=SMILE): vol.In(
                     {SMILE: FLOW_SMILE, STRETCH: FLOW_STRETCH}
                 ),
             }
         )
+    return schema
 async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> Smile:
     """Validate whether the user input allows us to connect to the gateway.
+    Data has the keys from base_schema() with values provided by the user.
     """
     websession = async_get_clientsession(hass, verify_ssl=False)
     api = Smile(
         password=data[CONF_PASSWORD],
         port=data[CONF_PORT],
         username=data[CONF_USERNAME],
         websession=websession,
     )
     await api.connect()
     VERSION = 1
     discovery_info: ZeroconfServiceInfo | None = None
+    product: str = "Unknown Smile"
     _username: str = DEFAULT_USERNAME
     async def async_step_zeroconf(
         if DEFAULT_USERNAME not in unique_id:
             self._username = STRETCH_USERNAME
+        self.product = _product = _properties.get("product", "Unknown Smile")
+        _version = _properties.get("version", "n/a")
         _name = f"{ZEROCONF_MAP.get(_product, _product)} v{_version}"
         # This is an Anna, but we already have config entries.
         # Assuming that the user has already configured Adam, aborting discovery.
+        if self._async_current_entries() and _product == "smile_thermo":
+            return self.async_abort(reason="anna_with_adam")
         # If we have discovered an Adam or Anna, both might be on the network.
         # In that case, we need to cancel the Anna flow, as the Adam should
         # be added.
+        if self.hass.config_entries.flow.async_has_matching_flow(self):
+            return self.async_abort(reason="anna_with_adam")
         self.context.update(
             {
+                "title_placeholders": {CONF_NAME: _name},
                 ATTR_CONFIGURATION_URL: (
                     f"http://{discovery_info.host}:{discovery_info.port}"
                 ),
             }
         )
         return await self.async_step_user()
+    def is_matching(self, other_flow: Self) -> bool:
+        """Return True if other_flow is matching this flow."""
+        # This is an Anna, and there is already an Adam flow in progress
+        if self.product == "smile_thermo" and other_flow.product == "smile_open_therm":
+            return True
+        # This is an Adam, and there is already an Anna flow in progress
+        if self.product == "smile_open_therm" and other_flow.product == "smile_thermo":
+            self.hass.config_entries.flow.async_abort(other_flow.flow_id)
+        return False
     async def async_step_user(
         self, user_input: dict[str, Any] | None = None
     ) -> ConfigFlowResult:
         """Handle the initial step when using network/gateway setups."""
         errors: dict[str, str] = {}
         if user_input is not None:
+            if self.discovery_info:
+                user_input[CONF_HOST] = self.discovery_info.host
+                user_input[CONF_PORT] = self.discovery_info.port
+                user_input[CONF_USERNAME] = self._username
+            try:
+                api = await validate_input(self.hass, user_input)
+            except ConnectionFailedError:
+                errors[CONF_BASE] = "cannot_connect"
+            except InvalidAuthentication:
+                errors[CONF_BASE] = "invalid_auth"
+            except InvalidSetupError:
+                errors[CONF_BASE] = "invalid_setup"
+            except (InvalidXMLError, ResponseError):
+                errors[CONF_BASE] = "response_error"
+            except UnsupportedDeviceError:
+                errors[CONF_BASE] = "unsupported"
+            except Exception:  # noqa: BLE001
+                errors[CONF_BASE] = "unknown"
+            else:
+                await self.async_set_unique_id(
+                    api.smile_hostname or api.gateway_id, raise_on_progress=False
+                )
+                self._abort_if_unique_id_configured()
+                return self.async_create_entry(title=api.smile_name, data=user_input)
+        return self.async_show_form(
+            step_id=SOURCE_USER,
+            data_schema=base_schema(self.discovery_info),
+            errors=errors,
+        )

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/const.py RENAMED Viewed

@@ -1,4 +1,7 @@
 """Constants for Plugwise component."""
 from datetime import timedelta
 import logging
 from typing import Final, Literal
@@ -10,154 +13,18 @@
 LOGGER = logging.getLogger(__package__)
 API: Final = "api"
-COORDINATOR: Final = "coordinator"
-CONF_HOMEKIT_EMULATION: Final = "homekit_emulation"  # pw-beta options
-CONF_REFRESH_INTERVAL: Final = "refresh_interval"  # pw-beta options
-CONF_MANUAL_PATH: Final = "Enter Manually"
-DEVICES: Final = "devices"
 GATEWAY: Final = "gateway"
 LOCATION: Final = "location"
-MAC_ADDRESS: Final = "mac_address"
 REBOOT: Final = "reboot"
 SMILE: Final = "smile"
 STRETCH: Final = "stretch"
 STRETCH_USERNAME: Final = "stretch"
-UNIQUE_IDS: Final = "unique_ids"
-ZIGBEE_MAC_ADDRESS: Final = "zigbee_mac_address"
-# Binary Sensor constants
-BINARY_SENSORS: Final = "binary_sensors"
-COMPRESSOR_STATE: Final = "compressor_state"
-COOLING_ENABLED: Final = "cooling_enabled"
-COOLING_STATE: Final = "cooling_state"
-DHW_STATE: Final = "dhw_state"
-FLAME_STATE: Final = "flame_state"
-HEATING_STATE: Final = "heating_state"
-NOTIFICATIONS: Final ="notifications"
-PLUGWISE_NOTIFICATION: Final = "plugwise_notification"
-SECONDARY_BOILER_STATE: Final = "secondary_boiler_state"
-# Climate constants
-ACTIVE_PRESET: Final = "active_preset"
-CONTROL_STATE: Final = "control_state"
-COOLING_PRESENT: Final ="cooling_present"
-DEV_CLASS: Final = "dev_class"
-NONE : Final = "None"
-MODE: Final = "mode"
-TARGET_TEMP: Final = "setpoint"
-TARGET_TEMP_HIGH: Final = "setpoint_high"
-TARGET_TEMP_LOW: Final = "setpoint_low"
-THERMOSTAT: Final = "thermostat"
-# Config_flow constants
-ANNA_WITH_ADAM: Final = "anna_with_adam"
-CONTEXT: Final = "context"
-FLOW_ID: Final = "flow_id"
-FLOW_NET: Final = "Network: Smile/Stretch"
-FLOW_SMILE: Final = "Smile (Adam/Anna/P1)"
-FLOW_STRETCH: Final = "Stretch (Stretch)"
-FLOW_TYPE: Final = "flow_type"
-INIT: Final = "init"
-PRODUCT: Final = "product"
-SMILE_OPEN_THERM: Final = "smile_open_therm"
-SMILE_THERMO: Final = "smile_thermo"
-TITLE_PLACEHOLDERS: Final = "title_placeholders"
-VERSION: Final = "version"
-# Entity constants
-AVAILABLE: Final = "available"
-FIRMWARE: Final = "firmware"
-GATEWAY_ID: Final = "gateway_id"
-HARDWARE: Final = "hardware"
-MODEL: Final = "model"
-SMILE_NAME: Final = "smile_name"
-VENDOR: Final = "vendor"
-# Number constants
-MAX_BOILER_TEMP: Final = "maximum_boiler_temperature"
-MAX_DHW_TEMP: Final = "max_dhw_temperature"
-LOWER_BOUND: Final = "lower_bound"
-RESOLUTION: Final = "resolution"
-TEMPERATURE_OFFSET: Final = "temperature_offset"
-UPPER_BOUND: Final = "upper_bound"
-# Sensor constants
-DHW_TEMP: Final = "dhw_temperature"
-DHW_SETPOINT: Final = "domestic_hot_water_setpoint"
-EL_CONSUMED: Final = "electricity_consumed"
-EL_CONS_INTERVAL: Final = "electricity_consumed_interval"
-EL_CONS_OP_CUMULATIVE: Final = "electricity_consumed_off_peak_cumulative"
-EL_CONS_OP_INTERVAL: Final = "electricity_consumed_off_peak_interval"
-EL_CONS_OP_POINT: Final = "electricity_consumed_off_peak_point"
-EL_CONS_P_CUMULATIVE: Final = "electricity_consumed_peak_cumulative"
-EL_CONS_P_INTERVAL: Final = "electricity_consumed_peak_interval"
-EL_CONS_P_POINT: Final = "electricity_consumed_peak_point"
-EL_CONS_POINT: Final = "electricity_consumed_point"
-EL_PH1_CONSUMED: Final = "electricity_phase_one_consumed"
-EL_PH2_CONSUMED: Final = "electricity_phase_two_consumed"
-EL_PH3_CONSUMED: Final = "electricity_phase_three_consumed"
-EL_PH1_PRODUCED: Final = "electricity_phase_one_produced"
-EL_PH2_PRODUCED: Final = "electricity_phase_two_produced"
-EL_PH3_PRODUCED: Final = "electricity_phase_three_produced"
-EL_PRODUCED: Final = "electricity_produced"
-EL_PROD_INTERVAL: Final = "electricity_produced_interval"
-EL_PROD_OP_CUMULATIVE: Final = "electricity_produced_off_peak_cumulative"
-EL_PROD_OP_INTERVAL: Final = "electricity_produced_off_peak_interval"
-EL_PROD_OP_POINT: Final = "electricity_produced_off_peak_point"
-EL_PROD_P_CUMULATIVE: Final = "electricity_produced_peak_cumulative"
-EL_PROD_P_INTERVAL: Final = "electricity_produced_peak_interval"
-EL_PROD_P_POINT: Final = "electricity_produced_peak_point"
-EL_PROD_POINT: Final = "electricity_produced_point"
-GAS_CONS_CUMULATIVE: Final = "gas_consumed_cumulative"
-GAS_CONS_INTERVAL: Final = "gas_consumed_interval"
-INTENDED_BOILER_TEMP: Final = "intended_boiler_temperature"
-MOD_LEVEL: Final = "modulation_level"
-NET_EL_POINT: Final = "net_electricity_point"
-NET_EL_CUMULATIVE: Final = "net_electricity_cumulative"
-OUTDOOR_AIR_TEMP: Final = "outdoor_air_temperature"
-OUTDOOR_TEMP: Final = "outdoor_temperature"
-RETURN_TEMP: Final = "return_temperature"
-SENSORS: Final = "sensors"
-TEMP_DIFF: Final = "temperature_difference"
-VALVE_POS: Final = "valve_position"
-VOLTAGE_PH1: Final = "voltage_phase_one"
-VOLTAGE_PH2: Final = "voltage_phase_two"
-VOLTAGE_PH3: Final = "voltage_phase_three"
-WATER_TEMP: Final = "water_temperature"
-WATER_PRESSURE: Final = "water_pressure"
-# Select constants
-AVAILABLE_SCHEDULES: Final = "available_schedules"
-DHW_MODE: Final = "dhw_mode"
-DHW_MODES: Final = "dhw_modes"
-GATEWAY_MODE: Final = "gateway_mode"
-GATEWAY_MODES: Final = "gateway_modes"
-REGULATION_MODE: Final = "regulation_mode"
-REGULATION_MODES: Final = "regulation_modes"
-SELECT_DHW_MODE: Final = "select_dhw_mode"
-SELECT_GATEWAY_MODE: Final = "select_gateway_mode"
-SELECT_REGULATION_MODE: Final = "select_regulation_mode"
-SELECT_SCHEDULE: Final = "select_schedule"
-# Switch constants
-DHW_CM_SWITCH: Final = "dhw_cm_switch"
-LOCK: Final = "lock"
-MEMBERS: Final ="members"
-RELAY: Final = "relay"
-COOLING_ENA_SWITCH: Final ="cooling_ena_switch"
-SWITCHES: Final = "switches"
-# Default directives
-DEFAULT_PORT: Final[int] = 80
-DEFAULT_SCAN_INTERVAL: Final[dict[str, timedelta]] = {
-    "power": timedelta(seconds=10),
-    "stretch": timedelta(seconds=60),
-    "thermostat": timedelta(seconds=60),
-}
-DEFAULT_TIMEOUT: Final[int] = 10
-DEFAULT_USERNAME: Final = "smile"
-# --- Const for Plugwise Smile and Stretch
 PLATFORMS: Final[list[str]] = [
     Platform.BINARY_SENSOR,
     Platform.BUTTON,
@@ -167,18 +34,6 @@
     Platform.SENSOR,
     Platform.SWITCH,
 ]
-SERVICE_DELETE: Final = "delete_notification"
-SEVERITIES: Final[list[str]] = ["other", "info", "message", "warning", "error"]
-# Climate const:
-MASTER_THERMOSTATS: Final[list[str]] = [
-    "thermostat",
-    "zone_thermometer",
-    "zone_thermostat",
-    "thermostatic_radiator_valve",
-]
-# Config_flow const:
 ZEROCONF_MAP: Final[dict[str, str]] = {
     "smile": "Smile P1",
     "smile_thermo": "Smile Anna",
@@ -186,21 +41,39 @@
     "stretch": "Stretch",
 }
-NumberType = Literal[
     "maximum_boiler_temperature",
     "max_dhw_temperature",
     "temperature_offset",
 ]
-SelectType = Literal[
     "select_dhw_mode",
     "select_gateway_mode",
     "select_regulation_mode",
     "select_schedule",
 ]
-SelectOptionsType = Literal[
     "dhw_modes",
     "gateway_modes",
     "regulation_modes",
     "available_schedules",
 ]

 """Constants for Plugwise component."""
+from __future__ import annotations
 from datetime import timedelta
 import logging
 from typing import Final, Literal
 LOGGER = logging.getLogger(__package__)
 API: Final = "api"
+FLOW_SMILE: Final = "smile (Adam/Anna/P1)"
+FLOW_STRETCH: Final = "stretch (Stretch)"
+FLOW_TYPE: Final = "flow_type"
 GATEWAY: Final = "gateway"
+GATEWAY_ID: Final = "gateway_id"
 LOCATION: Final = "location"
+PW_TYPE: Final = "plugwise_type"
 REBOOT: Final = "reboot"
 SMILE: Final = "smile"
 STRETCH: Final = "stretch"
 STRETCH_USERNAME: Final = "stretch"
 PLATFORMS: Final[list[str]] = [
     Platform.BINARY_SENSOR,
     Platform.BUTTON,
     Platform.SENSOR,
     Platform.SWITCH,
 ]
 ZEROCONF_MAP: Final[dict[str, str]] = {
     "smile": "Smile P1",
     "smile_thermo": "Smile Anna",
     "stretch": "Stretch",
 }
+type NumberType = Literal[
     "maximum_boiler_temperature",
     "max_dhw_temperature",
     "temperature_offset",
 ]
+type SelectType = Literal[
     "select_dhw_mode",
     "select_gateway_mode",
     "select_regulation_mode",
     "select_schedule",
 ]
+type SelectOptionsType = Literal[
     "dhw_modes",
     "gateway_modes",
     "regulation_modes",
     "available_schedules",
 ]
+# Default directives
+DEFAULT_MAX_TEMP: Final = 30
+DEFAULT_MIN_TEMP: Final = 4
+DEFAULT_PORT: Final = 80
+DEFAULT_SCAN_INTERVAL: Final[dict[str, timedelta]] = {
+    "power": timedelta(seconds=10),
+    "stretch": timedelta(seconds=60),
+    "thermostat": timedelta(seconds=60),
+}
+DEFAULT_USERNAME: Final = "smile"
+MASTER_THERMOSTATS: Final[list[str]] = [
+    "thermostat",
+    "thermostatic_radiator_valve",
+    "zone_thermometer",
+    "zone_thermostat",
+]

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/coordinator.py RENAMED Viewed

@@ -1,8 +1,8 @@
 """DataUpdateCoordinator for Plugwise."""
-from collections.abc import Callable
 from datetime import timedelta
 from plugwise import PlugwiseData, Smile
 from plugwise.exceptions import (
     ConnectionFailedError,
@@ -14,13 +14,7 @@
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import (
-    CONF_HOST,
-    CONF_PASSWORD,
-    CONF_PORT,
-    CONF_SCAN_INTERVAL,  # pw-beta options
-    CONF_USERNAME,
-)
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryError
 from homeassistant.helpers import device_registry as dr
@@ -28,14 +22,7 @@
 from homeassistant.helpers.debounce import Debouncer
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import (
-    DEFAULT_PORT,
-    DEFAULT_SCAN_INTERVAL,
-    DEFAULT_USERNAME,
-    DOMAIN,
-    GATEWAY_ID,
-    LOGGER,
-)
 class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[PlugwiseData]):
@@ -45,25 +32,19 @@
     config_entry: ConfigEntry
-    def __init__(
-        self,
-        hass: HomeAssistant,
-        cooldown: float,
-        update_interval: timedelta = timedelta(seconds=60),
-    ) -> None:  # pw-beta cooldown
         """Initialize the coordinator."""
         super().__init__(
             hass,
             LOGGER,
             name=DOMAIN,
-            # Core directly updates from const's DEFAULT_SCAN_INTERVAL
-            update_interval=update_interval,
             # Don't refresh immediately, give the device time to process
             # the change in state before we query it.
             request_refresh_debouncer=Debouncer(
                 hass,
                 LOGGER,
-                cooldown=cooldown,
                 immediate=False,
             ),
         )
@@ -73,28 +54,17 @@
             username=self.config_entry.data.get(CONF_USERNAME, DEFAULT_USERNAME),
             password=self.config_entry.data[CONF_PASSWORD],
             port=self.config_entry.data.get(CONF_PORT, DEFAULT_PORT),
-            timeout=30,
             websession=async_get_clientsession(hass, verify_ssl=False),
         )
         self._current_devices: set[str] = set()
         self.new_devices: set[str] = set()
-        self.new_devices_callbacks: list[Callable[[str], None]] = []
-        self.update_interval = update_interval
     async def _connect(self) -> None:
         """Connect to the Plugwise Smile."""
-        self._connected = await self.api.connect()
-        self.api.get_all_devices()
-        self.update_interval = DEFAULT_SCAN_INTERVAL.get(
-            self.api.smile_type, timedelta(seconds=60)
-        )  # pw-beta options scan-interval
-        if (custom_time := self.config_entry.options.get(CONF_SCAN_INTERVAL)) is not None:
-            self.update_interval = timedelta(
-                seconds=int(custom_time)
-            )  # pragma: no cover  # pw-beta options
-        LOGGER.debug("DUC update interval: %s", self.update_interval)  # pw-beta options
     async def _async_update_data(self) -> PlugwiseData:
         """Fetch data from Plugwise."""
@@ -104,56 +74,52 @@
                 await self._connect()
             data = await self.api.async_update()
         except ConnectionFailedError as err:
-                raise UpdateFailed("Failed to connect") from err
         except InvalidAuthentication as err:
-                raise ConfigEntryError("Authentication failed") from err
         except (InvalidXMLError, ResponseError) as err:
-                raise UpdateFailed(
-                    "Invalid XML data, or error indication received from the Plugwise Adam/Smile/Stretch"
-                ) from err
         except PlugwiseError as err:
-                raise UpdateFailed("Data incomplete or missing") from err
         except UnsupportedDeviceError as err:
-                raise ConfigEntryError("Device with unsupported firmware") from err
         else:
-            LOGGER.debug(f"{self.api.smile_name} data: %s", data)
             self._async_add_remove_devices(data, self.config_entry)
         return data
-    def _async_add_remove_devices(self, data:PlugwiseData, entry: ConfigEntry,) -> None:
         """Add new Plugwise devices, remove non-existing devices."""
         # Check for new or removed devices
-        if new_devices := set(data.devices) - self._current_devices:
-            self.new_devices = new_devices
-            LOGGER.debug("New Plugwise devices found: %s", new_devices)
-            for device_id in new_devices:
-                for callback in self.new_devices_callbacks:
-                    callback(device_id)
         removed_devices = self._current_devices - set(data.devices)
         self._current_devices = set(data.devices)
-        if not removed_devices:
-            return
-        # Clean device_registry when removed devices found
         device_reg = dr.async_get(self.hass)
         device_list = dr.async_entries_for_config_entry(
             device_reg, self.config_entry.entry_id
         )
-        # via_device cannot be None, this will result in the deletion
-        # of other Plugwise Gateways when present!
-        via_device: str = ""
         for device_entry in device_list:
-            if device_entry.identifiers:
-                item = list(list(device_entry.identifiers)[0])
-                if item[0] == DOMAIN:
-                    # First find the Plugwise via_device, this is always the first device
-                    if item[1] == data.gateway[GATEWAY_ID]:
-                        via_device = device_entry.id
-                    elif ( # then remove the connected orphaned device(s)
-                        device_entry.via_device_id == via_device
-                        and item[1] not in list(data.devices.keys())
                     ):
                         device_reg.async_update_device(
                             device_entry.id, remove_config_entry_id=entry.entry_id
@@ -162,5 +128,5 @@
                             "Removed %s device %s %s from device_registry",
                             DOMAIN,
                             device_entry.model,
-                            item[1],
                         )

 """DataUpdateCoordinator for Plugwise."""
 from datetime import timedelta
+from packaging.version import Version
 from plugwise import PlugwiseData, Smile
 from plugwise.exceptions import (
     ConnectionFailedError,
 )
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_USERNAME
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryError
 from homeassistant.helpers import device_registry as dr
 from homeassistant.helpers.debounce import Debouncer
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+from .const import DEFAULT_PORT, DEFAULT_USERNAME, DOMAIN, GATEWAY_ID, LOGGER
 class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[PlugwiseData]):
     config_entry: ConfigEntry
+    def __init__(self, hass: HomeAssistant) -> None:
         """Initialize the coordinator."""
         super().__init__(
             hass,
             LOGGER,
             name=DOMAIN,
+            update_interval=timedelta(seconds=60),
             # Don't refresh immediately, give the device time to process
             # the change in state before we query it.
             request_refresh_debouncer=Debouncer(
                 hass,
                 LOGGER,
+                cooldown=1.5,
                 immediate=False,
             ),
         )
             username=self.config_entry.data.get(CONF_USERNAME, DEFAULT_USERNAME),
             password=self.config_entry.data[CONF_PASSWORD],
             port=self.config_entry.data.get(CONF_PORT, DEFAULT_PORT),
             websession=async_get_clientsession(hass, verify_ssl=False),
         )
         self._current_devices: set[str] = set()
         self.new_devices: set[str] = set()
     async def _connect(self) -> None:
         """Connect to the Plugwise Smile."""
+        version = await self.api.connect()
+        self._connected = isinstance(version, Version)
+        if self._connected:
+            self.api.get_all_devices()
     async def _async_update_data(self) -> PlugwiseData:
         """Fetch data from Plugwise."""
                 await self._connect()
             data = await self.api.async_update()
         except ConnectionFailedError as err:
+            raise UpdateFailed("Failed to connect") from err
         except InvalidAuthentication as err:
+            raise ConfigEntryError("Authentication failed") from err
         except (InvalidXMLError, ResponseError) as err:
+            raise UpdateFailed(
+                "Invalid XML data, or error indication received from the Plugwise Adam/Smile/Stretch"
+            ) from err
         except PlugwiseError as err:
+            raise UpdateFailed("Data incomplete or missing") from err
         except UnsupportedDeviceError as err:
+            raise ConfigEntryError("Device with unsupported firmware") from err
         else:
             self._async_add_remove_devices(data, self.config_entry)
         return data
+    def _async_add_remove_devices(self, data: PlugwiseData, entry: ConfigEntry) -> None:
         """Add new Plugwise devices, remove non-existing devices."""
         # Check for new or removed devices
+        self.new_devices = set(data.devices) - self._current_devices
         removed_devices = self._current_devices - set(data.devices)
         self._current_devices = set(data.devices)
+        if removed_devices:
+            self._async_remove_devices(data, entry)
+    def _async_remove_devices(self, data: PlugwiseData, entry: ConfigEntry) -> None:
+        """Clean registries when removed devices found."""
         device_reg = dr.async_get(self.hass)
         device_list = dr.async_entries_for_config_entry(
             device_reg, self.config_entry.entry_id
         )
+        # First find the Plugwise via_device
+        gateway_device = device_reg.async_get_device(
+            {(DOMAIN, data.gateway[GATEWAY_ID])}
+        )
+        assert gateway_device is not None
+        via_device_id = gateway_device.id
+        # Then remove the connected orphaned device(s)
         for device_entry in device_list:
+            for identifier in device_entry.identifiers:
+                if identifier[0] == DOMAIN:
+                    if (
+                        device_entry.via_device_id == via_device_id
+                        and identifier[1] not in data.devices
                     ):
                         device_reg.async_update_device(
                             device_entry.id, remove_config_entry_id=entry.entry_id
                             "Removed %s device %s %s from device_registry",
                             DOMAIN,
                             device_entry.model,
+                            identifier[1],
                         )

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/diagnostics.py RENAMED Viewed

@@ -1,4 +1,5 @@
 """Diagnostics support for Plugwise."""
 from __future__ import annotations
 from typing import Any

 """Diagnostics support for Plugwise."""
 from __future__ import annotations
 from typing import Any

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/entity.py RENAMED Viewed

@@ -1,4 +1,5 @@
 """Generic Plugwise Entity Class."""
 from __future__ import annotations
 from plugwise.constants import DeviceData
@@ -11,18 +12,7 @@
 )
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .const import (
-    AVAILABLE,
-    DOMAIN,
-    FIRMWARE,
-    GATEWAY_ID,
-    HARDWARE,
-    MAC_ADDRESS,
-    MODEL,
-    SMILE_NAME,
-    VENDOR,
-    ZIGBEE_MAC_ADDRESS,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
@@ -46,29 +36,30 @@
         data = coordinator.data.devices[device_id]
         connections = set()
-        if mac := data.get(MAC_ADDRESS):
             connections.add((CONNECTION_NETWORK_MAC, mac))
-        if mac := data.get(ZIGBEE_MAC_ADDRESS):
             connections.add((CONNECTION_ZIGBEE, mac))
         self._attr_device_info = DeviceInfo(
             configuration_url=configuration_url,
             identifiers={(DOMAIN, device_id)},
             connections=connections,
-            manufacturer=data.get(VENDOR),
-            model=data.get(MODEL),
-            name=coordinator.data.gateway[SMILE_NAME],
-            sw_version=data.get(FIRMWARE),
-            hw_version=data.get(HARDWARE),
         )
-        if device_id != coordinator.data.gateway[GATEWAY_ID]:
             self._attr_device_info.update(
                 {
-                    ATTR_NAME: data.get(ATTR_NAME),
                     ATTR_VIA_DEVICE: (
                         DOMAIN,
-                        str(self.coordinator.data.gateway[GATEWAY_ID]),
                     ),
                 }
             )
@@ -78,9 +69,7 @@
         """Return if entity is available."""
         return (
             self._dev_id in self.coordinator.data.devices
-            # Do not change the below line: some Plugwise devices
-            # do not provide their availability-status!
-            and (AVAILABLE not in self.device or self.device[AVAILABLE] is True)
             and super().available
         )

 """Generic Plugwise Entity Class."""
 from __future__ import annotations
 from plugwise.constants import DeviceData
 )
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
+from .const import DOMAIN
 from .coordinator import PlugwiseDataUpdateCoordinator
         data = coordinator.data.devices[device_id]
         connections = set()
+        if mac := data.get("mac_address"):
             connections.add((CONNECTION_NETWORK_MAC, mac))
+        if mac := data.get("zigbee_mac_address"):
             connections.add((CONNECTION_ZIGBEE, mac))
         self._attr_device_info = DeviceInfo(
             configuration_url=configuration_url,
             identifiers={(DOMAIN, device_id)},
             connections=connections,
+            manufacturer=data.get("vendor"),
+            model=data.get("model"),
+            model_id=data.get("model_id"),
+            name=coordinator.data.gateway["smile_name"],
+            sw_version=data.get("firmware"),
+            hw_version=data.get("hardware"),
         )
+        if device_id != coordinator.data.gateway["gateway_id"]:
             self._attr_device_info.update(
                 {
+                    ATTR_NAME: data.get("name"),
                     ATTR_VIA_DEVICE: (
                         DOMAIN,
+                        str(self.coordinator.data.gateway["gateway_id"]),
                     ),
                 }
             )
         """Return if entity is available."""
         return (
             self._dev_id in self.coordinator.data.devices
+            and ("available" not in self.device or self.device["available"] is True)
             and super().available
         )

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/icons.json RENAMED Viewed

@@ -120,8 +120,5 @@
         "default": "mdi:lock"
       }
     }
-  },
-  "services": {
-    "delete_notification": "mdi:trash-can"
   }
 }

         "default": "mdi:lock"
       }
     }
   }
 }

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/manifest.json RENAMED Viewed

@@ -1,13 +1,12 @@
 {
   "domain": "plugwise",
-  "name": "Plugwise Smile/Stretch Beta",
-  "after_dependencies": ["zeroconf"],
-  "codeowners": ["@CoMPaTech", "@bouwew"],
   "config_flow": true,
-  "documentation": "https://github.com/plugwise/plugwise-beta",
   "integration_type": "hub",
   "iot_class": "local_polling",
   "loggers": ["plugwise"],
-  "requirements": ["plugwise==0.38.3"],
-  "version": "0.51.3"
 }

 {
   "domain": "plugwise",
+  "name": "Plugwise",
+  "codeowners": ["@CoMPaTech", "@bouwew", "@frenck"],
   "config_flow": true,
+  "documentation": "https://www.home-assistant.io/integrations/plugwise",
   "integration_type": "hub",
   "iot_class": "local_polling",
   "loggers": ["plugwise"],
+  "requirements": ["plugwise==1.5.0"],
+  "zeroconf": ["_plugwise._tcp.local."]
 }

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/number.py RENAMED Viewed

@@ -1,4 +1,5 @@
 """Number platform for Plugwise integration."""
 from __future__ import annotations
 from dataclasses import dataclass
@@ -9,21 +10,12 @@
     NumberEntityDescription,
     NumberMode,
 )
-from homeassistant.const import ATTR_NAME, EntityCategory, UnitOfTemperature
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
-from .const import (
-    LOGGER,
-    LOWER_BOUND,
-    MAX_BOILER_TEMP,
-    MAX_DHW_TEMP,
-    RESOLUTION,
-    TEMPERATURE_OFFSET,
-    UPPER_BOUND,
-    NumberType,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
@@ -38,22 +30,22 @@
 NUMBER_TYPES = (
     PlugwiseNumberEntityDescription(
-        key=MAX_BOILER_TEMP,
-        translation_key=MAX_BOILER_TEMP,
         device_class=NumberDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.CONFIG,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
     ),
     PlugwiseNumberEntityDescription(
-        key=MAX_DHW_TEMP,
-        translation_key=MAX_DHW_TEMP,
         device_class=NumberDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.CONFIG,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
     ),
     PlugwiseNumberEntityDescription(
-        key=TEMPERATURE_OFFSET,
-        translation_key=TEMPERATURE_OFFSET,
         device_class=NumberDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.CONFIG,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
@@ -66,7 +58,7 @@
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
-    """Set up the Plugwise numbers from a ConfigEntry."""
     coordinator = entry.runtime_data
     @callback
@@ -75,22 +67,15 @@
         if not coordinator.new_devices:
             return
-        entities: list[PlugwiseNumberEntity] = []
-        for device_id, device in coordinator.data.devices.items():
-            for description in NUMBER_TYPES:
-                if description.key in device:
-                    entities.append(
-                        PlugwiseNumberEntity(coordinator, device_id, description)
-                    )
-                    LOGGER.debug(
-                        "Add %s %s number", device[ATTR_NAME], description.translation_key
-                    )
-        async_add_entities(entities)
-    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
     _add_entities()
 class PlugwiseNumberEntity(PlugwiseEntity, NumberEntity):
@@ -106,16 +91,15 @@
     ) -> None:
         """Initiate Plugwise Number."""
         super().__init__(coordinator, device_id)
-        self.actuator = self.device[description.key]
         self.device_id = device_id
         self.entity_description = description
         self._attr_unique_id = f"{device_id}-{description.key}"
         self._attr_mode = NumberMode.BOX
-        self._attr_native_max_value = self.device[description.key][UPPER_BOUND]
-        self._attr_native_min_value = self.device[description.key][LOWER_BOUND]
-        native_step = self.device[description.key][RESOLUTION]
-        if description.key != TEMPERATURE_OFFSET:
             native_step = max(native_step, 0.5)
         self._attr_native_step = native_step
@@ -127,7 +111,6 @@
     @plugwise_command
     async def async_set_native_value(self, value: float) -> None:
         """Change to the new setpoint value."""
-        await self.coordinator.api.set_number(self.device_id, self.entity_description.key, value)
-        LOGGER.debug(
-            "Setting %s to %s was successful", self.entity_description.key, value
         )

 """Number platform for Plugwise integration."""
 from __future__ import annotations
 from dataclasses import dataclass
     NumberEntityDescription,
     NumberMode,
 )
+from homeassistant.const import EntityCategory, UnitOfTemperature
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
+from .const import NumberType
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
 NUMBER_TYPES = (
     PlugwiseNumberEntityDescription(
+        key="maximum_boiler_temperature",
+        translation_key="maximum_boiler_temperature",
         device_class=NumberDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.CONFIG,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
     ),
     PlugwiseNumberEntityDescription(
+        key="max_dhw_temperature",
+        translation_key="max_dhw_temperature",
         device_class=NumberDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.CONFIG,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
     ),
     PlugwiseNumberEntityDescription(
+        key="temperature_offset",
+        translation_key="temperature_offset",
         device_class=NumberDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.CONFIG,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
+    """Set up Plugwise number platform."""
     coordinator = entry.runtime_data
     @callback
         if not coordinator.new_devices:
             return
+        async_add_entities(
+            PlugwiseNumberEntity(coordinator, device_id, description)
+            for device_id in coordinator.new_devices
+            for description in NUMBER_TYPES
+            if description.key in coordinator.data.devices[device_id]
+        )
     _add_entities()
+    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
 class PlugwiseNumberEntity(PlugwiseEntity, NumberEntity):
     ) -> None:
         """Initiate Plugwise Number."""
         super().__init__(coordinator, device_id)
         self.device_id = device_id
         self.entity_description = description
         self._attr_unique_id = f"{device_id}-{description.key}"
         self._attr_mode = NumberMode.BOX
+        self._attr_native_max_value = self.device[description.key]["upper_bound"]
+        self._attr_native_min_value = self.device[description.key]["lower_bound"]
+        native_step = self.device[description.key]["resolution"]
+        if description.key != "temperature_offset":
             native_step = max(native_step, 0.5)
         self._attr_native_step = native_step
     @plugwise_command
     async def async_set_native_value(self, value: float) -> None:
         """Change to the new setpoint value."""
+        await self.coordinator.api.set_number(
+            self.device_id, self.entity_description.key, value
         )

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/select.py RENAMED Viewed

@@ -1,37 +1,20 @@
 """Plugwise Select component for Home Assistant."""
 from __future__ import annotations
 from dataclasses import dataclass
 from homeassistant.components.select import SelectEntity, SelectEntityDescription
-from homeassistant.const import ATTR_NAME, STATE_ON, EntityCategory
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
-from .const import (
-    AVAILABLE_SCHEDULES,
-    DHW_MODE,
-    DHW_MODES,
-    GATEWAY_MODE,
-    GATEWAY_MODES,
-    LOCATION,
-    LOGGER,
-    REGULATION_MODE,
-    REGULATION_MODES,
-    SELECT_DHW_MODE,
-    SELECT_GATEWAY_MODE,
-    SELECT_REGULATION_MODE,
-    SELECT_SCHEDULE,
-    SelectOptionsType,
-    SelectType,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
-PARALLEL_UPDATES = 0
 @dataclass(frozen=True, kw_only=True)
 class PlugwiseSelectEntityDescription(SelectEntityDescription):
@@ -43,27 +26,27 @@
 SELECT_TYPES = (
     PlugwiseSelectEntityDescription(
-        key=SELECT_SCHEDULE,
-        translation_key=SELECT_SCHEDULE,
-        options_key=AVAILABLE_SCHEDULES,
     ),
     PlugwiseSelectEntityDescription(
-        key=SELECT_REGULATION_MODE,
-        translation_key=REGULATION_MODE,
         entity_category=EntityCategory.CONFIG,
-        options_key=REGULATION_MODES,
     ),
     PlugwiseSelectEntityDescription(
-        key=SELECT_DHW_MODE,
-        translation_key=DHW_MODE,
         entity_category=EntityCategory.CONFIG,
-        options_key=DHW_MODES,
     ),
     PlugwiseSelectEntityDescription(
-        key=SELECT_GATEWAY_MODE,
-        translation_key=GATEWAY_MODE,
         entity_category=EntityCategory.CONFIG,
-        options_key=GATEWAY_MODES,
     ),
 )
@@ -73,7 +56,7 @@
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
-    """Set up the Smile selector from a ConfigEntry."""
     coordinator = entry.runtime_data
     @callback
@@ -82,22 +65,15 @@
         if not coordinator.new_devices:
             return
-        entities: list[PlugwiseSelectEntity] = []
-        for device_id, device in coordinator.data.devices.items():
-            for description in SELECT_TYPES:
-                if description.options_key in device:
-                    entities.append(
-                        PlugwiseSelectEntity(coordinator, device_id, description)
-                    )
-                    LOGGER.debug(
-                        "Add %s %s selector", device[ATTR_NAME], description.translation_key
-                    )
-        async_add_entities(entities)
-    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
     _add_entities()
 class PlugwiseSelectEntity(PlugwiseEntity, SelectEntity):
@@ -135,8 +111,3 @@
         await self.coordinator.api.set_select(
             self.entity_description.key, self.device[LOCATION], option, STATE_ON
         )
-        LOGGER.debug(
-            "Set %s to %s was successful.",
-            self.entity_description.key,
-            option,
-        )

 """Plugwise Select component for Home Assistant."""
 from __future__ import annotations
 from dataclasses import dataclass
 from homeassistant.components.select import SelectEntity, SelectEntityDescription
+from homeassistant.const import STATE_ON, EntityCategory
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
+from .const import LOCATION, SelectOptionsType, SelectType
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
 @dataclass(frozen=True, kw_only=True)
 class PlugwiseSelectEntityDescription(SelectEntityDescription):
 SELECT_TYPES = (
     PlugwiseSelectEntityDescription(
+        key="select_schedule",
+        translation_key="select_schedule",
+        options_key="available_schedules",
     ),
     PlugwiseSelectEntityDescription(
+        key="select_regulation_mode",
+        translation_key="regulation_mode",
         entity_category=EntityCategory.CONFIG,
+        options_key="regulation_modes",
     ),
     PlugwiseSelectEntityDescription(
+        key="select_dhw_mode",
+        translation_key="dhw_mode",
         entity_category=EntityCategory.CONFIG,
+        options_key="dhw_modes",
     ),
     PlugwiseSelectEntityDescription(
+        key="select_gateway_mode",
+        translation_key="gateway_mode",
         entity_category=EntityCategory.CONFIG,
+        options_key="gateway_modes",
     ),
 )
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
+    """Set up the Smile selector from a config entry."""
     coordinator = entry.runtime_data
     @callback
         if not coordinator.new_devices:
             return
+        async_add_entities(
+            PlugwiseSelectEntity(coordinator, device_id, description)
+            for device_id in coordinator.new_devices
+            for description in SELECT_TYPES
+            if description.options_key in coordinator.data.devices[device_id]
+        )
     _add_entities()
+    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
 class PlugwiseSelectEntity(PlugwiseEntity, SelectEntity):
         await self.coordinator.api.set_select(
             self.entity_description.key, self.device[LOCATION], option, STATE_ON
         )

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/sensor.py RENAMED Viewed

@@ -1,4 +1,5 @@
 """Plugwise Sensor component for Home Assistant."""
 from __future__ import annotations
 from dataclasses import dataclass
@@ -12,8 +13,6 @@
     SensorStateClass,
 )
 from homeassistant.const import (
-    ATTR_NAME,
-    ATTR_TEMPERATURE,
     LIGHT_LUX,
     PERCENTAGE,
     EntityCategory,
@@ -29,60 +28,9 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
-from .const import (
-    DHW_SETPOINT,
-    DHW_TEMP,
-    EL_CONS_INTERVAL,
-    EL_CONS_OP_CUMULATIVE,
-    EL_CONS_OP_INTERVAL,
-    EL_CONS_OP_POINT,
-    EL_CONS_P_CUMULATIVE,
-    EL_CONS_P_INTERVAL,
-    EL_CONS_P_POINT,
-    EL_CONS_POINT,
-    EL_CONSUMED,
-    EL_PH1_CONSUMED,
-    EL_PH1_PRODUCED,
-    EL_PH2_CONSUMED,
-    EL_PH2_PRODUCED,
-    EL_PH3_CONSUMED,
-    EL_PH3_PRODUCED,
-    EL_PROD_INTERVAL,
-    EL_PROD_OP_CUMULATIVE,
-    EL_PROD_OP_INTERVAL,
-    EL_PROD_OP_POINT,
-    EL_PROD_P_CUMULATIVE,
-    EL_PROD_P_INTERVAL,
-    EL_PROD_P_POINT,
-    EL_PROD_POINT,
-    EL_PRODUCED,
-    GAS_CONS_CUMULATIVE,
-    GAS_CONS_INTERVAL,
-    INTENDED_BOILER_TEMP,
-    LOGGER,
-    MOD_LEVEL,
-    NET_EL_CUMULATIVE,
-    NET_EL_POINT,
-    OUTDOOR_AIR_TEMP,
-    OUTDOOR_TEMP,
-    RETURN_TEMP,
-    SENSORS,
-    TARGET_TEMP,
-    TARGET_TEMP_HIGH,
-    TARGET_TEMP_LOW,
-    TEMP_DIFF,
-    VALVE_POS,
-    VOLTAGE_PH1,
-    VOLTAGE_PH2,
-    VOLTAGE_PH3,
-    WATER_PRESSURE,
-    WATER_TEMP,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
-PARALLEL_UPDATES = 0
 @dataclass(frozen=True)
 class PlugwiseSensorEntityDescription(SensorEntityDescription):
@@ -91,17 +39,17 @@
     key: SensorType
-PLUGWISE_SENSORS: tuple[PlugwiseSensorEntityDescription, ...] = (
     PlugwiseSensorEntityDescription(
-        key=TARGET_TEMP,
-        translation_key=TARGET_TEMP,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
-        key=TARGET_TEMP_HIGH,
         translation_key="cooling_setpoint",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
@@ -109,7 +57,7 @@
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
-        key=TARGET_TEMP_LOW,
         translation_key="heating_setpoint",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
@@ -117,277 +65,276 @@
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
-        key=ATTR_TEMPERATURE,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=INTENDED_BOILER_TEMP,
-        translation_key=INTENDED_BOILER_TEMP,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=TEMP_DIFF,
-        translation_key=TEMP_DIFF,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=OUTDOOR_TEMP,
-        translation_key=OUTDOOR_TEMP,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         state_class=SensorStateClass.MEASUREMENT,
-        suggested_display_precision=1,
     ),
     PlugwiseSensorEntityDescription(
-        key=OUTDOOR_AIR_TEMP,
-        translation_key=OUTDOOR_AIR_TEMP,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=WATER_TEMP,
-        translation_key=WATER_TEMP,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=RETURN_TEMP,
-        translation_key=RETURN_TEMP,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONSUMED,
-        translation_key=EL_CONSUMED,
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PRODUCED,
-        translation_key=EL_PRODUCED,
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_INTERVAL,
-        translation_key=EL_CONS_INTERVAL,
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_P_INTERVAL,
-        translation_key=EL_CONS_P_INTERVAL,
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_OP_INTERVAL,
-        translation_key=EL_CONS_OP_INTERVAL,
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_INTERVAL,
-        translation_key=EL_PROD_INTERVAL,
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_P_INTERVAL,
-        translation_key=EL_PROD_P_INTERVAL,
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_OP_INTERVAL,
-        translation_key=EL_PROD_OP_INTERVAL,
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_POINT,
-        translation_key=EL_CONS_POINT,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_OP_POINT,
-        translation_key=EL_CONS_OP_POINT,
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_P_POINT,
-        translation_key=EL_CONS_P_POINT,
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_OP_CUMULATIVE,
-        translation_key=EL_CONS_OP_CUMULATIVE,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_CONS_P_CUMULATIVE,
-        translation_key=EL_CONS_P_CUMULATIVE,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_POINT,
-        translation_key=EL_PROD_POINT,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_OP_POINT,
-        translation_key=EL_PROD_OP_POINT,
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_P_POINT,
-        translation_key=EL_PROD_P_POINT,
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_OP_CUMULATIVE,
-        translation_key=EL_PROD_OP_CUMULATIVE,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PROD_P_CUMULATIVE,
-        translation_key=EL_PROD_P_CUMULATIVE,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PH1_CONSUMED,
-        translation_key=EL_PH1_CONSUMED,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PH2_CONSUMED,
-        translation_key=EL_PH2_CONSUMED,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PH3_CONSUMED,
-        translation_key=EL_PH3_CONSUMED,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PH1_PRODUCED,
-        translation_key=EL_PH1_PRODUCED,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PH2_PRODUCED,
-        translation_key=EL_PH2_PRODUCED,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=EL_PH3_PRODUCED,
-        translation_key=EL_PH3_PRODUCED,
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=VOLTAGE_PH1,
-        translation_key=VOLTAGE_PH1,
         device_class=SensorDeviceClass.VOLTAGE,
         native_unit_of_measurement=UnitOfElectricPotential.VOLT,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
-        key=VOLTAGE_PH2,
-        translation_key=VOLTAGE_PH2,
         device_class=SensorDeviceClass.VOLTAGE,
         native_unit_of_measurement=UnitOfElectricPotential.VOLT,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
-        key=VOLTAGE_PH3,
-        translation_key=VOLTAGE_PH3,
         device_class=SensorDeviceClass.VOLTAGE,
         native_unit_of_measurement=UnitOfElectricPotential.VOLT,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
-        key=GAS_CONS_INTERVAL,
-        translation_key=GAS_CONS_INTERVAL,
         native_unit_of_measurement=UnitOfVolumeFlowRate.CUBIC_METERS_PER_HOUR,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=GAS_CONS_CUMULATIVE,
-        translation_key=GAS_CONS_CUMULATIVE,
         native_unit_of_measurement=UnitOfVolume.CUBIC_METERS,
         device_class=SensorDeviceClass.GAS,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
-        key=NET_EL_POINT,
-        translation_key=NET_EL_POINT,
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=NET_EL_CUMULATIVE,
-        translation_key=NET_EL_CUMULATIVE,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
@@ -407,22 +354,22 @@
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
-        key=MOD_LEVEL,
-        translation_key=MOD_LEVEL,
         native_unit_of_measurement=PERCENTAGE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=VALVE_POS,
-        translation_key=VALVE_POS,
         entity_category=EntityCategory.DIAGNOSTIC,
         native_unit_of_measurement=PERCENTAGE,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=WATER_PRESSURE,
-        translation_key=WATER_PRESSURE,
         native_unit_of_measurement=UnitOfPressure.BAR,
         device_class=SensorDeviceClass.PRESSURE,
         entity_category=EntityCategory.DIAGNOSTIC,
@@ -435,16 +382,16 @@
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=DHW_TEMP,
-        translation_key=DHW_TEMP,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
-        key=DHW_SETPOINT,
-        translation_key=DHW_SETPOINT,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
@@ -458,7 +405,7 @@
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
-    """Set up the Smile sensors from a ConfigEntry."""
     coordinator = entry.runtime_data
     @callback
@@ -467,29 +414,16 @@
         if not coordinator.new_devices:
             return
-        entities: list[PlugwiseSensorEntity] = []
-        for device_id, device in coordinator.data.devices.items():
-            if not (sensors := device.get(SENSORS)):
-                continue
-            for description in PLUGWISE_SENSORS:
-                if description.key not in sensors:
-                    continue
-                entities.append(
-                    PlugwiseSensorEntity(
-                        coordinator,
-                        device_id,
-                        description,
-                    )
-                )
-                LOGGER.debug(
-                    "Add %s %s sensor", device[ATTR_NAME], description.translation_key or description.key
-                )
-        async_add_entities(entities)
-    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
     _add_entities()
 class PlugwiseSensorEntity(PlugwiseEntity, SensorEntity):
@@ -511,4 +445,4 @@
     @property
     def native_value(self) -> int | float:
         """Return the value reported by the sensor."""
-        return self.device[SENSORS][self.entity_description.key]

 """Plugwise Sensor component for Home Assistant."""
 from __future__ import annotations
 from dataclasses import dataclass
     SensorStateClass,
 )
 from homeassistant.const import (
     LIGHT_LUX,
     PERCENTAGE,
     EntityCategory,
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 @dataclass(frozen=True)
 class PlugwiseSensorEntityDescription(SensorEntityDescription):
     key: SensorType
+SENSORS: tuple[PlugwiseSensorEntityDescription, ...] = (
     PlugwiseSensorEntityDescription(
+        key="setpoint",
+        translation_key="setpoint",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
+        key="setpoint_high",
         translation_key="cooling_setpoint",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
+        key="setpoint_low",
         translation_key="heating_setpoint",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
+        key="temperature",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="intended_boiler_temperature",
+        translation_key="intended_boiler_temperature",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="temperature_difference",
+        translation_key="temperature_difference",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="outdoor_temperature",
+        translation_key="outdoor_temperature",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="outdoor_air_temperature",
+        translation_key="outdoor_air_temperature",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="water_temperature",
+        translation_key="water_temperature",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="return_temperature",
+        translation_key="return_temperature",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed",
+        translation_key="electricity_consumed",
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced",
+        translation_key="electricity_produced",
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_interval",
+        translation_key="electricity_consumed_interval",
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_peak_interval",
+        translation_key="electricity_consumed_peak_interval",
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_off_peak_interval",
+        translation_key="electricity_consumed_off_peak_interval",
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_interval",
+        translation_key="electricity_produced_interval",
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_peak_interval",
+        translation_key="electricity_produced_peak_interval",
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_off_peak_interval",
+        translation_key="electricity_produced_off_peak_interval",
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_point",
+        translation_key="electricity_consumed_point",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_off_peak_point",
+        translation_key="electricity_consumed_off_peak_point",
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_peak_point",
+        translation_key="electricity_consumed_peak_point",
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_off_peak_cumulative",
+        translation_key="electricity_consumed_off_peak_cumulative",
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_consumed_peak_cumulative",
+        translation_key="electricity_consumed_peak_cumulative",
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_point",
+        translation_key="electricity_produced_point",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_off_peak_point",
+        translation_key="electricity_produced_off_peak_point",
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_peak_point",
+        translation_key="electricity_produced_peak_point",
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_off_peak_cumulative",
+        translation_key="electricity_produced_off_peak_cumulative",
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_produced_peak_cumulative",
+        translation_key="electricity_produced_peak_cumulative",
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_phase_one_consumed",
+        translation_key="electricity_phase_one_consumed",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_phase_two_consumed",
+        translation_key="electricity_phase_two_consumed",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_phase_three_consumed",
+        translation_key="electricity_phase_three_consumed",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_phase_one_produced",
+        translation_key="electricity_phase_one_produced",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_phase_two_produced",
+        translation_key="electricity_phase_two_produced",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="electricity_phase_three_produced",
+        translation_key="electricity_phase_three_produced",
         device_class=SensorDeviceClass.POWER,
         native_unit_of_measurement=UnitOfPower.WATT,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="voltage_phase_one",
+        translation_key="voltage_phase_one",
         device_class=SensorDeviceClass.VOLTAGE,
         native_unit_of_measurement=UnitOfElectricPotential.VOLT,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
+        key="voltage_phase_two",
+        translation_key="voltage_phase_two",
         device_class=SensorDeviceClass.VOLTAGE,
         native_unit_of_measurement=UnitOfElectricPotential.VOLT,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
+        key="voltage_phase_three",
+        translation_key="voltage_phase_three",
         device_class=SensorDeviceClass.VOLTAGE,
         native_unit_of_measurement=UnitOfElectricPotential.VOLT,
         state_class=SensorStateClass.MEASUREMENT,
         entity_registry_enabled_default=False,
     ),
     PlugwiseSensorEntityDescription(
+        key="gas_consumed_interval",
+        translation_key="gas_consumed_interval",
         native_unit_of_measurement=UnitOfVolumeFlowRate.CUBIC_METERS_PER_HOUR,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="gas_consumed_cumulative",
+        translation_key="gas_consumed_cumulative",
         native_unit_of_measurement=UnitOfVolume.CUBIC_METERS,
         device_class=SensorDeviceClass.GAS,
         state_class=SensorStateClass.TOTAL,
     ),
     PlugwiseSensorEntityDescription(
+        key="net_electricity_point",
+        translation_key="net_electricity_point",
         native_unit_of_measurement=UnitOfPower.WATT,
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="net_electricity_cumulative",
+        translation_key="net_electricity_cumulative",
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY,
         state_class=SensorStateClass.TOTAL,
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     PlugwiseSensorEntityDescription(
+        key="modulation_level",
+        translation_key="modulation_level",
         native_unit_of_measurement=PERCENTAGE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="valve_position",
+        translation_key="valve_position",
         entity_category=EntityCategory.DIAGNOSTIC,
         native_unit_of_measurement=PERCENTAGE,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="water_pressure",
+        translation_key="water_pressure",
         native_unit_of_measurement=UnitOfPressure.BAR,
         device_class=SensorDeviceClass.PRESSURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="dhw_temperature",
+        translation_key="dhw_temperature",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
         state_class=SensorStateClass.MEASUREMENT,
     ),
     PlugwiseSensorEntityDescription(
+        key="domestic_hot_water_setpoint",
+        translation_key="domestic_hot_water_setpoint",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         device_class=SensorDeviceClass.TEMPERATURE,
         entity_category=EntityCategory.DIAGNOSTIC,
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
+    """Set up the Smile sensors from a config entry."""
     coordinator = entry.runtime_data
     @callback
         if not coordinator.new_devices:
             return
+        async_add_entities(
+            PlugwiseSensorEntity(coordinator, device_id, description)
+            for device_id in coordinator.new_devices
+            if (sensors := coordinator.data.devices[device_id].get("sensors"))
+            for description in SENSORS
+            if description.key in sensors
+        )
     _add_entities()
+    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
 class PlugwiseSensorEntity(PlugwiseEntity, SensorEntity):
     @property
     def native_value(self) -> int | float:
         """Return the value reported by the sensor."""
+        return self.device["sensors"][self.entity_description.key]

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/strings.json RENAMED Viewed

@@ -1,52 +1,38 @@
 {
-  "options": {
-    "step": {
-      "none": {
-        "title": "No Options available",
-        "description": "This Integration does not provide any Options"
-      },
-      "init": {
-        "description": "Adjust Smile/Stretch Options",
-        "data": {
-          "cooling_on": "Anna: cooling-mode is on",
-          "scan_interval": "Scan Interval (seconds) *) beta-only option",
-          "homekit_emulation": "Homekit emulation (i.e. on hvac_off => Away) *) beta-only option",
-          "refresh_interval": "Frontend refresh-time (1.5 - 5 seconds) *) beta-only option"
-        }
-      }
-    }
-  },
   "config": {
     "step": {
       "user": {
-        "title": "Connect to the Plugwise Adam/Smile/Stretch",
-        "description": "Please enter:",
         "data": {
-          "password": "ID",
-          "username": "Username",
-          "host": "IP-address",
-          "port": "Port number"
         }
       }
     },
     "error": {
-      "cannot_connect": "Failed to connect",
-      "invalid_auth": "Authentication failed",
       "invalid_setup": "Add your Adam instead of your Anna, see the documentation",
-      "network_down": "Plugwise Zigbee network is down",
-      "network_timeout": "Network communication timeout",
       "response_error": "Invalid XML data, or error indication received",
-      "stick_init": "Initialization of Plugwise USB-stick failed",
-      "unknown": "Unknown error!",
       "unsupported": "Device with unsupported firmware"
     },
     "abort": {
-      "already_configured": "This device is already configured",
       "anna_with_adam": "Both Anna and Adam detected. Add your Adam instead of your Anna"
     }
   },
   "entity": {
     "binary_sensor": {
       "compressor_state": {
         "name": "Compressor state"
       },
@@ -60,10 +46,10 @@
         "name": "Flame state"
       },
       "heating_state": {
-        "name": "Heating"
       },
       "cooling_state": {
-        "name": "Cooling"
       },
       "secondary_boiler_state": {
         "name": "Secondary boiler state"
@@ -80,14 +66,20 @@
     "climate": {
       "plugwise": {
         "state_attributes": {
           "preset_mode": {
             "state": {
               "asleep": "Night",
-              "away": "Away",
-              "home": "Home",
               "no_frost": "Anti-frost",
               "vacation": "Vacation"
             }
           }
         }
       }
@@ -107,20 +99,10 @@
       "dhw_mode": {
         "name": "DHW mode",
         "state": {
           "auto": "Auto",
-          "boost": "Boost",
-          "comfort": "Comfort",
-          "off": "Off"
-        }
-      },
-      "regulation_mode": {
-        "name": "Regulation mode",
-        "state": {
-          "bleeding_cold": "Bleeding cold",
-          "bleeding_hot": "Bleeding hot",
-          "cooling": "Cooling",
-          "heating": "Heating",
-          "off": "Off"
         }
       },
       "gateway_mode": {
@@ -131,6 +113,16 @@
           "vacation": "Vacation"
         }
       },
       "select_schedule": {
         "name": "Thermostat schedule",
         "state": {
@@ -146,7 +138,7 @@
         "name": "Cooling setpoint"
       },
       "heating_setpoint": {
-        "name": " Heating setpoint"
       },
       "intended_boiler_temperature": {
         "name": "Intended boiler temperature"
@@ -172,12 +164,6 @@
       "electricity_produced": {
         "name": "Electricity produced"
       },
-      "electricity_consumed_point": {
-        "name": "Electricity consumed point"
-      },
-      "electricity_produced_point": {
-        "name": "Electricity produced point"
-      },
       "electricity_consumed_interval": {
         "name": "Electricity consumed interval"
       },
@@ -196,6 +182,9 @@
       "electricity_produced_off_peak_interval": {
         "name": "Electricity produced off peak interval"
       },
       "electricity_consumed_off_peak_point": {
         "name": "Electricity consumed off peak point"
       },
@@ -208,6 +197,9 @@
       "electricity_consumed_peak_cumulative": {
         "name": "Electricity consumed peak cumulative"
       },
       "electricity_produced_off_peak_point": {
         "name": "Electricity produced off peak point"
       },
@@ -275,28 +267,22 @@
         "name": "DHW setpoint"
       },
       "maximum_boiler_temperature": {
-        "name": "Maximum boiler temperature setpoint"
       }
     },
     "switch": {
       "cooling_ena_switch": {
-        "name": "Cooling"
       },
       "dhw_cm_switch": {
         "name": "DHW comfort mode"
       },
       "lock": {
-        "name": "Lock"
       },
       "relay": {
         "name": "Relay"
       }
     }
-  },
-  "services": {
-    "delete_notification": {
-      "name": "Delete Plugwise notification",
-      "description": "Deletes a Plugwise Notification"
-    }
   }
 }

 {
   "config": {
     "step": {
       "user": {
+        "title": "Connect to the Smile",
+        "description": "Please enter",
         "data": {
+          "password": "Smile ID",
+          "host": "[%key:common::config_flow::data::ip%]",
+          "port": "[%key:common::config_flow::data::port%]",
+          "username": "Smile Username"
+        },
+        "data_description": {
+          "host": "Leave empty if using Auto Discovery"
         }
       }
     },
     "error": {
+      "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
+      "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
       "invalid_setup": "Add your Adam instead of your Anna, see the documentation",
       "response_error": "Invalid XML data, or error indication received",
+      "unknown": "[%key:common::config_flow::error::unknown%]",
       "unsupported": "Device with unsupported firmware"
     },
     "abort": {
+      "already_configured": "[%key:common::config_flow::abort::already_configured_service%]",
       "anna_with_adam": "Both Anna and Adam detected. Add your Adam instead of your Anna"
     }
   },
   "entity": {
     "binary_sensor": {
+      "low_battery": {
+        "name": "Battery state"
+      },
       "compressor_state": {
         "name": "Compressor state"
       },
         "name": "Flame state"
       },
       "heating_state": {
+        "name": "[%key:component::climate::entity_component::_::state_attributes::hvac_action::state::heating%]"
       },
       "cooling_state": {
+        "name": "[%key:component::climate::entity_component::_::state_attributes::hvac_action::state::cooling%]"
       },
       "secondary_boiler_state": {
         "name": "Secondary boiler state"
     "climate": {
       "plugwise": {
         "state_attributes": {
+          "available_schemas": {
+            "name": "Available schemas"
+          },
           "preset_mode": {
             "state": {
               "asleep": "Night",
+              "away": "[%key:component::climate::entity_component::_::state_attributes::preset_mode::state::away%]",
+              "home": "[%key:common::state::home%]",
               "no_frost": "Anti-frost",
               "vacation": "Vacation"
             }
+          },
+          "selected_schema": {
+            "name": "Selected schema"
           }
         }
       }
       "dhw_mode": {
         "name": "DHW mode",
         "state": {
+          "off": "[%key:common::state::off%]",
           "auto": "Auto",
+          "boost": "[%key:component::climate::entity_component::_::state_attributes::preset_mode::state::boost%]",
+          "comfort": "[%key:component::climate::entity_component::_::state_attributes::preset_mode::state::comfort%]"
         }
       },
       "gateway_mode": {
           "vacation": "Vacation"
         }
       },
+      "regulation_mode": {
+        "name": "Regulation mode",
+        "state": {
+          "bleeding_cold": "Bleeding cold",
+          "bleeding_hot": "Bleeding hot",
+          "cooling": "[%key:component::climate::entity_component::_::state_attributes::hvac_action::state::cooling%]",
+          "heating": "[%key:component::climate::entity_component::_::state_attributes::hvac_action::state::heating%]",
+          "off": "[%key:common::state::off%]"
+        }
+      },
       "select_schedule": {
         "name": "Thermostat schedule",
         "state": {
         "name": "Cooling setpoint"
       },
       "heating_setpoint": {
+        "name": "Heating setpoint"
       },
       "intended_boiler_temperature": {
         "name": "Intended boiler temperature"
       "electricity_produced": {
         "name": "Electricity produced"
       },
       "electricity_consumed_interval": {
         "name": "Electricity consumed interval"
       },
       "electricity_produced_off_peak_interval": {
         "name": "Electricity produced off peak interval"
       },
+      "electricity_consumed_point": {
+        "name": "Electricity consumed point"
+      },
       "electricity_consumed_off_peak_point": {
         "name": "Electricity consumed off peak point"
       },
       "electricity_consumed_peak_cumulative": {
         "name": "Electricity consumed peak cumulative"
       },
+      "electricity_produced_point": {
+        "name": "Electricity produced point"
+      },
       "electricity_produced_off_peak_point": {
         "name": "Electricity produced off peak point"
       },
         "name": "DHW setpoint"
       },
       "maximum_boiler_temperature": {
+        "name": "Maximum boiler temperature"
       }
     },
     "switch": {
       "cooling_ena_switch": {
+        "name": "[%key:component::climate::entity_component::_::state_attributes::hvac_action::state::cooling%]"
       },
       "dhw_cm_switch": {
         "name": "DHW comfort mode"
       },
       "lock": {
+        "name": "[%key:component::lock::title%]"
       },
       "relay": {
         "name": "Relay"
       }
     }
   }
 }

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/switch.py RENAMED Viewed

@@ -1,4 +1,5 @@
 """Plugwise Switch component for HomeAssistant."""
 from __future__ import annotations
 from dataclasses import dataclass
@@ -11,20 +12,11 @@
     SwitchEntity,
     SwitchEntityDescription,
 )
-from homeassistant.const import ATTR_NAME, EntityCategory
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
-from .const import (
-    COOLING_ENA_SWITCH,
-    DHW_CM_SWITCH,
-    LOCK,
-    LOGGER,
-    MEMBERS,
-    RELAY,
-    SWITCHES,
-)
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
@@ -37,28 +29,26 @@
     key: SwitchType
-PLUGWISE_SWITCHES: tuple[PlugwiseSwitchEntityDescription, ...] = (
     PlugwiseSwitchEntityDescription(
-        key=DHW_CM_SWITCH,
-        translation_key=DHW_CM_SWITCH,
-        device_class=SwitchDeviceClass.SWITCH,
         entity_category=EntityCategory.CONFIG,
     ),
     PlugwiseSwitchEntityDescription(
-        key=LOCK,
-        translation_key=LOCK,
-        device_class=SwitchDeviceClass.SWITCH,
         entity_category=EntityCategory.CONFIG,
     ),
     PlugwiseSwitchEntityDescription(
-        key=RELAY,
-        translation_key=RELAY,
         device_class=SwitchDeviceClass.SWITCH,
     ),
     PlugwiseSwitchEntityDescription(
-        key=COOLING_ENA_SWITCH,
-        translation_key=COOLING_ENA_SWITCH,
-        device_class=SwitchDeviceClass.SWITCH,
         entity_category=EntityCategory.CONFIG,
     ),
 )
@@ -69,7 +59,7 @@
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
-    """Set up the Smile switches from a ConfigEntry."""
     coordinator = entry.runtime_data
     @callback
@@ -78,23 +68,16 @@
         if not coordinator.new_devices:
             return
-        entities: list[PlugwiseSwitchEntity] = []
-        for device_id, device in coordinator.data.devices.items():
-            if not (switches := device.get(SWITCHES)):
-                continue
-            for description in PLUGWISE_SWITCHES:
-                if description.key not in switches:
-                    continue
-                entities.append(PlugwiseSwitchEntity(coordinator, device_id, description))
-                LOGGER.debug(
-                    "Add %s %s switch", device[ATTR_NAME], description.translation_key
-                )
-        async_add_entities(entities)
-    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
     _add_entities()
 class PlugwiseSwitchEntity(PlugwiseEntity, SwitchEntity):
@@ -116,14 +99,14 @@
     @property
     def is_on(self) -> bool:
         """Return True if entity is on."""
-        return self.device[SWITCHES][self.entity_description.key]
     @plugwise_command
     async def async_turn_on(self, **kwargs: Any) -> None:
         """Turn the device on."""
         await self.coordinator.api.set_switch_state(
             self._dev_id,
-            self.device.get(MEMBERS),
             self.entity_description.key,
             "on",
         )
@@ -133,7 +116,7 @@
         """Turn the device off."""
         await self.coordinator.api.set_switch_state(
             self._dev_id,
-            self.device.get(MEMBERS),
             self.entity_description.key,
             "off",
         )

 """Plugwise Switch component for HomeAssistant."""
 from __future__ import annotations
 from dataclasses import dataclass
     SwitchEntity,
     SwitchEntityDescription,
 )
+from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from . import PlugwiseConfigEntry
 from .coordinator import PlugwiseDataUpdateCoordinator
 from .entity import PlugwiseEntity
 from .util import plugwise_command
     key: SwitchType
+SWITCHES: tuple[PlugwiseSwitchEntityDescription, ...] = (
     PlugwiseSwitchEntityDescription(
+        key="dhw_cm_switch",
+        translation_key="dhw_cm_switch",
         entity_category=EntityCategory.CONFIG,
     ),
     PlugwiseSwitchEntityDescription(
+        key="lock",
+        translation_key="lock",
         entity_category=EntityCategory.CONFIG,
     ),
     PlugwiseSwitchEntityDescription(
+        key="relay",
+        translation_key="relay",
         device_class=SwitchDeviceClass.SWITCH,
     ),
     PlugwiseSwitchEntityDescription(
+        key="cooling_ena_switch",
+        translation_key="cooling_ena_switch",
+        name="Cooling",
         entity_category=EntityCategory.CONFIG,
     ),
 )
     entry: PlugwiseConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
+    """Set up the Smile switches from a config entry."""
     coordinator = entry.runtime_data
     @callback
         if not coordinator.new_devices:
             return
+        async_add_entities(
+            PlugwiseSwitchEntity(coordinator, device_id, description)
+            for device_id in coordinator.new_devices
+            if (switches := coordinator.data.devices[device_id].get("switches"))
+            for description in SWITCHES
+            if description.key in switches
+        )
     _add_entities()
+    entry.async_on_unload(coordinator.async_add_listener(_add_entities))
 class PlugwiseSwitchEntity(PlugwiseEntity, SwitchEntity):
     @property
     def is_on(self) -> bool:
         """Return True if entity is on."""
+        return self.device["switches"][self.entity_description.key]
     @plugwise_command
     async def async_turn_on(self, **kwargs: Any) -> None:
         """Turn the device on."""
         await self.coordinator.api.set_switch_state(
             self._dev_id,
+            self.device.get("members"),
             self.entity_description.key,
             "on",
         )
         """Turn the device off."""
         await self.coordinator.api.set_switch_state(
             self._dev_id,
+            self.device.get("members"),
             self.entity_description.key,
             "off",
         )

/home/runner/work/progress/progress/clones/beta/{beta/custom_components → ha-core/homeassistant/components}/plugwise/util.py RENAMED Viewed

@@ -1,8 +1,7 @@
 """Utilities for Plugwise."""
-from __future__ import annotations
 from collections.abc import Awaitable, Callable, Coroutine
-from typing import Any, Concatenate, ParamSpec, TypeVar
 from plugwise.exceptions import PlugwiseException
@@ -10,13 +9,9 @@
 from .entity import PlugwiseEntity
-_PlugwiseEntityT = TypeVar("_PlugwiseEntityT", bound=PlugwiseEntity)
-_R = TypeVar("_R")
-_P = ParamSpec("_P")
-def plugwise_command(
-    func: Callable[Concatenate[_PlugwiseEntityT, _P], Awaitable[_R]]
 ) -> Callable[Concatenate[_PlugwiseEntityT, _P], Coroutine[Any, Any, _R]]:
     """Decorate Plugwise calls that send commands/make changes to the device.

 """Utilities for Plugwise."""
 from collections.abc import Awaitable, Callable, Coroutine
+from typing import Any, Concatenate
 from plugwise.exceptions import PlugwiseException
 from .entity import PlugwiseEntity
+def plugwise_command[_PlugwiseEntityT: PlugwiseEntity, **_P, _R](
+    func: Callable[Concatenate[_PlugwiseEntityT, _P], Awaitable[_R]],
 ) -> Callable[Concatenate[_PlugwiseEntityT, _P], Coroutine[Any, Any, _R]]:
     """Decorate Plugwise calls that send commands/make changes to the device.