diff --git a/initializers/dcim_interfaces.yml b/initializers/dcim_interfaces.yml index 4030530..4db416f 100644 --- a/initializers/dcim_interfaces.yml +++ b/initializers/dcim_interfaces.yml @@ -8,6 +8,15 @@ ## ## Examples: +# - device: server01 +# name: ath0 +# type: 1000base-t +# lag: ae0 +# bridge: br0 +# - device: server01 +# name: ath1 +# type: 1000base-t +# parent: ath0 # - device: server01 # enabled: true # type: virtual diff --git a/startup_scripts/210_dcim_interfaces.py b/startup_scripts/210_dcim_interfaces.py index b217f3b..6e57e73 100644 --- a/startup_scripts/210_dcim_interfaces.py +++ b/startup_scripts/210_dcim_interfaces.py @@ -15,10 +15,17 @@ if interfaces is None: match_params = ["device", "name"] required_assocs = {"device": (Device, "name")} +related_assocs = { + "bridge": (Interface, "name"), + "lag": (Interface, "name"), + "parent": (Interface, "name"), +} for params in interfaces: custom_field_data = pop_custom_fields(params) + related_interfaces = {k: params.pop(k, None) for k in related_assocs} + for assoc, details in required_assocs.items(): model, field = details query = {field: params.pop(assoc)} @@ -29,6 +36,35 @@ for params in interfaces: interface, created = Interface.objects.get_or_create(**matching_params, defaults=defaults) if created: - print("🧷 Created interface", interface.name, interface.device.name) + print(f"🧷 Created interface {interface} on {interface.device}") set_custom_fields_values(interface, custom_field_data) + + for related_field, related_value in related_interfaces.items(): + if not related_value: + continue + + r_model, r_field = related_assocs[related_field] + + if related_field == "parent" and not interface.parent_id: + query = {r_field: related_value, "device": interface.device} + try: + related_obj = r_model.objects.get(**query) + except Interface.DoesNotExist: + print(f"⚠️ Could not find parent interface with: {query} for interface {interface}") + raise + + interface.parent_id = related_obj.id + interface.save() + print( + f"🧷 Attached interface {interface} on {interface.device} " + f"to parent {related_obj}" + ) + else: + query = {r_field: related_value, "device": interface.device, "type": related_field} + related_obj, rel_obj_created = r_model.objects.get_or_create(**query) + + if rel_obj_created: + setattr(interface, f"{related_field}_id", related_obj.id) + interface.save() + print(f"🧷 Created {related_field} interface {interface} on {interface.device}")