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

Fix some sensor classes (#48254)

Browse source 
* Fix some sensor classes

* Tweak

* Tweak
This commit is contained in:
Erik Montnemery 2021-03-23 15:56:33 +01:00 • committed by GitHub
parent 6932cf9534
commit a09c8eecb7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
19 changed files with 88 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -212,6 +212,11 @@ class ShellySensor(ShellyBlockAttributeEntity, SensorEntity):
"""Return value of sensor."""
return self.attribute_value
@property
def unit_of_measurement(self):
"""Return unit of sensor."""
return self._unit
class ShellyRestSensor(ShellyRestAttributeEntity, SensorEntity):
"""Represent a shelly REST sensor."""
@ -221,6 +226,11 @@ class ShellyRestSensor(ShellyRestAttributeEntity, SensorEntity):
"""Return value of sensor."""
return self.attribute_value
@property
def unit_of_measurement(self):
"""Return unit of sensor."""
return self.description.unit
class ShellySleepingSensor(ShellySleepingBlockAttributeEntity, SensorEntity):
"""Represent a shelly sleeping sensor."""
@ -232,3 +242,8 @@ class ShellySleepingSensor(ShellySleepingBlockAttributeEntity, SensorEntity):
return self.attribute_value
return self.last_state
@property
def unit_of_measurement(self):
"""Return unit of sensor."""
return self._unit