mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-07-08 16:34:28 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

cleanup

Browse source
This commit is contained in:
Tomaae 2023-08-10 02:21:16 +02:00
parent 9c5a270b4f
commit e1e9f61e08
No known key found for this signature in database
GPG key ID: 2F97770867DAA4E6
2 changed files with 28 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

12
custom_components/mikrotik_router/coordinator.py
View file

@ -2090,8 +2090,7 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]):
self.ds["host"][uid]["bypassed"] = self.ds["hostspot_host"][uid][ self.ds["host"][uid]["bypassed"] = self.ds["hostspot_host"][uid][
"bypassed" "bypassed"
] ]
else: elif "authorized" in self.ds["host"][uid]:
if "authorized" in self.ds["host"][uid]:
del self.ds["host"][uid]["authorized"] del self.ds["host"][uid]["authorized"]
del self.ds["host"][uid]["bypassed"] del self.ds["host"][uid]["bypassed"]
@ -2152,10 +2151,10 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]):
].split(".")[0] ].split(".")[0]
break break
if self.ds["host"][uid]["host-name"] == "unknown":
# Resolve hostname from DHCP comment # Resolve hostname from DHCP comment
if ( if (
self.ds["host"][uid]["host-name"] == "unknown" uid in self.ds["dhcp"]
and uid in self.ds["dhcp"]
and self.ds["dhcp"][uid]["enabled"] and self.ds["dhcp"][uid]["enabled"]
and self.ds["dhcp"][uid]["comment"].split("#", 1)[0] != "" and self.ds["dhcp"][uid]["comment"].split("#", 1)[0] != ""
): ):
@ -2164,8 +2163,7 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]):
].split("#", 1)[0] ].split("#", 1)[0]
# Resolve hostname from DHCP hostname # Resolve hostname from DHCP hostname
elif ( elif (
self.ds["host"][uid]["host-name"] == "unknown" uid in self.ds["dhcp"]
and uid in self.ds["dhcp"]
and self.ds["dhcp"][uid]["enabled"] and self.ds["dhcp"][uid]["enabled"]
and self.ds["dhcp"][uid]["host-name"] != "unknown" and self.ds["dhcp"][uid]["host-name"] != "unknown"
): ):
@ -2173,7 +2171,7 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]):
"host-name" "host-name"
] ]
# Fallback to mac address for hostname # Fallback to mac address for hostname
elif self.ds["host"][uid]["host-name"] == "unknown": else:
self.ds["host"][uid]["host-name"] = uid self.ds["host"][uid]["host-name"] = uid
# Resolve manufacturer # Resolve manufacturer

8
custom_components/mikrotik_router/entity.py
View file

@ -245,7 +245,7 @@ class MikrotikEntity(CoordinatorEntity[_MikrotikCoordinatorT], Entity):
dev_connection_value = self._data[dev_connection_value] dev_connection_value = self._data[dev_connection_value]
if self.entity_description.ha_group == "System": if self.entity_description.ha_group == "System":
info = DeviceInfo( return DeviceInfo(
connections={(dev_connection, f"{dev_connection_value}")}, connections={(dev_connection, f"{dev_connection_value}")},
identifiers={(dev_connection, f"{dev_connection_value}")}, identifiers={(dev_connection, f"{dev_connection_value}")},
name=f"{self._inst} {dev_group}", name=f"{self._inst} {dev_group}",
@ -265,7 +265,7 @@ class MikrotikEntity(CoordinatorEntity[_MikrotikCoordinatorT], Entity):
"manufacturer" "manufacturer"
] ]
info = DeviceInfo( return DeviceInfo(
connections={(dev_connection, f"{dev_connection_value}")}, connections={(dev_connection, f"{dev_connection_value}")},
default_name=f"{dev_group}", default_name=f"{dev_group}",
default_manufacturer=f"{dev_manufacturer}", default_manufacturer=f"{dev_manufacturer}",
@ -275,7 +275,7 @@ class MikrotikEntity(CoordinatorEntity[_MikrotikCoordinatorT], Entity):
), ),
) )
else: else:
info = DeviceInfo( return DeviceInfo(
connections={(dev_connection, f"{dev_connection_value}")}, connections={(dev_connection, f"{dev_connection_value}")},
default_name=f"{self._inst} {dev_group}", default_name=f"{self._inst} {dev_group}",
default_model=f"{self.coordinator.data['resource']['board-name']}", default_model=f"{self.coordinator.data['resource']['board-name']}",
@ -286,8 +286,6 @@ class MikrotikEntity(CoordinatorEntity[_MikrotikCoordinatorT], Entity):
), ),
) )
return info
@property @property
def extra_state_attributes(self) -> Mapping[str, Any]: def extra_state_attributes(self) -> Mapping[str, Any]:
"""Return the state attributes.""" """Return the state attributes."""