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

converted attributes to constants

Browse source
This commit is contained in:
tomaae 2019-12-04 15:48:53 +01:00
parent c75edb8b69
commit ecee4043e3
1 changed files with 15 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

27
custom_components/mikrotik_router/sensor.py
View file

@ -21,6 +21,9 @@ _LOGGER = logging.getLogger(__name__)
ATTR_ICON = "icon" ATTR_ICON = "icon"
ATTR_LABEL = "label" ATTR_LABEL = "label"
ATTR_UNIT = "unit" ATTR_UNIT = "unit"
ATTR_GROUP = "group"
ATTR_PATH = "data_path"
ATTR_ATTR = "data_attr"
SENSOR_TYPES = { SENSOR_TYPES = {
'system_cpu-load': { 'system_cpu-load': {
@ -28,27 +31,27 @@ SENSOR_TYPES = {
ATTR_ICON: "mdi:speedometer", ATTR_ICON: "mdi:speedometer",
ATTR_LABEL: 'CPU load', ATTR_LABEL: 'CPU load',
ATTR_UNIT: "%", ATTR_UNIT: "%",
'group': "System", ATTR_GROUP: "System",
'data_path': "resource", ATTR_PATH: "resource",
'data_attr': "cpu-load", ATTR_ATTR: "cpu-load",
}, },
'system_memory-usage': { 'system_memory-usage': {
ATTR_DEVICE_CLASS: None, ATTR_DEVICE_CLASS: None,
ATTR_ICON: "mdi:memory", ATTR_ICON: "mdi:memory",
ATTR_LABEL: 'Memory usage', ATTR_LABEL: 'Memory usage',
ATTR_UNIT: "%", ATTR_UNIT: "%",
'group': "System", ATTR_GROUP: "System",
'data_path': "resource", ATTR_PATH: "resource",
'data_attr': "memory-usage", ATTR_ATTR: "memory-usage",
}, },
'system_hdd-usage': { 'system_hdd-usage': {
ATTR_DEVICE_CLASS: None, ATTR_DEVICE_CLASS: None,
ATTR_ICON: "mdi:harddisk", ATTR_ICON: "mdi:harddisk",
ATTR_LABEL: 'HDD usage', ATTR_LABEL: 'HDD usage',
ATTR_UNIT: "%", ATTR_UNIT: "%",
'group': "System", ATTR_GROUP: "System",
'data_path': "resource", ATTR_PATH: "resource",
'data_attr': "hdd-usage", ATTR_ATTR: "hdd-usage",
}, },
} }
@ -129,8 +132,8 @@ class MikrotikControllerSensor(Entity):
def state(self): def state(self):
"""Return the state.""" """Return the state."""
val = "unknown" val = "unknown"
if SENSOR_TYPES[self.kind]['data_path'] in self.mikrotik_controller.data and SENSOR_TYPES[self.kind]['data_attr'] in self.mikrotik_controller.data[SENSOR_TYPES[self.kind]['data_path']]: if SENSOR_TYPES[self.kind][ATTR_PATH] in self.mikrotik_controller.data and SENSOR_TYPES[self.kind][ATTR_ATTR] in self.mikrotik_controller.data[SENSOR_TYPES[self.kind][ATTR_PATH]]:
val = self.mikrotik_controller.data[SENSOR_TYPES[self.kind]['data_path']][SENSOR_TYPES[self.kind]['data_attr']] val = self.mikrotik_controller.data[SENSOR_TYPES[self.kind][ATTR_PATH]][SENSOR_TYPES[self.kind][ATTR_ATTR]]
return val return val
@ -174,7 +177,7 @@ class MikrotikControllerSensor(Entity):
"identifiers": {(DOMAIN, "serial-number", self.mikrotik_controller.data['routerboard']['serial-number'], "switch", "PORT")}, "identifiers": {(DOMAIN, "serial-number", self.mikrotik_controller.data['routerboard']['serial-number'], "switch", "PORT")},
"manufacturer": self.mikrotik_controller.data['resource']['platform'], "manufacturer": self.mikrotik_controller.data['resource']['platform'],
"model": self.mikrotik_controller.data['resource']['board-name'], "model": self.mikrotik_controller.data['resource']['board-name'],
"name": SENSOR_TYPES[self.kind]['group'], "name": SENSOR_TYPES[self.kind][ATTR_GROUP],
} }
return info return info