evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use shorthand attributes for SRP Energy (#98953)

Browse source
This commit is contained in:
Joost Lekkerkerker 2023-08-24 19:54:49 +02:00 • committed by GitHub
parent 2066cf6b31
commit 969063ccf8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

22
homeassistant/components/srp_energy/sensor.py
View file

@ -101,6 +101,9 @@ class SrpEntity(SensorEntity):
_attr_attribution = "Powered by SRP Energy"
_attr_icon = "mdi:flash"
_attr_native_unit_of_measurement = UnitOfEnergy.KILO_WATT_HOUR
_attr_device_class = SensorDeviceClass.ENERGY
_attr_state_class = SensorStateClass.TOTAL_INCREASING
_attr_should_poll = False
def __init__(self, coordinator) -> None:
@ -108,8 +111,6 @@ class SrpEntity(SensorEntity):
self._name = SENSOR_NAME
self.type = SENSOR_TYPE
self.coordinator = coordinator
self._unit_of_measurement = UnitOfEnergy.KILO_WATT_HOUR
self._state = None
@property
def name(self) -> str:
@ -121,33 +122,16 @@ class SrpEntity(SensorEntity):
"""Return the state of the device."""
return self.coordinator.data
@property
def native_unit_of_measurement(self) -> str:
"""Return the unit of measurement of this entity, if any."""
return self._unit_of_measurement
@property
def available(self) -> bool:
"""Return if entity is available."""
return self.coordinator.last_update_success
@property
def device_class(self) -> SensorDeviceClass:
"""Return the device class."""
return SensorDeviceClass.ENERGY
@property
def state_class(self) -> SensorStateClass:
"""Return the state class."""
return SensorStateClass.TOTAL_INCREASING
async def async_added_to_hass(self) -> None:
"""When entity is added to hass."""
self.async_on_remove(
self.coordinator.async_add_listener(self.async_write_ha_state)
)
if self.coordinator.data:
self._state = self.coordinator.data
async def async_update(self) -> None:
"""Update the entity.