Merge pull request #2 from tellerbop/dev

Merge dev to master for v1.11
tellerbop · May 26, 2020 · 14a4077 · 14a4077
2 parents 446f1dc + 752895e
commit 14a4077
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/custom_components/vistapool/binary_sensor.py b/custom_components/vistapool/binary_sensor.py
@@ -1,7 +1,7 @@
 """Support for Vista pool sensors."""
 import logging
 
-from homeassistant.components.binary_sensor import DEVICE_CLASSES, BinarySensorDevice
+from homeassistant.components.binary_sensor import DEVICE_CLASSES, BinarySensorEntity
 from homeassistant.const import CONF_USERNAME
 
 from .vistapool_entity import VistaPoolEntity
@@ -27,7 +27,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
     async_add_entities(sensors)
 
 
-class VistaPoolSensor(VistaPoolEntity, BinarySensorDevice):
+class VistaPoolSensor(VistaPoolEntity, BinarySensorEntity):
     """Representation of an Vista Pool sensor."""
 
     @property

diff --git a/custom_components/vistapool/vistapool_entity.py b/custom_components/vistapool/vistapool_entity.py
@@ -69,7 +69,8 @@ def unique_id(self):
     @property
     def device_info(self):
         return {
-            "identifiers": {(DOMAIN, self._instrument.name)},            
-            "name": self.name,
+            "identifiers": {(DOMAIN, self._instrument.pool_name)},            
+            "manufacturer": "VistaPool",
+            "name": self._pool_name,
             "device_type": self._component,
         }