Fixed traffic sensor names, ref #306

This commit is contained in:
Tomaae 2023-09-18 08:48:45 +02:00
parent 6079affcfa
commit 4b610ba4e5
No known key found for this signature in database
GPG key ID: 2F97770867DAA4E6
2 changed files with 31 additions and 31 deletions

View file

@ -39,7 +39,7 @@ async def async_setup_entry(
dispatcher = { dispatcher = {
"MikrotikSensor": MikrotikSensor, "MikrotikSensor": MikrotikSensor,
"MikrotikInterfaceTrafficSensor": MikrotikInterfaceTrafficSensor, "MikrotikInterfaceTrafficSensor": MikrotikInterfaceTrafficSensor,
# "MikrotikClientTrafficSensor": MikrotikClientTrafficSensor, "MikrotikClientTrafficSensor": MikrotikClientTrafficSensor,
} }
await async_add_entities(hass, config_entry, dispatcher) await async_add_entities(hass, config_entry, dispatcher)
@ -109,27 +109,27 @@ class MikrotikInterfaceTrafficSensor(MikrotikSensor):
return attributes return attributes
# # --------------------------- # ---------------------------
# # MikrotikClientTrafficSensor # MikrotikClientTrafficSensor
# # --------------------------- # ---------------------------
# class MikrotikClientTrafficSensor(MikrotikSensor): class MikrotikClientTrafficSensor(MikrotikSensor):
# """Define an Mikrotik MikrotikClientTrafficSensor sensor.""" """Define an Mikrotik MikrotikClientTrafficSensor sensor."""
#
@property
def custom_name(self) -> str:
"""Return the name for this entity"""
return f"{self.entity_description.name}"
# @property # @property
# def name(self) -> str: # def available(self) -> bool:
# """Return the name.""" # """Return if controller and accounting feature in Mikrotik is available.
# return f"{self.entity_description.name}" # Additional check for lan-tx/rx sensors
# # """
# # @property # if self.entity_description.data_attribute in ["lan-tx", "lan-rx"]:
# # def available(self) -> bool: # return (
# # """Return if controller and accounting feature in Mikrotik is available. # self.coordinator.connected()
# # Additional check for lan-tx/rx sensors # and self._data["available"]
# # """ # and self._data["local_accounting"]
# # if self.entity_description.data_attribute in ["lan-tx", "lan-rx"]: # )
# # return ( # else:
# # self.coordinator.connected() # return self.coordinator.connected() and self._data["available"]
# # and self._data["available"]
# # and self._data["local_accounting"]
# # )
# # else:
# # return self.coordinator.connected() and self._data["available"]

View file

@ -505,7 +505,7 @@ SENSOR_TYPES: tuple[MikrotikSensorEntityDescription, ...] = (
data_uid="", data_uid="",
data_reference="mac-address", data_reference="mac-address",
data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC, data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC,
# func="MikrotikClientTrafficSensor", func="MikrotikClientTrafficSensor",
), ),
MikrotikSensorEntityDescription( MikrotikSensorEntityDescription(
key="client_traffic_lan_rx", key="client_traffic_lan_rx",
@ -526,7 +526,7 @@ SENSOR_TYPES: tuple[MikrotikSensorEntityDescription, ...] = (
data_uid="", data_uid="",
data_reference="mac-address", data_reference="mac-address",
data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC, data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC,
# func="MikrotikClientTrafficSensor", func="MikrotikClientTrafficSensor",
), ),
MikrotikSensorEntityDescription( MikrotikSensorEntityDescription(
key="client_traffic_wan_tx", key="client_traffic_wan_tx",
@ -547,7 +547,7 @@ SENSOR_TYPES: tuple[MikrotikSensorEntityDescription, ...] = (
data_uid="", data_uid="",
data_reference="mac-address", data_reference="mac-address",
data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC, data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC,
# func="MikrotikClientTrafficSensor", func="MikrotikClientTrafficSensor",
), ),
MikrotikSensorEntityDescription( MikrotikSensorEntityDescription(
key="client_traffic_wan_rx", key="client_traffic_wan_rx",
@ -568,7 +568,7 @@ SENSOR_TYPES: tuple[MikrotikSensorEntityDescription, ...] = (
data_uid="", data_uid="",
data_reference="mac-address", data_reference="mac-address",
data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC, data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC,
# func="MikrotikClientTrafficSensor", func="MikrotikClientTrafficSensor",
), ),
MikrotikSensorEntityDescription( MikrotikSensorEntityDescription(
key="client_traffic_tx", key="client_traffic_tx",
@ -589,7 +589,7 @@ SENSOR_TYPES: tuple[MikrotikSensorEntityDescription, ...] = (
data_uid="", data_uid="",
data_reference="mac-address", data_reference="mac-address",
data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC, data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC,
# func="MikrotikClientTrafficSensor", func="MikrotikClientTrafficSensor",
), ),
MikrotikSensorEntityDescription( MikrotikSensorEntityDescription(
key="client_traffic_rx", key="client_traffic_rx",
@ -610,7 +610,7 @@ SENSOR_TYPES: tuple[MikrotikSensorEntityDescription, ...] = (
data_uid="", data_uid="",
data_reference="mac-address", data_reference="mac-address",
data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC, data_attributes_list=DEVICE_ATTRIBUTES_CLIENT_TRAFFIC,
# func="MikrotikClientTrafficSensor", func="MikrotikClientTrafficSensor",
), ),
MikrotikSensorEntityDescription( MikrotikSensorEntityDescription(
key="environment", key="environment",