Spring cleaning ☀️ (#23)

* Spring cleaning

* Actions

* Fix branches

* Changes for config_flow

custom_components/blueprint/switch.py

@@ -1,72 +1,34 @@
 """Switch platform for blueprint."""
 from homeassistant.components.switch import SwitchDevice
-from .const import ATTRIBUTION, DEFAULT_NAME, DOMAIN_DATA, ICON, DOMAIN
+
+from custom_components.blueprint.const import DEFAULT_NAME, DOMAIN, ICON, SWITCH
+
+from custom_components.blueprint.entity import BlueprintEntity
 
 
-async def async_setup_platform(
-    hass, config, async_add_entities, discovery_info=None
-):  # pylint: disable=unused-argument
-    """Setup switch platform."""
-    async_add_entities([BlueprintBinarySwitch(hass, discovery_info)], True)
-
-
-async def async_setup_entry(hass, config_entry, async_add_devices):
+async def async_setup_entry(hass, entry, async_add_devices):
     """Setup sensor platform."""
-    async_add_devices([BlueprintBinarySwitch(hass, {})], True)
+    coordinator = hass.data[DOMAIN][entry.entry_id]
+    async_add_devices([BlueprintBinarySwitch(coordinator, entry)])
 
 
-class BlueprintBinarySwitch(SwitchDevice):
+class BlueprintBinarySwitch(BlueprintEntity, SwitchDevice):
     """blueprint switch 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 switch."""
-        # 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.
-        self._status = self.hass.data[DOMAIN_DATA]["client"].client.something
-
-        # Set/update attributes
-        self.attr["attribution"] = ATTRIBUTION
-        self.attr["time"] = str(updated.get("time"))
-        self.attr["static"] = updated.get("static")
-
     async def async_turn_on(self, **kwargs):  # pylint: disable=unused-argument
         """Turn on the switch."""
-        await self.hass.data[DOMAIN_DATA]["client"].client.change_something(True)
+        await self.coordinator.api.async_change_something(True)
+        await self.coordinator.async_request_refresh()
 
     async def async_turn_off(self, **kwargs):  # pylint: disable=unused-argument
         """Turn off the switch."""
-        await self.hass.data[DOMAIN_DATA]["client"].client.change_something(False)
-
-    @property
-    def unique_id(self):
-        """Return a unique ID to use for this switch."""
-        return (
-            "0818a0cd-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",
-        }
+        await self.coordinator.api.async_change_something(False)
+        await self.coordinator.async_request_refresh()
 
     @property
     def name(self):
         """Return the name of the switch."""
-        return self._name
+        return f"{DEFAULT_NAME}_{SWITCH}"
 
     @property
     def icon(self):
@@ -76,9 +38,4 @@ class BlueprintBinarySwitch(SwitchDevice):
     @property
     def is_on(self):
         """Return true if the switch is on."""
-        return self._status
-
-    @property
-    def device_state_attributes(self):
-        """Return the state attributes."""
-        return self.attr
+        return self.coordinator.api.something