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

fixed switch control

Browse source
This commit is contained in:
Tomaae 2023-08-09 09:48:08 +02:00
parent 727a075dbb
commit eae9fd7e2d
No known key found for this signature in database
GPG key ID: 2F97770867DAA4E6
1 changed files with 68 additions and 68 deletions
Download patch file Download diff file Expand all files Collapse all files

136
custom_components/mikrotik_router/switch.py
View file

@ -74,27 +74,27 @@ class MikrotikSwitch(MikrotikEntity, SwitchEntity, RestoreEntity):
async def async_turn_on(self) -> None:
"""Turn on the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = self.entity_description.data_reference
value = self._data[self.entity_description.data_reference]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, False)
await self._ctrl.force_update()
self.coordinator.set_value(path, param, value, mod_param, False)
await self.coordinator.async_refresh()
async def async_turn_off(self) -> None:
"""Turn off the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = self.entity_description.data_reference
value = self._data[self.entity_description.data_reference]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, True)
await self._ctrl.async_update()
self.coordinator.set_value(path, param, value, mod_param, True)
await self.coordinator.async_refresh()
# ---------------------------
@ -140,7 +140,7 @@ class MikrotikPortSwitch(MikrotikSwitch):
async def async_turn_on(self) -> Optional[str]:
"""Turn on the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
@ -152,17 +152,17 @@ class MikrotikPortSwitch(MikrotikSwitch):
param = "name"
value = self._data[self.entity_description.data_reference]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, False)
self.coordinator.set_value(path, param, value, mod_param, False)
if "poe-out" in self._data and self._data["poe-out"] == "off":
path = "/interface/ethernet"
self._ctrl.set_value(path, param, value, "poe-out", "auto-on")
self.coordinator.set_value(path, param, value, "poe-out", "auto-on")
await self._ctrl.force_update()
await self.coordinator.async_refresh()
async def async_turn_off(self) -> Optional[str]:
"""Turn off the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
@ -174,13 +174,13 @@ class MikrotikPortSwitch(MikrotikSwitch):
param = "name"
value = self._data[self.entity_description.data_reference]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, True)
self.coordinator.set_value(path, param, value, mod_param, True)
if "poe-out" in self._data and self._data["poe-out"] == "auto-on":
path = "/interface/ethernet"
self._ctrl.set_value(path, param, value, "poe-out", "off")
self.coordinator.set_value(path, param, value, "poe-out", "off")
await self._ctrl.async_update()
await self.coordinator.async_refresh()
# ---------------------------
@ -191,43 +191,43 @@ class MikrotikNATSwitch(MikrotikSwitch):
async def async_turn_on(self) -> None:
"""Turn on the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["nat"]:
if self._ctrl.data["nat"][uid]["uniq-id"] == (
for uid in self.coordinator.data["nat"]:
if self.coordinator.data["nat"][uid]["uniq-id"] == (
f"{self._data['chain']},{self._data['action']},{self._data['protocol']},"
f"{self._data['in-interface']}:{self._data['dst-port']}-"
f"{self._data['out-interface']}:{self._data['to-addresses']}:{self._data['to-ports']}"
):
value = self._ctrl.data["nat"][uid][".id"]
value = self.coordinator.data["nat"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, False)
await self._ctrl.force_update()
self.coordinator.set_value(path, param, value, mod_param, False)
await self.coordinator.async_refresh()
async def async_turn_off(self) -> None:
"""Turn off the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["nat"]:
if self._ctrl.data["nat"][uid]["uniq-id"] == (
for uid in self.coordinator.data["nat"]:
if self.coordinator.data["nat"][uid]["uniq-id"] == (
f"{self._data['chain']},{self._data['action']},{self._data['protocol']},"
f"{self._data['in-interface']}:{self._data['dst-port']}-"
f"{self._data['out-interface']}:{self._data['to-addresses']}:{self._data['to-ports']}"
):
value = self._ctrl.data["nat"][uid][".id"]
value = self.coordinator.data["nat"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, True)
await self._ctrl.async_update()
self.coordinator.set_value(path, param, value, mod_param, True)
await self.coordinator.async_refresh()
# ---------------------------
@ -238,45 +238,45 @@ class MikrotikMangleSwitch(MikrotikSwitch):
async def async_turn_on(self) -> None:
"""Turn on the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["mangle"]:
if self._ctrl.data["mangle"][uid]["uniq-id"] == (
for uid in self.coordinator.data["mangle"]:
if self.coordinator.data["mangle"][uid]["uniq-id"] == (
f"{self._data['chain']},{self._data['action']},{self._data['protocol']},"
f"{self._data['src-address']}:{self._data['src-port']}-"
f"{self._data['dst-address']}:{self._data['dst-port']},"
f"{self._data['src-address-list']}-{self._data['dst-address-list']}"
):
value = self._ctrl.data["mangle"][uid][".id"]
value = self.coordinator.data["mangle"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, False)
await self._ctrl.force_update()
self.coordinator.set_value(path, param, value, mod_param, False)
await self.coordinator.async_refresh()
async def async_turn_off(self) -> None:
"""Turn off the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["mangle"]:
if self._ctrl.data["mangle"][uid]["uniq-id"] == (
for uid in self.coordinator.data["mangle"]:
if self.coordinator.data["mangle"][uid]["uniq-id"] == (
f"{self._data['chain']},{self._data['action']},{self._data['protocol']},"
f"{self._data['src-address']}:{self._data['src-port']}-"
f"{self._data['dst-address']}:{self._data['dst-port']},"
f"{self._data['src-address-list']}-{self._data['dst-address-list']}"
):
value = self._ctrl.data["mangle"][uid][".id"]
value = self.coordinator.data["mangle"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, True)
await self._ctrl.async_update()
self.coordinator.set_value(path, param, value, mod_param, True)
await self.coordinator.async_refresh()
# ---------------------------
@ -287,43 +287,43 @@ class MikrotikFilterSwitch(MikrotikSwitch):
async def async_turn_on(self) -> None:
"""Turn on the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["filter"]:
if self._ctrl.data["filter"][uid]["uniq-id"] == (
for uid in self.coordinator.data["filter"]:
if self.coordinator.data["filter"][uid]["uniq-id"] == (
f"{self._data['chain']},{self._data['action']},{self._data['protocol']},{self._data['layer7-protocol']},"
f"{self._data['in-interface']},{self._data['in-interface-list']}:{self._data['src-address']},{self._data['src-address-list']}:{self._data['src-port']}-"
f"{self._data['out-interface']},{self._data['out-interface-list']}:{self._data['dst-address']},{self._data['dst-address-list']}:{self._data['dst-port']}"
):
value = self._ctrl.data["filter"][uid][".id"]
value = self.coordinator.data["filter"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, False)
await self._ctrl.force_update()
self.coordinator.set_value(path, param, value, mod_param, False)
await self.coordinator.async_refresh()
async def async_turn_off(self) -> None:
"""Turn off the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["filter"]:
if self._ctrl.data["filter"][uid]["uniq-id"] == (
for uid in self.coordinator.data["filter"]:
if self.coordinator.data["filter"][uid]["uniq-id"] == (
f"{self._data['chain']},{self._data['action']},{self._data['protocol']},{self._data['layer7-protocol']},"
f"{self._data['in-interface']},{self._data['in-interface-list']}:{self._data['src-address']},{self._data['src-address-list']}:{self._data['src-port']}-"
f"{self._data['out-interface']},{self._data['out-interface-list']}:{self._data['dst-address']},{self._data['dst-address-list']}:{self._data['dst-port']}"
):
value = self._ctrl.data["filter"][uid][".id"]
value = self.coordinator.data["filter"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, True)
await self._ctrl.async_update()
self.coordinator.set_value(path, param, value, mod_param, True)
await self.coordinator.async_refresh()
# ---------------------------
@ -334,35 +334,35 @@ class MikrotikQueueSwitch(MikrotikSwitch):
async def async_turn_on(self) -> None:
"""Turn on the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["queue"]:
if self._ctrl.data["queue"][uid]["name"] == f"{self._data['name']}":
value = self._ctrl.data["queue"][uid][".id"]
for uid in self.coordinator.data["queue"]:
if self.coordinator.data["queue"][uid]["name"] == f"{self._data['name']}":
value = self.coordinator.data["queue"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, False)
await self._ctrl.force_update()
self.coordinator.set_value(path, param, value, mod_param, False)
await self.coordinator.async_refresh()
async def async_turn_off(self) -> None:
"""Turn off the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = ".id"
value = None
for uid in self._ctrl.data["queue"]:
if self._ctrl.data["queue"][uid]["name"] == f"{self._data['name']}":
value = self._ctrl.data["queue"][uid][".id"]
for uid in self.coordinator.data["queue"]:
if self.coordinator.data["queue"][uid]["name"] == f"{self._data['name']}":
value = self.coordinator.data["queue"][uid][".id"]
mod_param = self.entity_description.data_switch_parameter
self._ctrl.set_value(path, param, value, mod_param, True)
await self._ctrl.async_update()
self.coordinator.set_value(path, param, value, mod_param, True)
await self.coordinator.async_refresh()
# ---------------------------
@ -373,24 +373,24 @@ class MikrotikKidcontrolPauseSwitch(MikrotikSwitch):
async def async_turn_on(self) -> None:
"""Turn on the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = self.entity_description.data_reference
value = self._data[self.entity_description.data_reference]
command = "resume"
self._ctrl.execute(path, command, param, value)
await self._ctrl.force_update()
self.coordinator.execute(path, command, param, value)
await self.coordinator.async_refresh()
async def async_turn_off(self) -> None:
"""Turn off the switch."""
if "write" not in self._ctrl.data["access"]:
if "write" not in self.coordinator.data["access"]:
return
path = self.entity_description.data_switch_path
param = self.entity_description.data_reference
value = self._data[self.entity_description.data_reference]
command = "pause"
self._ctrl.execute(path, command, param, value)
await self._ctrl.async_update()
self.coordinator.execute(path, command, param, value)
await self.coordinator.async_refresh()