mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-06-26 10:48:41 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

added support for caps-man for device_tracking #24

Browse source
This commit is contained in:
tomaae 2020-04-09 22:09:38 +02:00
parent 6a0536225b
commit 3886346d07
1 changed files with 53 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

59
custom_components/mikrotik_router/mikrotik_controller.py
View file

@ -63,6 +63,7 @@ class MikrotikControllerData:
"dhcp-server": {},
"dhcp-network": {},
"dhcp": {},
"capsman": {},
"host": {},
"host_hass": {},
"accounting": {}
@ -238,6 +239,9 @@ class MikrotikControllerData:
if not self.data["host_hass"]:
await self.process_host_hass()
if self.support_capsman:
await self.hass.async_add_executor_job(self.get_capsman)
await self.hass.async_add_executor_job(self.get_interface)
await self.hass.async_add_executor_job(self.get_arp)
await self.hass.async_add_executor_job(self.get_dns)
@ -807,6 +811,22 @@ class MikrotikControllerData:
self.data["dhcp"][uid]["interface"] = \
self.data["dhcp-server"][self.data["dhcp"][uid]["server"]]["interface"]
# ---------------------------
# get_capsman
# ---------------------------
def get_capsman(self):
"""Get CAPS-MAN data from Mikrotik"""
self.data["capsman"] = parse_api(
data=self.data["capsman"],
source=self.api.path("/caps-man/registration-table"),
key="mac-address",
vals=[
{"name": "mac-address"},
{"name": "interface", "default": "unknown"},
{"name": "ssid", "default": "unknown"},
]
)
# ---------------------------
# process_host_hass
# ---------------------------
@ -825,6 +845,26 @@ class MikrotikControllerData:
# ---------------------------
def process_host(self):
"""Get host tracking data"""
# Add hosts from CAPS-MAN
if self.support_capsman:
capsman_detected = {}
for uid, vals in self.data["capsman"].items():
if uid not in self.data["host"]:
self.data["host"][uid] = {}
self.data["host"][uid]["source"] = "capsman"
for key, key_data in zip(
["mac-address", "interface"],
["mac-address", "interface"],
):
if key not in self.data["host"][uid] or self.data["capsman"][uid][key] == "unknown":
self.data["capsman"][uid][key] = vals[key_data]
# Update last seen
capsman_detected[uid] = True
self.data["host"][uid]["available"] = True
self.data["host"][uid]["last-seen"] = utcnow()
# Add hosts from DHCP
for uid, vals in self.data["dhcp"].items():
if uid not in self.data["host"]:
@ -872,18 +912,25 @@ class MikrotikControllerData:
if key not in self.data["host"][uid]:
self.data["host"][uid][key] = default
# CAPS-MAN
if vals["source"] == "capsman" and uid not in capsman_detected:
self.data["host"][uid]["available"] = False
# Update IP and interface (DHCP/returned host)
if uid in self.data["dhcp"] and "." in self.data["dhcp"][uid]["address"]:
if self.data["dhcp"][uid]["address"] != self.data["host"][uid]["address"]:
self.data["host"][uid]["source"] = "dhcp"
if vals["source"] != "capsman":
self.data["host"][uid]["source"] = "dhcp"
self.data["host"][uid]["interface"] = self.data["dhcp"][uid]["interface"]
self.data["host"][uid]["address"] = self.data["dhcp"][uid]["address"]
self.data["host"][uid]["interface"] = self.data["dhcp"][uid]["interface"]
elif uid in self.data["arp"] and "." in self.data["arp"][uid]["address"] \
and self.data["arp"][uid]["arp"] != self.data["host"][uid]["address"]:
self.data["host"][uid]["source"] = "dhcp"
and self.data["arp"][uid]["address"] != self.data["host"][uid]["address"]:
if vals["source"] != "capsman":
self.data["host"][uid]["source"] = "dhcp"
self.data["host"][uid]["interface"] = self.data["arp"][uid]["interface"]
self.data["host"][uid]["address"] = self.data["arp"][uid]["address"]
self.data["host"][uid]["interface"] = self.data["arp"][uid]["interface"]
# Resolve hostname
if vals["host-name"] == "unknown":
@ -905,7 +952,7 @@ class MikrotikControllerData:
self.data["host"][uid]["host-name"] = uid
# Check host availability
if vals["address"] != "unknown" and vals["interface"] != "unknown":
if vals["source"] != "capsman" and vals["address"] != "unknown" and vals["interface"] != "unknown":
self.data["host"][uid]["available"] = \
self.api.arp_ping(vals["address"], vals["interface"])