mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 05:07:41 +00:00
Update powerwall to use CoordinatorEntity (#39389)
This commit is contained in:
parent
ba75856f2b
commit
f7d1cfb625
@ -74,7 +74,7 @@ class PowerWallRunningSensor(PowerWallEntity, BinarySensorEntity):
|
|||||||
@property
|
@property
|
||||||
def is_on(self):
|
def is_on(self):
|
||||||
"""Get the powerwall running state."""
|
"""Get the powerwall running state."""
|
||||||
return self._coordinator.data[POWERWALL_API_SITEMASTER].running
|
return self.coordinator.data[POWERWALL_API_SITEMASTER].running
|
||||||
|
|
||||||
|
|
||||||
class PowerWallConnectedSensor(PowerWallEntity, BinarySensorEntity):
|
class PowerWallConnectedSensor(PowerWallEntity, BinarySensorEntity):
|
||||||
@ -98,7 +98,7 @@ class PowerWallConnectedSensor(PowerWallEntity, BinarySensorEntity):
|
|||||||
@property
|
@property
|
||||||
def is_on(self):
|
def is_on(self):
|
||||||
"""Get the powerwall connected to tesla state."""
|
"""Get the powerwall connected to tesla state."""
|
||||||
return self._coordinator.data[POWERWALL_API_SITEMASTER].connected_to_tesla
|
return self.coordinator.data[POWERWALL_API_SITEMASTER].connected_to_tesla
|
||||||
|
|
||||||
|
|
||||||
class PowerWallGridStatusSensor(PowerWallEntity, BinarySensorEntity):
|
class PowerWallGridStatusSensor(PowerWallEntity, BinarySensorEntity):
|
||||||
@ -122,7 +122,7 @@ class PowerWallGridStatusSensor(PowerWallEntity, BinarySensorEntity):
|
|||||||
@property
|
@property
|
||||||
def is_on(self):
|
def is_on(self):
|
||||||
"""Grid is online."""
|
"""Grid is online."""
|
||||||
return self._coordinator.data[POWERWALL_API_GRID_STATUS] == GridStatus.CONNECTED
|
return self.coordinator.data[POWERWALL_API_GRID_STATUS] == GridStatus.CONNECTED
|
||||||
|
|
||||||
|
|
||||||
class PowerWallChargingStatusSensor(PowerWallEntity, BinarySensorEntity):
|
class PowerWallChargingStatusSensor(PowerWallEntity, BinarySensorEntity):
|
||||||
@ -147,6 +147,6 @@ class PowerWallChargingStatusSensor(PowerWallEntity, BinarySensorEntity):
|
|||||||
def is_on(self):
|
def is_on(self):
|
||||||
"""Powerwall is charging."""
|
"""Powerwall is charging."""
|
||||||
# is_sending_to returns true for values greater than 100 watts
|
# is_sending_to returns true for values greater than 100 watts
|
||||||
return self._coordinator.data[POWERWALL_API_METERS][
|
return self.coordinator.data[POWERWALL_API_METERS][
|
||||||
POWERWALL_BATTERY_METER
|
POWERWALL_BATTERY_METER
|
||||||
].is_sending_to()
|
].is_sending_to()
|
||||||
|
@ -1,19 +1,18 @@
|
|||||||
"""The Tesla Powerwall integration base entity."""
|
"""The Tesla Powerwall integration base entity."""
|
||||||
|
|
||||||
from homeassistant.helpers.entity import Entity
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||||||
|
|
||||||
from .const import DOMAIN, MANUFACTURER, MODEL
|
from .const import DOMAIN, MANUFACTURER, MODEL
|
||||||
|
|
||||||
|
|
||||||
class PowerWallEntity(Entity):
|
class PowerWallEntity(CoordinatorEntity):
|
||||||
"""Base class for powerwall entities."""
|
"""Base class for powerwall entities."""
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self, coordinator, site_info, status, device_type, powerwalls_serial_numbers
|
self, coordinator, site_info, status, device_type, powerwalls_serial_numbers
|
||||||
):
|
):
|
||||||
"""Initialize the sensor."""
|
"""Initialize the sensor."""
|
||||||
super().__init__()
|
super().__init__(coordinator)
|
||||||
self._coordinator = coordinator
|
|
||||||
self._site_info = site_info
|
self._site_info = site_info
|
||||||
self._device_type = device_type
|
self._device_type = device_type
|
||||||
self._version = status.version
|
self._version = status.version
|
||||||
@ -33,26 +32,3 @@ class PowerWallEntity(Entity):
|
|||||||
device_info["model"] = model
|
device_info["model"] = model
|
||||||
device_info["sw_version"] = self._version
|
device_info["sw_version"] = self._version
|
||||||
return device_info
|
return device_info
|
||||||
|
|
||||||
@property
|
|
||||||
def available(self):
|
|
||||||
"""Return True if entity is available."""
|
|
||||||
return self._coordinator.last_update_success
|
|
||||||
|
|
||||||
@property
|
|
||||||
def should_poll(self):
|
|
||||||
"""Return False, updates are controlled via coordinator."""
|
|
||||||
return False
|
|
||||||
|
|
||||||
async def async_update(self):
|
|
||||||
"""Update the entity.
|
|
||||||
|
|
||||||
Only used by the generic entity update service.
|
|
||||||
"""
|
|
||||||
await self._coordinator.async_request_refresh()
|
|
||||||
|
|
||||||
async def async_added_to_hass(self):
|
|
||||||
"""Subscribe to updates."""
|
|
||||||
self.async_on_remove(
|
|
||||||
self._coordinator.async_add_listener(self.async_write_ha_state)
|
|
||||||
)
|
|
||||||
|
@ -89,7 +89,7 @@ class PowerWallChargeSensor(PowerWallEntity):
|
|||||||
@property
|
@property
|
||||||
def state(self):
|
def state(self):
|
||||||
"""Get the current value in percentage."""
|
"""Get the current value in percentage."""
|
||||||
return self._coordinator.data[POWERWALL_API_CHARGE]
|
return self.coordinator.data[POWERWALL_API_CHARGE]
|
||||||
|
|
||||||
|
|
||||||
class PowerWallEnergySensor(PowerWallEntity):
|
class PowerWallEnergySensor(PowerWallEntity):
|
||||||
@ -134,7 +134,7 @@ class PowerWallEnergySensor(PowerWallEntity):
|
|||||||
def state(self):
|
def state(self):
|
||||||
"""Get the current value in kW."""
|
"""Get the current value in kW."""
|
||||||
return (
|
return (
|
||||||
self._coordinator.data[POWERWALL_API_METERS]
|
self.coordinator.data[POWERWALL_API_METERS]
|
||||||
.get(self._meter)
|
.get(self._meter)
|
||||||
.get_power(precision=3)
|
.get_power(precision=3)
|
||||||
)
|
)
|
||||||
@ -142,7 +142,7 @@ class PowerWallEnergySensor(PowerWallEntity):
|
|||||||
@property
|
@property
|
||||||
def device_state_attributes(self):
|
def device_state_attributes(self):
|
||||||
"""Return the device specific state attributes."""
|
"""Return the device specific state attributes."""
|
||||||
meter = self._coordinator.data[POWERWALL_API_METERS].get(self._meter)
|
meter = self.coordinator.data[POWERWALL_API_METERS].get(self._meter)
|
||||||
return {
|
return {
|
||||||
ATTR_FREQUENCY: round(meter.frequency, 1),
|
ATTR_FREQUENCY: round(meter.frequency, 1),
|
||||||
ATTR_ENERGY_EXPORTED: convert_to_kw(meter.energy_exported),
|
ATTR_ENERGY_EXPORTED: convert_to_kw(meter.energy_exported),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user