mirror of
https://github.com/home-assistant/core.git
synced 2025-04-27 10:47:51 +00:00
ca0648afe8
* Fix powerwall units (kW) * Fix test Co-authored-by: Paulus Schoutsen <balloob@gmail.com>
123 lines
3.4 KiB
Python
123 lines
3.4 KiB
Python
"""Support for August sensors."""
|
|
import logging
|
|
|
|
from homeassistant.const import (
|
|
DEVICE_CLASS_BATTERY,
|
|
DEVICE_CLASS_POWER,
|
|
UNIT_PERCENTAGE,
|
|
)
|
|
|
|
from .const import (
|
|
ATTR_ENERGY_EXPORTED,
|
|
ATTR_ENERGY_IMPORTED,
|
|
ATTR_FREQUENCY,
|
|
ATTR_INSTANT_AVERAGE_VOLTAGE,
|
|
DOMAIN,
|
|
ENERGY_KILO_WATT,
|
|
POWERWALL_API_CHARGE,
|
|
POWERWALL_API_DEVICE_TYPE,
|
|
POWERWALL_API_METERS,
|
|
POWERWALL_API_SITE_INFO,
|
|
POWERWALL_API_STATUS,
|
|
POWERWALL_COORDINATOR,
|
|
)
|
|
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_API_SITE_INFO]
|
|
device_type = powerwall_data[POWERWALL_API_DEVICE_TYPE]
|
|
status = powerwall_data[POWERWALL_API_STATUS]
|
|
|
|
entities = []
|
|
for meter in coordinator.data[POWERWALL_API_METERS]:
|
|
entities.append(
|
|
PowerWallEnergySensor(meter, coordinator, site_info, status, device_type)
|
|
)
|
|
|
|
entities.append(PowerWallChargeSensor(coordinator, site_info, status, device_type))
|
|
|
|
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, status, device_type):
|
|
"""Initialize the sensor."""
|
|
super().__init__(coordinator, site_info, status, device_type)
|
|
self._meter = meter
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measurement."""
|
|
return ENERGY_KILO_WATT
|
|
|
|
@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 kW."""
|
|
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,
|
|
}
|