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

Add more sensors to the Shelly integration (#39368)

Browse Source
This commit is contained in:
Maciej Bieniek 2020-08-30 14:18:35 +02:00 committed by GitHub
parent 4aa8b6cad8
commit 0cffba77cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 38 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
homeassistant/components/shelly/sensor.py
View File

@ -2,15 +2,31 @@
import aioshelly import aioshelly
from homeassistant.components import sensor from homeassistant.components import sensor
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT, UNIT_PERCENTAGE from homeassistant.const import (
ELECTRICAL_CURRENT_AMPERE,
ENERGY_KILO_WATT_HOUR,
POWER_WATT,
TEMP_CELSIUS,
TEMP_FAHRENHEIT,
UNIT_PERCENTAGE,
VOLT,
)
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from . import ShellyBlockEntity, ShellyDeviceWrapper from . import ShellyBlockEntity, ShellyDeviceWrapper
from .const import DOMAIN from .const import DOMAIN
SENSORS = { SENSORS = {
"battery": [UNIT_PERCENTAGE, sensor.DEVICE_CLASS_BATTERY],
"current": [ELECTRICAL_CURRENT_AMPERE, sensor.DEVICE_CLASS_CURRENT],
"deviceTemp": [None, sensor.DEVICE_CLASS_TEMPERATURE],
"energy": [ENERGY_KILO_WATT_HOUR, sensor.DEVICE_CLASS_ENERGY],
"energyReturned": [ENERGY_KILO_WATT_HOUR, sensor.DEVICE_CLASS_ENERGY],
"extTemp": [None, sensor.DEVICE_CLASS_TEMPERATURE], "extTemp": [None, sensor.DEVICE_CLASS_TEMPERATURE],
"humidity": [UNIT_PERCENTAGE, sensor.DEVICE_CLASS_HUMIDITY], "humidity": [UNIT_PERCENTAGE, sensor.DEVICE_CLASS_HUMIDITY],
"overpowerValue": [POWER_WATT, sensor.DEVICE_CLASS_POWER],
"power": [POWER_WATT, sensor.DEVICE_CLASS_POWER],
"voltage": [VOLT, sensor.DEVICE_CLASS_VOLTAGE],
} }
@ -20,9 +36,6 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
sensors = [] sensors = []
for block in wrapper.device.blocks: for block in wrapper.device.blocks:
if block.type != "sensor":
continue
for attr in SENSORS: for attr in SENSORS:
if not hasattr(block, attr): if not hasattr(block, attr):
continue continue
@ -46,13 +59,18 @@ class ShellySensor(ShellyBlockEntity, Entity):
super().__init__(wrapper, block) super().__init__(wrapper, block)
self.attribute = attribute self.attribute = attribute
unit, device_class = SENSORS[attribute] unit, device_class = SENSORS[attribute]
info = block.info(attribute) self.info = block.info(attribute)
if info[aioshelly.BLOCK_VALUE_TYPE] == aioshelly.BLOCK_VALUE_TYPE_TEMPERATURE: if (
if info[aioshelly.BLOCK_VALUE_UNIT] == "C": self.info[aioshelly.BLOCK_VALUE_TYPE]
== aioshelly.BLOCK_VALUE_TYPE_TEMPERATURE
):
if self.info[aioshelly.BLOCK_VALUE_UNIT] == "C":
unit = TEMP_CELSIUS unit = TEMP_CELSIUS
else: else:
unit = TEMP_FAHRENHEIT unit = TEMP_FAHRENHEIT
elif self.info[aioshelly.BLOCK_VALUE_TYPE] == aioshelly.BLOCK_VALUE_TYPE_ENERGY:
unit = ENERGY_KILO_WATT_HOUR
self._unit = unit self._unit = unit
self._device_class = device_class self._device_class = device_class
@ -70,6 +88,19 @@ class ShellySensor(ShellyBlockEntity, Entity):
@property @property
def state(self): def state(self):
"""Value of sensor.""" """Value of sensor."""
if self.attribute in [
"deviceTemp",
"extTemp",
"humidity",
"overpowerValue",
"power",
]:
return round(getattr(self.block, self.attribute), 1)
# Energy unit change from Wmin or Wh to kWh
if self.info[aioshelly.BLOCK_VALUE_UNIT] == "Wmin":
return round(getattr(self.block, self.attribute) / 60 / 1000, 2)
if self.info[aioshelly.BLOCK_VALUE_UNIT] == "Wh":
return round(getattr(self.block, self.attribute) / 1000, 2)
return getattr(self.block, self.attribute) return getattr(self.block, self.attribute)
@property @property