Compatible with Netbox 3.4 (#39)

Author: tobiasge

README.md

@@ -6,7 +6,7 @@ Load data from YAML files into Netbox
 
 First activate your virtual environment where Netbox is installed, the install the plugin version correspondig to your Netbox version.
 ```bash
-pip install "netbox-initializers==3.3.*"
+pip install "netbox-initializers==3.4.*"
 ```
 Then you need to add the plugin to the `PLUGINS` array in the Netbox configuration.
 ```python
@@ -36,6 +36,6 @@ The initializers where a part of the Docker image and where then extracted into
 To use the new plugin in a the Netbox Docker image, it musst be installad into the image. To this, the following example can be used as a starting point:
 
 ```dockerfile
-FROM netboxcommunity/netbox:v3.3
-RUN /opt/netbox/venv/bin/pip install "netbox-initializers==3.3.*"
+FROM netboxcommunity/netbox:v3.4
+RUN /opt/netbox/venv/bin/pip install "netbox-initializers==3.4.*"
 ```

pyproject.toml

@@ -11,7 +11,7 @@ license = "Apache-2.0"
 name = "netbox-initializers"
 readme = "README.md"
 repository = "https://github.com/tobiasge/netbox-initializers"
-version = "3.3.3"
+version = "3.4.0"
 
 [tool.poetry.dependencies]
 python = "^3.8"

src/netbox_initializers/__init__.py

@@ -5,10 +5,10 @@ class NetBoxInitializersConfig(PluginConfig):
     name = "netbox_initializers"
     verbose_name = "NetBox Initializers"
     description = "Load initial data into Netbox"
-    version = "3.3.3"
+    version = "3.4.0"
     base_url = "initializers"
-    min_version = "3.3.0"
-    max_version = "3.3.99"
+    min_version = "3.4.0"
+    max_version = "3.4.99"
 
 
 config = NetBoxInitializersConfig

src/netbox_initializers/initializers/custom_links.py

@@ -4,11 +4,12 @@
 from . import BaseInitializer, register_initializer
 
 
-def get_content_type_id(content_type):
+def get_content_type(content_type):
     try:
-        return ContentType.objects.get(model=content_type).id
+        return ContentType.objects.get(model=content_type)
     except ContentType.DoesNotExist:
         pass
+    return None
 
 
 class CustomLinkInitializer(BaseInitializer):
@@ -19,9 +20,9 @@ def load_data(self):
         if custom_links is None:
             return
         for link in custom_links:
-            content_type = link.pop("content_type")
-            link["content_type_id"] = get_content_type_id(content_type)
-            if link["content_type_id"] is None:
+            content_type_name = link.pop("content_type")
+            content_type = get_content_type(content_type_name)
+            if content_type is None:
                 print(
                     "⚠️ Unable to create Custom Link '{0}': The content_type '{1}' is unknown".format(
                         link.get("name"), content_type
@@ -35,6 +36,8 @@ def load_data(self):
             )
 
             if created:
+                custom_link.content_types.add(content_type)
+                custom_link.save()
                 print("🔗 Created Custom Link '{0}'".format(custom_link.name))
 
 

src/netbox_initializers/initializers/providers.py

@@ -1,4 +1,5 @@
 from circuits.models import Provider
+from ipam.models import ASN
 
 from . import BaseInitializer, register_initializer
 
@@ -13,10 +14,22 @@ def load_data(self):
         for params in providers:
             custom_field_data = self.pop_custom_fields(params)
 
+            asn_number = params.pop("asn")
+            asn = ASN.objects.filter(asn=asn_number).first()
+            if asn is None:
+                print(
+                    "⚠️ Unable to create Provider '{0}': The ASN '{1}' is unknown".format(
+                        params.get("name"), asn_number
+                    )
+                )
+                continue
+
             matching_params, defaults = self.split_params(params)
             provider, created = Provider.objects.get_or_create(**matching_params, defaults=defaults)
 
             if created:
+                provider.asns.add(asn)
+                provider.save()
                 print("📡 Created provider", provider.name)
 
             self.set_custom_fields_values(provider, custom_field_data)

src/netbox_initializers/initializers/yaml/providers.yml

@@ -1,6 +1,6 @@
 # - name: Provider1
 #   slug: provider1
-#   asn: 121
+#   asn: 1
 # - name: Provider2
 #   slug: provider2
-#   asn: 122
+#   asn: 3

test/Dockerfile

@@ -1,4 +1,4 @@
-FROM netboxcommunity/netbox:v3.3
+FROM netboxcommunity/netbox:v3.4
 
 COPY ../ /opt/netbox-initializers/
 COPY ./test/config/plugins.py /etc/netbox/config/

test/docker-compose.yml

@@ -15,7 +15,7 @@ services:
       dockerfile: test/Dockerfile
   # postgres
   postgres:
-    image: postgres:14-alpine
+    image: postgres:15-alpine
     env_file: env/postgres.env
     volumes:
     - netbox-postgres-data:/var/lib/postgresql/data