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

Handle state unknown if last state is missing in Shelly (#86813)

Browse source 
Shelly - handle state unknown if last state is missing
This commit is contained in:
Shay Levy 2023-01-28 00:34:56 +02:00 • committed by GitHub
parent 50c2992f36
commit 803cd8d9a3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 170 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

21
homeassistant/components/shelly/sensor.py
View file

@ -15,6 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ATTR_UNIT_OF_MEASUREMENT,
CONCENTRATION_PARTS_PER_MILLION,
DEGREE,
LIGHT_LUX,
@ -669,7 +670,10 @@ class BlockSleepingSensor(ShellySleepingBlockAttributeEntity, SensorEntity):
if self.block is not None:
return self.attribute_value
return self.last_state
if self.last_state is None:
return None
return self.last_state.state
@property
def native_unit_of_measurement(self) -> str | None:
@ -677,7 +681,10 @@ class BlockSleepingSensor(ShellySleepingBlockAttributeEntity, SensorEntity):
if self.block is not None:
return self.entity_description.native_unit_of_measurement
return self.last_unit
if self.last_state is None:
return None
return self.last_state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
class RpcSleepingSensor(ShellySleepingRpcAttributeEntity, SensorEntity):
@ -691,12 +698,12 @@ class RpcSleepingSensor(ShellySleepingRpcAttributeEntity, SensorEntity):
if self.coordinator.device.initialized:
return self.attribute_value
return self.last_state
if self.last_state is None:
return None
return self.last_state.state
@property
def native_unit_of_measurement(self) -> str | None:
"""Return the unit of measurement of the sensor, if any."""
if self.coordinator.device.initialized:
return self.entity_description.native_unit_of_measurement
return self.last_unit
return self.entity_description.native_unit_of_measurement