diff --git a/custom_components/mikrotik_router/binary_sensor_types.py b/custom_components/mikrotik_router/binary_sensor_types.py index 6657907..040822c 100644 --- a/custom_components/mikrotik_router/binary_sensor_types.py +++ b/custom_components/mikrotik_router/binary_sensor_types.py @@ -118,6 +118,7 @@ DEVICE_ATTRIBUTES_NETWATCH = [ "comment", ] + @dataclass class MikrotikBinarySensorEntityDescription(BinarySensorEntityDescription): """Class describing mikrotik entities.""" diff --git a/custom_components/mikrotik_router/config_flow.py b/custom_components/mikrotik_router/config_flow.py index 83d1f97..297679d 100644 --- a/custom_components/mikrotik_router/config_flow.py +++ b/custom_components/mikrotik_router/config_flow.py @@ -285,7 +285,8 @@ class MikrotikControllerOptionsFlowHandler(OptionsFlow): vol.Optional( CONF_SENSOR_NETWATCH_TRACKER, default=self.config_entry.options.get( - CONF_SENSOR_NETWATCH_TRACKER, DEFAULT_SENSOR_NETWATCH_TRACKER + CONF_SENSOR_NETWATCH_TRACKER, + DEFAULT_SENSOR_NETWATCH_TRACKER, ), ): bool, vol.Optional( diff --git a/custom_components/mikrotik_router/coordinator.py b/custom_components/mikrotik_router/coordinator.py index a072e77..4568fcb 100644 --- a/custom_components/mikrotik_router/coordinator.py +++ b/custom_components/mikrotik_router/coordinator.py @@ -59,7 +59,9 @@ from .const import ( CONF_SENSOR_SCRIPTS, DEFAULT_SENSOR_SCRIPTS, CONF_SENSOR_ENVIRONMENT, - DEFAULT_SENSOR_ENVIRONMENT, CONF_SENSOR_NETWATCH_TRACKER, DEFAULT_SENSOR_NETWATCH_TRACKER, + DEFAULT_SENSOR_ENVIRONMENT, + CONF_SENSOR_NETWATCH_TRACKER, + DEFAULT_SENSOR_NETWATCH_TRACKER, ) from .exceptions import ApiEntryNotFound from .apiparser import parse_api @@ -258,7 +260,7 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]): "environment": {}, "ups": {}, "gps": {}, - "netwatch": {} + "netwatch": {}, } self.notified_flags = [] @@ -393,7 +395,9 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]): @property def option_sensor_netwatch(self): """Config entry option to not track ARP.""" - return self.config_entry.options.get(CONF_SENSOR_NETWATCH_TRACKER, DEFAULT_SENSOR_NETWATCH_TRACKER) + return self.config_entry.options.get( + CONF_SENSOR_NETWATCH_TRACKER, DEFAULT_SENSOR_NETWATCH_TRACKER + ) # --------------------------- # option_sensor_ppp diff --git a/custom_components/mikrotik_router/entity.py b/custom_components/mikrotik_router/entity.py index daf171f..9925238 100644 --- a/custom_components/mikrotik_router/entity.py +++ b/custom_components/mikrotik_router/entity.py @@ -72,11 +72,8 @@ def _skip_sensor(config_entry, entity_description, data, uid) -> bool: ): return True - if ( - entity_description.data_path == "netwatch" - and not config_entry.options.get( - CONF_SENSOR_NETWATCH_TRACKER, DEFAULT_SENSOR_NETWATCH_TRACKER - ) + if entity_description.data_path == "netwatch" and not config_entry.options.get( + CONF_SENSOR_NETWATCH_TRACKER, DEFAULT_SENSOR_NETWATCH_TRACKER ): return True