diff options
Diffstat (limited to 'tools/netbox/scripts/create-switch/create-switch.py')
| -rw-r--r-- | tools/netbox/scripts/create-switch/create-switch.py | 55 |
1 files changed, 37 insertions, 18 deletions
diff --git a/tools/netbox/scripts/create-switch/create-switch.py b/tools/netbox/scripts/create-switch/create-switch.py index b4c894e..3fad7c3 100644 --- a/tools/netbox/scripts/create-switch/create-switch.py +++ b/tools/netbox/scripts/create-switch/create-switch.py @@ -29,6 +29,10 @@ DEFAULT_L1_SWITCH = Device.objects.get(name='d1.ring') # Site.objects.first() # DEFAULT_DEVICE_TYPE = DeviceType.objects.get(model='EX2200-48T') # Site.objects.first() # TODO: pick default site ? DEFAULT_NETWORK_TAGS = [Tag.objects.get(name='dhcp-client')] +DEFAULT_IPV4_RING_DELIVERY = Prefix.objects.get(prefix='151.216.160.0/20') +DEFAULT_IPV6_RING_DELIVERY = Prefix.objects.get(prefix='2a06:5841:e:2000::/52') + + UPLINK_TYPES = ( (InterfaceTypeChoices.TYPE_10GE_SFP_PLUS, '10G SFP+'), (InterfaceTypeChoices.TYPE_1GE_FIXED, '1G CAT'), @@ -200,10 +204,6 @@ class CreateSwitch(Script): def run(self, data, commit): - self.log_success(f"{self.request.__dir__()}") - self.log_success(f"{self.request.id.__dir__()}") - self.log_success(f"{self.request.user}") - self.log_success(f"{self.request.META}") # Unfuck shit # Choice var apparently only gives you a string, not an object. # Or i might be stooopid @@ -252,16 +252,25 @@ class CreateSwitch(Script): # # Prefixes Part # - - prefixes = Prefix.objects.filter( - site = data['site'], - status = PrefixStatusChoices.STATUS_CONTAINER, - #family = IPAddressFamilyChoices.FAMILY_4, - role = Role.objects.get(slug='crew').id - ) - - if len(prefixes) > 2 or len(prefixes) == 0: - raise AbortScript(f"Got two or none prefixes. Run to Simen and ask for help!") + prefixes = [] + if data['site'].slug == "ring": + prefixes.append(DEFAULT_IPV4_RING_DELIVERY) + prefixes.append(DEFAULT_IPV6_RING_DELIVERY) + else: + raise AbortScript(f"Atm, i only support provisioning on Ring") + + + # This is code for automatically choosing prefix based on Role and Site. + # We decided to hardcode this instead :) + #prefixes = Prefix.objects.filter( + # site = data['site'], + # status = PrefixStatusChoices.STATUS_CONTAINER, + # #family = IPAddressFamilyChoices.FAMILY_4, + # role = Role.objects.get(slug='crew').id + #) + + #if len(prefixes) > 2 or len(prefixes) == 0: + # raise AbortScript(f"Got two or none prefixes. Run to Simen and ask for help!") # Doesn't support anything else than crew networks for prefix in prefixes: @@ -353,8 +362,18 @@ class CreateSwitch(Script): if data['leveranse'].name == "Access Switch": dest_ae_id = vlan.vid elif data['leveranse'].name == "Distribution Switch": - dest_ae_id = str(random.randint(5000,6000)) - self.log_warning("SCRIPT IS GENERATING AE WITH RANDOM NUMBER. PLS FIX ACCORDING TO TEMPLATE :(") + match data['switch_name']: + case 'd1.bird': + dest_ae_id = '100' + case 'd1.north': + dest_ae_id = '101' + case 'd1.sponsor': + dest_ae_id = '102' + case 'd1.resepsjon': + dest_ae_id = '103' + case _: + raise AbortScript(f"NO, this 'utskutt distro' is not supported >:(") + destination_ae = Interface.objects.create( device=data['destination_device'], @@ -363,10 +382,10 @@ class CreateSwitch(Script): type=InterfaceTypeChoices.TYPE_LAG, mode=InterfaceModeChoices.MODE_TAGGED, ) - if data['leveranse'].name == "Access Switch": + if data['leveranse'].name == "Access Switch": destination_ae.tagged_vlans.add(mgmt_vlan.id) destination_ae.tagged_vlans.add(vlan.id) - self.log_success("Created AE and VLAN interfaces for both ends") + self.log_success(f"Created ae{dest_ae_id} and VLAN interfaces for both ends") mgmt_prefix_v4 = mgmt_vlan.prefixes.get(prefix__family=4) mgmt_prefix_v6 = mgmt_vlan.prefixes.get(prefix__family=6) |