jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

Add device class attribute to modbus sensors (#30030)

Browse source
This commit is contained in:
Vladimír Záhradník 2019-12-21 10:54:48 +01:00 • committed by Paulus Schoutsen
parent 6604680793
commit 0fc92928a4
2 changed files with 32 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
homeassistant/components/modbus/sensor.py
View file

@ -1,12 +1,13 @@
"""Support for Modbus Register sensors."""
import logging
import struct
from typing import Any, Union
from typing import Any, Optional, Union
import voluptuous as vol
from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.components.sensor import DEVICE_CLASSES_SCHEMA, PLATFORM_SCHEMA
from homeassistant.const import (
CONF_DEVICE_CLASS,
CONF_NAME,
CONF_OFFSET,
CONF_SLAVE,
@ -67,6 +68,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
vol.Optional(CONF_DATA_TYPE, default=DATA_TYPE_INT): vol.In(
[DATA_TYPE_INT, DATA_TYPE_UINT, DATA_TYPE_FLOAT, DATA_TYPE_CUSTOM]
),
vol.Optional(CONF_DEVICE_CLASS): DEVICE_CLASSES_SCHEMA,
vol.Optional(CONF_HUB, default=DEFAULT_HUB): cv.string,
vol.Optional(CONF_OFFSET, default=0): number,
vol.Optional(CONF_PRECISION, default=0): cv.positive_int,
@ -139,6 +141,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
register.get(CONF_OFFSET),
structure,
register.get(CONF_PRECISION),
register.get(CONF_DEVICE_CLASS),
)
)
@ -164,6 +167,7 @@ class ModbusRegisterSensor(RestoreEntity):
offset,
structure,
precision,
device_class,
):
"""Initialize the modbus register sensor."""
self._hub = hub
@ -178,6 +182,7 @@ class ModbusRegisterSensor(RestoreEntity):
self._offset = offset
self._precision = precision
self._structure = structure
self._device_class = device_class
self._value = None
async def async_added_to_hass(self):
@ -202,6 +207,11 @@ class ModbusRegisterSensor(RestoreEntity):
"""Return the unit of measurement."""
return self._unit_of_measurement
@property
def device_class(self) -> Optional[str]:
"""Return the device class of the sensor."""
return self._device_class
def update(self):
"""Update the state of the sensor."""
if self._register_type == REGISTER_TYPE_INPUT: