Add Powerwalls integration (#32851)

* Create an integration for Powerwalls * Self review :: remove unused code * Remove debug * Update homeassistant/components/powerwall/__init__.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * _call_site_info to module level * Update homeassistant/components/powerwall/binary_sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/binary_sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/binary_sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update tests/components/powerwall/test_binary_sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/binary_sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/binary_sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * remove sensors that I added tests for from the comment * Update homeassistant/components/powerwall/config_flow.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Update homeassistant/components/powerwall/sensor.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> * Switch to UNIT_PERCENTAGE * reduce code * Add test for import * Adjust tests * Add missing file Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
2020-03-19 10:50:17 -05:00 · 2020-03-19 10:50:17 -05:00 · 5b4d2aed64
commit 5b4d2aed64
parent 242aff9269
20 changed files with 990 additions and 0 deletions
--- a/homeassistant/components/powerwall/sensor.py
+++ b/homeassistant/components/powerwall/sensor.py
@ -0,0 +1,116 @@
+"""Support for August sensors."""
+import logging
+
+from homeassistant.const import (
+    DEVICE_CLASS_BATTERY,
+    DEVICE_CLASS_POWER,
+    ENERGY_KILO_WATT_HOUR,
+    UNIT_PERCENTAGE,
+)
+
+from .const import (
+    ATTR_ENERGY_EXPORTED,
+    ATTR_ENERGY_IMPORTED,
+    ATTR_FREQUENCY,
+    ATTR_INSTANT_AVERAGE_VOLTAGE,
+    DOMAIN,
+    POWERWALL_API_CHARGE,
+    POWERWALL_API_METERS,
+    POWERWALL_COORDINATOR,
+    POWERWALL_SITE_INFO,
+)
+from .entity import PowerWallEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    """Set up the August sensors."""
+    powerwall_data = hass.data[DOMAIN][config_entry.entry_id]
+    _LOGGER.debug("Powerwall_data: %s", powerwall_data)
+
+    coordinator = powerwall_data[POWERWALL_COORDINATOR]
+    site_info = powerwall_data[POWERWALL_SITE_INFO]
+
+    entities = []
+    for meter in coordinator.data[POWERWALL_API_METERS]:
+        entities.append(PowerWallEnergySensor(meter, coordinator, site_info))
+
+    entities.append(PowerWallChargeSensor(coordinator, site_info))
+
+    async_add_entities(entities, True)
+
+
+class PowerWallChargeSensor(PowerWallEntity):
+    """Representation of an Powerwall charge sensor."""
+
+    @property
+    def unit_of_measurement(self):
+        """Return the unit of measurement."""
+        return UNIT_PERCENTAGE
+
+    @property
+    def name(self):
+        """Device Name."""
+        return "Powerwall Charge"
+
+    @property
+    def device_class(self):
+        """Device Class."""
+        return DEVICE_CLASS_BATTERY
+
+    @property
+    def unique_id(self):
+        """Device Uniqueid."""
+        return f"{self.base_unique_id}_charge"
+
+    @property
+    def state(self):
+        """Get the current value in percentage."""
+        return round(self._coordinator.data[POWERWALL_API_CHARGE], 3)
+
+
+class PowerWallEnergySensor(PowerWallEntity):
+    """Representation of an Powerwall Energy sensor."""
+
+    def __init__(self, meter, coordinator, site_info):
+        """Initialize the sensor."""
+        super().__init__(coordinator, site_info)
+        self._meter = meter
+
+    @property
+    def unit_of_measurement(self):
+        """Return the unit of measurement."""
+        return ENERGY_KILO_WATT_HOUR
+
+    @property
+    def name(self):
+        """Device Name."""
+        return f"Powerwall {self._meter.title()} Now"
+
+    @property
+    def device_class(self):
+        """Device Class."""
+        return DEVICE_CLASS_POWER
+
+    @property
+    def unique_id(self):
+        """Device Uniqueid."""
+        return f"{self.base_unique_id}_{self._meter}_instant_power"
+
+    @property
+    def state(self):
+        """Get the current value in kWh."""
+        meter = self._coordinator.data[POWERWALL_API_METERS][self._meter]
+        return round(float(meter.instant_power / 1000), 3)
+
+    @property
+    def device_state_attributes(self):
+        """Return the device specific state attributes."""
+        meter = self._coordinator.data[POWERWALL_API_METERS][self._meter]
+        return {
+            ATTR_FREQUENCY: meter.frequency,
+            ATTR_ENERGY_EXPORTED: meter.energy_exported,
+            ATTR_ENERGY_IMPORTED: meter.energy_imported,
+            ATTR_INSTANT_AVERAGE_VOLTAGE: meter.instant_average_voltage,
+        }