From 797f26b58e1c91876ff5de63590ab32aca0b0f29 Mon Sep 17 00:00:00 2001 From: Timotej Lazar Date: Mon, 10 Feb 2025 16:40:49 +0100 Subject: [PATCH] vlans: allow specifying native VLAN for tagged interfaces --- vlans.py | 35 ++++++++++++++++------------------- 1 file changed, 16 insertions(+), 19 deletions(-) diff --git a/vlans.py b/vlans.py index 2736836..c09e8c4 100644 --- a/vlans.py +++ b/vlans.py @@ -119,8 +119,13 @@ class SetVLANScript(Script): def run(self, data, commit): all_ports = list(data['switch_ports']) + untagged_vlan = data['untagged_vlan'] + tagged_vlans = list(data['tagged_vlans']) modified_switches = set() + if not untagged_vlan and not tagged_vlans: + raise AbortScript('at least one VLAN must be specified') + # Trace doesn’t work rear ports for some reason, so do it manually. # Assumes this layout (f=front port, r=rear port, i=interface, ---=cable): # 1f:012.23:r1 --- 23r:panel-012:f23 --- 46i:sw-xyzzy @@ -132,27 +137,19 @@ class SetVLANScript(Script): for port in all_ports: port.enabled = data['enable'] - match len(data['vlans']): - case 0: - port.mode = 'access' - port.save() - port.tagged_vlans.clear() - port.untagged_vlan = None - case 1: - port.mode = 'access' - port.save() - port.tagged_vlans.clear() - port.untagged_vlan = data['vlans'][0] - case _: - port.mode = 'tagged' - port.save() - port.tagged_vlans.set(data['vlans']) - port.untagged_vlan = None + if tagged_vlans: + self.log_info(f'{port.device.name} {port} is tagged') + port.mode = 'tagged' + port.save() + port.tagged_vlans.set(tagged_vlans) + else: + self.log_info(f'{port.device.name} {port} is access') + port.mode = 'access' + port.save() + port.untagged_vlan = untagged_vlan port.full_clean() port.save() modified_switches.add(port.device.name) - - self.log_info(f'{port.device.name} {port} is {port.mode} for {",".join(str(vlan.vid) for vlan in data["vlans"])}') + self.log_info(f'{port.device.name} {port} is {port.mode} with untagged_vlan: {port.untagged_vlan} tagged_vlans: {[v.vid for v in port.tagged_vlans.all()]}') self.log_success(f'modified switches {",".join(sorted(modified_switches))}') -