Spring cleaning ☀️ (#23)

* Spring cleaning

* Actions

* Fix branches

* Changes for config_flow

custom_components/blueprint/binary_sensor.py

@@ -1,73 +1,28 @@
 """Binary sensor platform for blueprint."""
 from homeassistant.components.binary_sensor import BinarySensorDevice
-from .const import (
-    ATTRIBUTION,
+
+from custom_components.blueprint.const import (
+    BINARY_SENSOR,
     BINARY_SENSOR_DEVICE_CLASS,
     DEFAULT_NAME,
-    DOMAIN_DATA,
     DOMAIN,
 )
+from custom_components.blueprint.entity import BlueprintEntity
 
 
-async def async_setup_platform(
-    hass, config, async_add_entities, discovery_info=None
-):  # pylint: disable=unused-argument
+async def async_setup_entry(hass, entry, async_add_devices):
     """Setup binary_sensor platform."""
-    async_add_entities([BlueprintBinarySensor(hass, discovery_info)], True)
+    coordinator = hass.data[DOMAIN][entry.entry_id]
+    async_add_devices([BlueprintBinarySensor(coordinator, entry)])
 
 
-async def async_setup_entry(hass, config_entry, async_add_devices):
-    """Setup sensor platform."""
-    async_add_devices([BlueprintBinarySensor(hass, {})], True)
-
-
-class BlueprintBinarySensor(BinarySensorDevice):
+class BlueprintBinarySensor(BlueprintEntity, BinarySensorDevice):
     """blueprint binary_sensor class."""
 
-    def __init__(self, hass, config):
-        self.hass = hass
-        self.attr = {}
-        self._status = False
-        self._name = config.get("name", DEFAULT_NAME)
-
-    async def async_update(self):
-        """Update the binary_sensor."""
-        # Send update "signal" to the component
-        await self.hass.data[DOMAIN_DATA]["client"].update_data()
-
-        # Get new data (if any)
-        updated = self.hass.data[DOMAIN_DATA]["data"].get("data", {})
-
-        # Check the data and update the value.
-        if updated.get("bool_on") is None:
-            self._status = self._status
-        else:
-            self._status = updated.get("bool_on")
-
-        # Set/update attributes
-        self.attr["attribution"] = ATTRIBUTION
-        self.attr["time"] = str(updated.get("time"))
-        self.attr["static"] = updated.get("static")
-
-    @property
-    def unique_id(self):
-        """Return a unique ID to use for this binary_sensor."""
-        return (
-            "0919a0cd-745c-48fd"
-        )  # Don't hard code this, use something from the device/service.
-
-    @property
-    def device_info(self):
-        return {
-            "identifiers": {(DOMAIN, self.unique_id)},
-            "name": self.name,
-            "manufacturer": "Blueprint",
-        }
-
     @property
     def name(self):
         """Return the name of the binary_sensor."""
-        return self._name
+        return f"{DEFAULT_NAME}_{BINARY_SENSOR}"
 
     @property
     def device_class(self):
@@ -77,9 +32,4 @@ class BlueprintBinarySensor(BinarySensorDevice):
     @property
     def is_on(self):
         """Return true if the binary_sensor is on."""
-        return self._status
-
-    @property
-    def device_state_attributes(self):
-        """Return the state attributes."""
-        return self.attr
+        return self.coordinator.data.get("bool_on", False)