jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-16 09:47:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Mill cleanup (#54396)

Browse Source
This commit is contained in:
Daniel Hjelseth Høyer 2021-08-10 16:28:39 +02:00 committed by GitHub
parent cf8f27bb44
commit f03b160c46
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
homeassistant/components/mill/sensor.py
View File

@ -5,7 +5,7 @@ from homeassistant.components.sensor import (
STATE_CLASS_MEASUREMENT, STATE_CLASS_MEASUREMENT,
SensorEntity, SensorEntity,
) )
from homeassistant.const import ENERGY_KILO_WATT_HOUR, STATE_UNKNOWN from homeassistant.const import ENERGY_KILO_WATT_HOUR
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
from .const import CONSUMPTION_TODAY, CONSUMPTION_YEAR, DOMAIN, MANUFACTURER from .const import CONSUMPTION_TODAY, CONSUMPTION_YEAR, DOMAIN, MANUFACTURER
@ -16,13 +16,12 @@ async def async_setup_entry(hass, entry, async_add_entities):
mill_data_connection = hass.data[DOMAIN] mill_data_connection = hass.data[DOMAIN]
dev = [] entities = [
for heater in mill_data_connection.heaters.values(): MillHeaterEnergySensor(heater, mill_data_connection, sensor_type)
for sensor_type in (CONSUMPTION_TODAY, CONSUMPTION_YEAR): for sensor_type in (CONSUMPTION_TODAY, CONSUMPTION_YEAR)
dev.append( for heater in mill_data_connection.heaters.values()
MillHeaterEnergySensor(heater, mill_data_connection, sensor_type) ]
) async_add_entities(entities)
async_add_entities(dev)
class MillHeaterEnergySensor(SensorEntity): class MillHeaterEnergySensor(SensorEntity):
@ -71,7 +70,7 @@ class MillHeaterEnergySensor(SensorEntity):
self._attr_state = _state self._attr_state = _state
return return
if self.state not in [STATE_UNKNOWN, None] and _state < self.state: if self.state is not None and _state < self.state:
if self._sensor_type == CONSUMPTION_TODAY: if self._sensor_type == CONSUMPTION_TODAY:
self._attr_last_reset = dt_util.as_utc( self._attr_last_reset = dt_util.as_utc(
dt_util.now().replace(hour=0, minute=0, second=0, microsecond=0) dt_util.now().replace(hour=0, minute=0, second=0, microsecond=0)