Add device_info

Added device_info so the switch gets assigned to the device properly
apetrycki · Apr 17, 2024 · 8d5523e · 8d5523e
1 parent 9495e17
commit 8d5523e
Show file tree

Hide file tree

Showing 9 changed files with 6 additions and 1 deletion.
diff --git a/custom_components/daikinskyport/__pycache__/__init__.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/__init__.cpython-312.pyc
diff --git a/custom_components/daikinskyport/__pycache__/climate.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/climate.cpython-312.pyc
diff --git a/custom_components/daikinskyport/__pycache__/config_flow.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/config_flow.cpython-312.pyc
diff --git a/custom_components/daikinskyport/__pycache__/const.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/const.cpython-312.pyc
diff --git a/custom_components/daikinskyport/__pycache__/daikinskyport.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/daikinskyport.cpython-312.pyc
diff --git a/custom_components/daikinskyport/__pycache__/sensor.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/sensor.cpython-312.pyc
diff --git a/custom_components/daikinskyport/__pycache__/switch.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/switch.cpython-312.pyc
diff --git a/custom_components/daikinskyport/__pycache__/weather.cpython-312.pyc b/custom_components/daikinskyport/__pycache__/weather.cpython-312.pyc
diff --git a/custom_components/daikinskyport/switch.py b/custom_components/daikinskyport/switch.py
@@ -4,6 +4,7 @@
 from homeassistant.components.switch import SwitchEntity
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.exceptions import HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
@@ -20,7 +21,7 @@
 async def async_setup_entry(
     hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
 ) -> None:
-    """Add a Daikin Skyport Weather entity from a config_entry."""
+    """Add a Daikin Skyport Switch entity from a config_entry."""
 
     data = hass.data[DOMAIN][entry.entry_id]
     coordinator: DaikinSkyportData = data[COORDINATOR]
@@ -69,6 +70,10 @@ def turn_off(self, **kwargs: Any) -> None:
         self.aux_on = False
         self.async_write_ha_state()
 
+    @property
+    def device_info(self) -> DeviceInfo:
+        return self.data.device_info
+
     async def async_update(self) -> None:
         """Get the latest state of the switch."""
         _LOGGER.debug("Updating switch entity")