removed unnecessery MikrotikClientTrafficSensor

This commit is contained in:
Tomaae 2023-08-09 09:25:33 +02:00
parent 7a892b6800
commit cace2ae3e5
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 # @property
def name(self) -> str: # def name(self) -> str:
"""Return the name.""" # """Return the name."""
return f"{self.entity_description.name}" # return f"{self.entity_description.name}"
#
@property # # @property
def available(self) -> bool: # # def available(self) -> bool:
"""Return if controller and accounting feature in Mikrotik is available. # # """Return if controller and accounting feature in Mikrotik is available.
Additional check for lan-tx/rx sensors # # Additional check for lan-tx/rx sensors
""" # # """
if self.entity_description.data_attribute in ["lan-tx", "lan-rx"]: # # if self.entity_description.data_attribute in ["lan-tx", "lan-rx"]:
return ( # # return (
self.coordinator.connected() # # self.coordinator.connected()
and self._data["available"] # # and self._data["available"]
and self._data["local_accounting"] # # and self._data["local_accounting"]
) # # )
else: # # else:
return self.coordinator.connected() and self._data["available"] # # return self.coordinator.connected() and self._data["available"]

View file

@ -503,7 +503,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",
@ -524,7 +524,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",
@ -545,7 +545,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",
@ -566,7 +566,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",
@ -587,7 +587,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",
@ -608,7 +608,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",