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

simplified update_items

Browse source
This commit is contained in:
Tomaae 2019-12-09 09:52:45 +01:00 • committed by GitHub
parent c49e81a208
commit 5de05f1927
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

15
custom_components/mikrotik_router/switch.py
View file

@ -99,7 +99,10 @@ def update_items(inst, mikrotik_controller, async_add_entities, switches):
new_switches = []
# Add switches
for sid in SWITCH_LIST:
for sid, sid_func in zip(
["interface", "nat", "script"],
[MikrotikControllerPortSwitch, MikrotikControllerNATSwitch, MikrotikControllerScriptSwitch]
):
for uid in mikrotik_controller.data[sid]:
item_id = "{}-{}-{}".format(inst, sid, mikrotik_controller.data[sid][uid]['name'])
if item_id in switches:
@ -107,15 +110,7 @@ def update_items(inst, mikrotik_controller, async_add_entities, switches):
switches[item_id].async_schedule_update_ha_state()
continue
if sid == 'interface':
switches[item_id] = MikrotikControllerPortSwitch(inst, uid, mikrotik_controller)
if sid == 'nat':
switches[item_id] = MikrotikControllerNATSwitch(inst, uid, mikrotik_controller)
if sid == 'script':
switches[item_id] = MikrotikControllerScriptSwitch(inst, uid, mikrotik_controller)
switches[item_id] = sid_func(inst, uid, mikrotik_controller)
new_switches.append(switches[item_id])
if new_switches: