mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-07-10 01:14:33 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add device_class attributes to valid sensors

Browse source 
Fixes #119
This commit is contained in:
Sergio Mayoral Martinez 2021-05-13 13:35:54 +02:00
parent cfd26adeee
commit 2257a563bf
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
custom_components/mikrotik_router/sensor.py
View file

@ -7,6 +7,8 @@ from homeassistant.const import (
CONF_NAME,
ATTR_ATTRIBUTION,
ATTR_DEVICE_CLASS,
DEVICE_CLASS_TEMPERATURE,
DEVICE_CLASS_POWER,
TEMP_CELSIUS,
)
@ -51,7 +53,7 @@ ATTR_ATTR = "data_attr"
SENSOR_TYPES = {
"system_temperature": {
ATTR_DEVICE_CLASS: None,
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
ATTR_ICON: "mdi:thermometer",
ATTR_LABEL: "Temperature",
ATTR_UNIT: TEMP_CELSIUS,
@ -60,7 +62,7 @@ SENSOR_TYPES = {
ATTR_ATTR: "temperature",
},
"system_cpu-temperature": {
ATTR_DEVICE_CLASS: None,
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
ATTR_ICON: "mdi:thermometer",
ATTR_LABEL: "CPU temperature",
ATTR_UNIT: TEMP_CELSIUS,
@ -69,7 +71,7 @@ SENSOR_TYPES = {
ATTR_ATTR: "cpu-temperature",
},
"system_board-temperature1": {
ATTR_DEVICE_CLASS: None,
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
ATTR_ICON: "mdi:thermometer",
ATTR_LABEL: "Board temperature",
ATTR_UNIT: TEMP_CELSIUS,
@ -78,7 +80,7 @@ SENSOR_TYPES = {
ATTR_ATTR: "board-temperature1",
},
"system_power-consumption": {
ATTR_DEVICE_CLASS: None,
ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER,
ATTR_ICON: "mdi:transmission-tower",
ATTR_LABEL: "Power consumption",
ATTR_UNIT: "W",
@ -396,15 +398,15 @@ class MikrotikControllerSensor(SensorEntity):
if self._icon:
return self._icon
return ""
return None
@property
def device_class(self) -> Optional[str]:
"""Return the device class."""
if ATTR_UNIT_ATTR in self._type:
if ATTR_DEVICE_CLASS in self._type:
return self._type[ATTR_DEVICE_CLASS]
else:
return None
return None
@property
def unique_id(self) -> str: