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

Implemented mangle switches #75

Browse source
This commit is contained in:
tomaae 2020-12-18 19:58:54 +01:00
parent 117fec7654
commit 68a47ffa3c
6 changed files with 178 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

59
custom_components/mikrotik_router/mikrotik_controller.py
View file

@ -40,6 +40,8 @@ from .const import (
DEFAULT_SENSOR_SIMPLE_QUEUES,
CONF_SENSOR_NAT,
DEFAULT_SENSOR_NAT,
CONF_SENSOR_MANGLE,
DEFAULT_SENSOR_MANGLE,
CONF_SENSOR_SCRIPTS,
DEFAULT_SENSOR_SCRIPTS,
CONF_SENSOR_ENVIRONMENT,
@ -74,6 +76,7 @@ class MikrotikControllerData:
"bridge_host": {},
"arp": {},
"nat": {},
"mangle": {},
"fw-update": {},
"script": {},
"queue": {},
@ -192,6 +195,14 @@ class MikrotikControllerData:
"""Config entry option to not track ARP."""
return self.config_entry.options.get(CONF_SENSOR_NAT, DEFAULT_SENSOR_NAT)
# ---------------------------
# option_sensor_nat
# ---------------------------
@property
def option_sensor_mangle(self):
"""Config entry option to not track ARP."""
return self.config_entry.options.get(CONF_SENSOR_MANGLE, DEFAULT_SENSOR_MANGLE)
# ---------------------------
# option_sensor_scripts
# ---------------------------
@ -489,6 +500,9 @@ class MikrotikControllerData:
if self.api.connected() and self.option_sensor_nat:
await self.hass.async_add_executor_job(self.get_nat)
if self.api.connected() and self.option_sensor_mangle:
await self.hass.async_add_executor_job(self.get_mangle)
if self.api.connected():
await self.hass.async_add_executor_job(self.get_system_resource)
@ -739,6 +753,51 @@ class MikrotikControllerData:
del self.data["nat"][uid]
# ---------------------------
# get_mangle
# ---------------------------
def get_mangle(self):
"""Get Mangle data from Mikrotik"""
self.data["mangle"] = parse_api(
data=self.data["mangle"],
source=self.api.path("/ip/firewall/mangle"),
key=".id",
vals=[
{"name": ".id"},
{"name": "chain"},
{"name": "action"},
{"name": "comment"},
{"name": "address-list"},
{"name": "passthrough", "type": "bool", "default": False},
{"name": "new-packet-mark"},
{"name": "src-address-list"},
{"name": "protocol", "default": "any"},
{"name": "src-port", "default": "any"},
{"name": "dst-port", "default": "any"},
{
"name": "enabled",
"source": "disabled",
"type": "bool",
"reverse": True,
},
],
val_proc=[
[
{"name": "name"},
{"action": "combine"},
{"key": "protocol"},
{"text": ":"},
{"key": "dst-port"},
]
],
skip=[
{"name": "dynamic", "value": True},
{"name": "action", "value": "jump"},
{"name": "protocol", "value": ""},
{"name": "dst-port", "value": ""},
],
)
# ---------------------------
# get_system_routerboard
# ---------------------------