add garden device to group entities (#10)

custom_components/aerogarden/binary_sensor.py

@@ -1,6 +1,7 @@
 import logging
 
 from homeassistant.components.binary_sensor import BinarySensorEntity
+from homeassistant.helpers.entity import DeviceInfo
 
 from .. import aerogarden
 
@@ -48,6 +49,18 @@
         self._aerogarden.update()
         self._state = self._aerogarden.garden_property(self._macaddr, self._field)
 
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return the device info."""
+        return DeviceInfo(
+            identifiers={
+                # Serial numbers are unique identifiers within a specific domain
+                self._macaddr
+            },
+            name=self._garden_name,
+            model=self._aerogarden.garden_property,
+        )
+
 
 def setup_platform(hass, config, add_entities, discovery_info=None):
     ag = hass.data[aerogarden.DOMAIN]

custom_components/aerogarden/light.py

@@ -1,6 +1,7 @@
 import logging
 
 from homeassistant.components.light import LightEntity
+from homeassistant.helpers.entity import DeviceInfo
 
 from .. import aerogarden
 
@@ -54,6 +55,18 @@
         self._aerogarden.update()
         self._state = self._aerogarden.garden_property(self._macaddr, self._field)
 
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return the device info."""
+        return DeviceInfo(
+            identifiers={
+                # Serial numbers are unique identifiers within a specific domain
+                self._macaddr
+            },
+            name=self._garden_name,
+            model=self._aerogarden.garden_property,
+        )
+
 
 def setup_platform(hass, config, add_devices, discovery_info=None):
     """Setup the Aerogarden platform"""

custom_components/aerogarden/sensor.py

@@ -1,6 +1,6 @@
 import logging
 
-from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.entity import DeviceInfo, Entity
 
 from .. import aerogarden
 
@@ -54,6 +54,18 @@
         self._aerogarden.update()
         self._state = self._aerogarden.garden_property(self._macaddr, self._field)
 
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return the device info."""
+        return DeviceInfo(
+            identifiers={
+                # Serial numbers are unique identifiers within a specific domain
+                self._macaddr
+            },
+            name=self._garden_name,
+            model=self._aerogarden.garden_property,
+        )
+
 
 def setup_platform(hass, config, add_entities, discovery_info=None):
     """Setup the aerogarden platform"""