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

sensor entity code cleanup

Browse source
This commit is contained in:
tomaae 2021-12-15 02:06:27 +01:00
parent 86a4dd358f
commit e372ebbfff
2 changed files with 44 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

34
custom_components/mikrotik_router/binary_sensor.py
View file

@ -227,28 +227,24 @@ def update_items(inst, config_entry, mikrotik_controller, async_add_entities, se
class MikrotikControllerBinarySensor(BinarySensorEntity):
"""Define an Mikrotik Controller Binary Sensor."""
def __init__(self, mikrotik_controller, inst, sensor):
def __init__(self, mikrotik_controller, inst, sid_data):
"""Initialize."""
self._inst = inst
self._sensor = sensor
self._sensor = sid_data
self._ctrl = mikrotik_controller
if sensor in SENSOR_TYPES:
self._data = mikrotik_controller.data[SENSOR_TYPES[sensor][ATTR_PATH]]
self._type = SENSOR_TYPES[sensor]
self._attr = SENSOR_TYPES[sensor][ATTR_ATTR]
if sid_data in SENSOR_TYPES:
self._data = mikrotik_controller.data[SENSOR_TYPES[sid_data][ATTR_PATH]]
self._type = SENSOR_TYPES[sid_data]
self._attr = SENSOR_TYPES[sid_data][ATTR_ATTR]
self._dcls = SENSOR_TYPES[sid_data][ATTR_DEVICE_CLASS]
self._ctgr = SENSOR_TYPES[sid_data][ATTR_CTGR]
else:
self._type = {}
self._attr = None
self._dcls = None
self._ctgr = None
if ATTR_CTGR in self._type:
self._entity_category = self._type[ATTR_CTGR]
else:
self._entity_category = None
self._device_class = None
self._state = None
self._icon = None
self._unit_of_measurement = None
self._attrs = {ATTR_ATTRIBUTION: ATTRIBUTION}
@property
@ -264,10 +260,7 @@ class MikrotikControllerBinarySensor(BinarySensorEntity):
@property
def device_class(self) -> Optional[str]:
"""Return the device class."""
if ATTR_DEVICE_CLASS in self._type:
return self._type[ATTR_DEVICE_CLASS]
return None
return self._dcls
@property
def unique_id(self) -> str:
@ -282,10 +275,7 @@ class MikrotikControllerBinarySensor(BinarySensorEntity):
@property
def entity_category(self) -> str:
"""Return entity category"""
if self._entity_category:
return self._entity_category
return None
return self._ctgr
@property
def device_info(self) -> Dict[str, Any]: