"""Support for the Mikrotik Router sensor service.""" import logging from homeassistant.const import (CONF_NAME, ATTR_ATTRIBUTION, ATTR_DEVICE_CLASS) from homeassistant.core import callback from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.entity import Entity from .const import (DOMAIN, DATA_CLIENT, ATTRIBUTION) _LOGGER = logging.getLogger(__name__) ATTR_ICON = "icon" ATTR_LABEL = "label" ATTR_UNIT = "unit" ATTR_UNIT_ATTR = "unit_attr" ATTR_GROUP = "group" ATTR_PATH = "data_path" ATTR_ATTR = "data_attr" SENSOR_TYPES = { "system_cpu-load": { ATTR_DEVICE_CLASS: None, ATTR_ICON: "mdi:speedometer", ATTR_LABEL: "CPU load", ATTR_UNIT: "%", ATTR_GROUP: "System", ATTR_PATH: "resource", ATTR_ATTR: "cpu-load", }, "system_memory-usage": { ATTR_DEVICE_CLASS: None, ATTR_ICON: "mdi:memory", ATTR_LABEL: "Memory usage", ATTR_UNIT: "%", ATTR_GROUP: "System", ATTR_PATH: "resource", ATTR_ATTR: "memory-usage", }, "system_hdd-usage": { ATTR_DEVICE_CLASS: None, ATTR_ICON: "mdi:harddisk", ATTR_LABEL: "HDD usage", ATTR_UNIT: "%", ATTR_GROUP: "System", ATTR_PATH: "resource", ATTR_ATTR: "hdd-usage", }, "traffic_tx": { ATTR_DEVICE_CLASS: None, ATTR_ICON: "mdi:upload-network-outline", ATTR_LABEL: "TX", ATTR_UNIT: "ps", ATTR_UNIT_ATTR: "tx-bits-per-second-attr", ATTR_PATH: "interface", ATTR_ATTR: "tx-bits-per-second", }, "traffic_rx": { ATTR_DEVICE_CLASS: None, ATTR_ICON: "mdi:download-network-outline", ATTR_LABEL: "RX", ATTR_UNIT: "ps", ATTR_UNIT_ATTR: "rx-bits-per-second-attr", ATTR_PATH: "interface", ATTR_ATTR: "rx-bits-per-second", }, } # --------------------------- # async_setup_entry # --------------------------- async def async_setup_entry(hass, config_entry, async_add_entities): """Set up device tracker for Mikrotik Router component.""" inst = config_entry.data[CONF_NAME] mikrotik_controller = hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id] sensors = {} @callback def update_controller(): """Update the values of the controller.""" update_items(inst, mikrotik_controller, async_add_entities, sensors) mikrotik_controller.listeners.append( async_dispatcher_connect( hass, mikrotik_controller.signal_update, update_controller ) ) update_controller() # --------------------------- # update_items # --------------------------- @callback def update_items(inst, mikrotik_controller, async_add_entities, sensors): """Update sensor state from the controller.""" new_sensors = [] for sensor in SENSOR_TYPES: if "traffic_" not in sensor: item_id = f"{inst}-{sensor}" _LOGGER.debug("Updating sensor %s", item_id) if item_id in sensors: if sensors[item_id].enabled: sensors[item_id].async_schedule_update_ha_state() continue sensors[item_id] = MikrotikControllerSensor( mikrotik_controller=mikrotik_controller, inst=inst, sensor=sensor ) new_sensors.append(sensors[item_id]) if "traffic_" in sensor: for uid in mikrotik_controller.data["interface"]: if mikrotik_controller.data["interface"][uid]["type"] == "ether": item_id = f"{inst}-{sensor}-{mikrotik_controller.data['interface'][uid]['default-name']}" _LOGGER.debug("Updating sensor %s", item_id) if item_id in sensors: if sensors[item_id].enabled: sensors[item_id].async_schedule_update_ha_state() continue sensors[item_id] = MikrotikControllerTrafficSensor( mikrotik_controller=mikrotik_controller, inst=inst, sensor=sensor, uid=uid, ) new_sensors.append(sensors[item_id]) if new_sensors: async_add_entities(new_sensors, True) # --------------------------- # MikrotikControllerSensor # --------------------------- class MikrotikControllerSensor(Entity): """Define an Mikrotik Controller sensor.""" def __init__(self, mikrotik_controller, inst, sensor): """Initialize.""" self._inst = inst self._sensor = sensor self._ctrl = mikrotik_controller self._data = mikrotik_controller.data[SENSOR_TYPES[sensor][ATTR_PATH]] self._type = SENSOR_TYPES[sensor] self._attr = SENSOR_TYPES[sensor][ATTR_ATTR] self._device_class = None self._state = None self._icon = None self._unit_of_measurement = None self._attrs = {ATTR_ATTRIBUTION: ATTRIBUTION} @property def name(self): """Return the name.""" return f"{self._inst} {self._type[ATTR_LABEL]}" @property def state(self): """Return the state.""" val = "unknown" if self._attr in self._data: val = self._data[self._attr] return val @property def device_state_attributes(self): """Return the state attributes.""" return self._attrs @property def icon(self): """Return the icon.""" self._icon = self._type[ATTR_ICON] return self._icon @property def device_class(self): """Return the device_class.""" return self._type[ATTR_DEVICE_CLASS] @property def unique_id(self): """Return a unique_id for this entity.""" return f"{self._inst.lower()}-{self._sensor.lower()}" @property def unit_of_measurement(self): """Return the unit the value is expressed in.""" if ATTR_UNIT_ATTR in self._type: return self._data[SENSOR_TYPES[self._sensor][ATTR_UNIT_ATTR]] if ATTR_UNIT in self._type: return self._type[ATTR_UNIT] @property def available(self) -> bool: """Return if controller is available.""" return self._ctrl.connected() @property def device_info(self): """Return a port description for device registry.""" info = { "identifiers": { ( DOMAIN, "serial-number", self._ctrl.data["routerboard"]["serial-number"], "switch", "PORT", ) }, "manufacturer": self._ctrl.data["resource"]["platform"], "model": self._ctrl.data["resource"]["board-name"], "name": self._type[ATTR_GROUP], } return info async def async_update(self): """Synchronize state with controller.""" async def async_added_to_hass(self): """Port entity created.""" _LOGGER.debug("New sensor %s (%s)", self._inst, self._sensor) # --------------------------- # MikrotikControllerTrafficSensor # --------------------------- class MikrotikControllerTrafficSensor(MikrotikControllerSensor): """Define an Mikrotik Controller sensor.""" def __init__(self, mikrotik_controller, inst, sensor, uid): """Initialize.""" super().__init__(mikrotik_controller, inst, sensor) self._uid = uid self._data = mikrotik_controller.data[SENSOR_TYPES[sensor][ATTR_PATH]][ uid] @property def name(self): """Return the name.""" return f"{self._inst} {self._data['name']} {self._type[ATTR_LABEL]}" @property def unique_id(self): """Return a unique_id for this entity.""" return f"{self._inst.lower()}-{self._sensor.lower()}-{self._data['default-name'].lower()}" @property def device_info(self): """Return a port description for device registry.""" info = { "connections": { (CONNECTION_NETWORK_MAC, self._data["port-mac-address"])}, "manufacturer": self._ctrl.data["resource"]["platform"], "model": self._ctrl.data["resource"]["board-name"], "name": self._data["default-name"], } return info async def async_added_to_hass(self): """Port entity created.""" _LOGGER.debug( "New sensor %s (%s %s)", self._inst, self._data["default-name"], self._sensor, )