Implement bulk imports and edits for SLM models (#46)

wkoot · Eric Hester · web-flow · commit a0a98bb4070e · 2024-10-14T13:41:22.000+02:00
Partially implements #4 Based on @erichester76's #44 Signed-off-by: wkoot <3715211+wkoot@users.noreply.github.com> Co-authored-by: Eric Hester <eric.hester@umbrella.tech>
diff --git a/netbox_slm/forms/__init__.py b/netbox_slm/forms/__init__.py
@@ -1,4 +1,12 @@
-from .software_license import SoftwareLicenseForm, SoftwareLicenseFilterForm
-from .software_product import SoftwareProductForm, SoftwareProductFilterForm
-from .software_product_installation import SoftwareProductInstallationForm, SoftwareProductInstallationFilterForm
-from .software_product_version import SoftwareProductVersionForm, SoftwareProductVersionFilterForm
+from .software_license import SoftwareLicenseForm, SoftwareLicenseFilterForm, SoftwareLicenseBulkImportForm
+from .software_product import SoftwareProductForm, SoftwareProductFilterForm, SoftwareProductBulkImportForm
+from .software_product_installation import (
+    SoftwareProductInstallationForm,
+    SoftwareProductInstallationFilterForm,
+    SoftwareProductInstallationBulkImportForm,
+)
+from .software_product_version import (
+    SoftwareProductVersionForm,
+    SoftwareProductVersionFilterForm,
+    SoftwareProductVersionBulkImportForm,
+)
diff --git a/netbox_slm/forms/software_license.py b/netbox_slm/forms/software_license.py
@@ -1,7 +1,7 @@
 from django.forms import DateField
 from django.urls import reverse_lazy
 
-from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm
+from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm, NetBoxModelImportForm
 from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation, SoftwareLicense
 from utilities.forms.fields import CommentField, DynamicModelChoiceField, TagFilterField, LaxURLField
 from utilities.forms.rendering import FieldSet
@@ -46,14 +46,14 @@ class Meta:
         fields = (
             "name",
             "description",
+            "software_product",
             "type",
             "stored_location",
             "stored_location_url",
             "start_date",
             "expiration_date",
             "support",
             "license_amount",
-            "software_product",
             "version",
             "installation",
             "tags",
@@ -65,3 +65,22 @@ class SoftwareLicenseFilterForm(NetBoxModelFilterSetForm):
     model = SoftwareLicense
     fieldsets = (FieldSet(None, ("q", "tag")),)
     tag = TagFilterField(model)
+
+
+class SoftwareLicenseBulkImportForm(NetBoxModelImportForm):
+    class Meta:
+        model = SoftwareLicense
+        fields = (
+            "name",
+            "description",
+            "software_product",
+            "type",
+            "start_date",
+            "expiration_date",
+            "support",
+            "license_amount",
+            "version",
+            "installation",
+            "tags",
+            "comments",
+        )
diff --git a/netbox_slm/forms/software_product.py b/netbox_slm/forms/software_product.py
@@ -1,5 +1,5 @@
 from dcim.models import Manufacturer
-from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm
+from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm, NetBoxModelImportForm
 from netbox_slm.models import SoftwareProduct
 from utilities.forms.fields import CommentField, DynamicModelChoiceField, TagFilterField
 from utilities.forms.rendering import FieldSet
@@ -30,3 +30,15 @@ class SoftwareProductFilterForm(NetBoxModelFilterSetForm):
     model = SoftwareProduct
     fieldsets = (FieldSet(None, ("q", "tag")),)
     tag = TagFilterField(model)
+
+
+class SoftwareProductBulkImportForm(NetBoxModelImportForm):
+    class Meta:
+        model = SoftwareProduct
+        fields = (
+            "name",
+            "description",
+            "manufacturer",
+            "tags",
+            "comments",
+        )
diff --git a/netbox_slm/forms/software_product_installation.py b/netbox_slm/forms/software_product_installation.py
@@ -2,7 +2,7 @@
 from django.urls import reverse_lazy
 
 from dcim.models import Device
-from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm
+from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm, NetBoxModelImportForm
 from netbox_slm.models import SoftwareProductInstallation, SoftwareProduct, SoftwareProductVersion
 from utilities.forms.fields import CommentField, DynamicModelChoiceField, TagFilterField
 from utilities.forms.rendering import FieldSet
@@ -59,3 +59,17 @@ class SoftwareProductInstallationFilterForm(NetBoxModelFilterSetForm):
     model = SoftwareProductInstallation
     fieldsets = (FieldSet(None, ("q", "tag")),)
     tag = TagFilterField(model)
+
+
+class SoftwareProductInstallationBulkImportForm(NetBoxModelImportForm):
+    class Meta:
+        model = SoftwareProductInstallation
+        fields = (
+            "device",
+            "virtualmachine",
+            "cluster",
+            "software_product",
+            "version",
+            "tags",
+            "comments",
+        )
diff --git a/netbox_slm/forms/software_product_version.py b/netbox_slm/forms/software_product_version.py
@@ -1,7 +1,7 @@
 from django.forms import DateField
 from django.urls import reverse_lazy
 
-from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm
+from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm, NetBoxModelImportForm
 from netbox_slm.models import SoftwareProduct, SoftwareProductVersion
 from utilities.forms.fields import CommentField, DynamicModelChoiceField, TagFilterField
 from utilities.forms.rendering import FieldSet
@@ -25,14 +25,14 @@ class Meta:
         model = SoftwareProductVersion
         fields = (
             "name",
+            "software_product",
             "release_date",
             "documentation_url",
             "end_of_support",
             "filename",
             "file_checksum",
             "file_link",
             "release_type",
-            "software_product",
             "tags",
             "comments",
         )
@@ -42,3 +42,19 @@ class SoftwareProductVersionFilterForm(NetBoxModelFilterSetForm):
     model = SoftwareProductVersion
     fieldsets = (FieldSet(None, ("q", "tag")),)
     tag = TagFilterField(model)
+
+
+class SoftwareProductVersionBulkImportForm(NetBoxModelImportForm):
+    class Meta:
+        model = SoftwareProductVersion
+        fields = (
+            "name",
+            "software_product",
+            "release_date",
+            "end_of_support",
+            "filename",
+            "file_checksum",
+            "release_type",
+            "tags",
+            "comments",
+        )
diff --git a/netbox_slm/navigation.py b/netbox_slm/navigation.py
@@ -15,6 +15,12 @@
                 "mdi mdi-plus-thick",
                 permissions=["netbox_slm.add_softwareproduct"],
             ),
+            PluginMenuButton(
+                "plugins:netbox_slm:softwareproduct_import",
+                "Import",
+                "mdi mdi-upload",
+                permissions=["netbox_slm.import_softwareproduct"],
+            ),
         ),
     ),
     PluginMenuItem(
@@ -28,6 +34,12 @@
                 "mdi mdi-plus-thick",
                 permissions=["netbox_slm.add_softwareproductversion"],
             ),
+            PluginMenuButton(
+                "plugins:netbox_slm:softwareproductversion_import",
+                "Import",
+                "mdi mdi-upload",
+                permissions=["netbox_slm.import_softwareproductversion"],
+            ),
         ),
     ),
     PluginMenuItem(
@@ -41,6 +53,12 @@
                 "mdi mdi-plus-thick",
                 permissions=["netbox_slm.add_softwareproductinstallation"],
             ),
+            PluginMenuButton(
+                "plugins:netbox_slm:softwareproductinstallation_import",
+                "Import",
+                "mdi mdi-upload",
+                permissions=["netbox_slm.import_softwareproductinstallation"],
+            ),
         ),
     ),
     PluginMenuItem(
@@ -54,6 +72,12 @@
                 "mdi mdi-plus-thick",
                 permissions=["netbox_slm.add_softwarelicense"],
             ),
+            PluginMenuButton(
+                "plugins:netbox_slm:softwarelicense_import",
+                "Import",
+                "mdi mdi-upload",
+                permissions=["netbox_slm.import_softwarelicense"],
+            ),
         ),
     ),
 )
diff --git a/netbox_slm/urls.py b/netbox_slm/urls.py
@@ -11,6 +11,7 @@
     path(
         "software-products/delete/", views.SoftwareProductBulkDeleteView.as_view(), name="softwareproduct_bulk_delete"
     ),
+    path("software-products/import/", views.SoftwareProductBulkImportView.as_view(), name="softwareproduct_import"),
     path("software-products/<int:pk>/", views.SoftwareProductView.as_view(), name="softwareproduct"),
     path(
         "software-products/<int:pk>/delete/", views.SoftwareProductDeleteView.as_view(), name="softwareproduct_delete"
@@ -30,6 +31,9 @@
         views.SoftwareProductVersionBulkDeleteView.as_view(),
         name="softwareproductversion_bulk_delete",
     ),
+    path(
+        "versions/import/", views.SoftwareProductVersionBulkImportView.as_view(), name="softwareproductversion_import"
+    ),
     path("versions/<int:pk>/", views.SoftwareProductVersionView.as_view(), name="softwareproductversion"),
     path(
         "versions/<int:pk>/delete/",
@@ -59,6 +63,11 @@
         views.SoftwareProductInstallationBulkDeleteView.as_view(),
         name="softwareproductinstallation_bulk_delete",
     ),
+    path(
+        "installations/import/",
+        views.SoftwareProductInstallationBulkImportView.as_view(),
+        name="softwareproductinstallation_import",
+    ),
     path(
         "installations/<int:pk>/",
         views.SoftwareProductInstallationView.as_view(),
@@ -84,6 +93,7 @@
     path("licenses/", views.SoftwareLicenseListView.as_view(), name="softwarelicense_list"),
     path("licenses/add/", views.SoftwareLicenseEditView.as_view(), name="softwarelicense_add"),
     path("licenses/delete/", views.SoftwareLicenseBulkDeleteView.as_view(), name="softwarelicense_bulk_delete"),
+    path("licenses/import/", views.SoftwareLicenseBulkImportView.as_view(), name="softwarelicense_import"),
     path("licenses/<int:pk>/", views.SoftwareLicenseView.as_view(), name="softwarelicense"),
     path("licenses/<int:pk>/delete/", views.SoftwareLicenseDeleteView.as_view(), name="softwarelicense_delete"),
     path("licenses/<int:pk>/edit/", views.SoftwareLicenseEditView.as_view(), name="softwarelicense_edit"),
diff --git a/netbox_slm/views/__init__.py b/netbox_slm/views/__init__.py
@@ -4,25 +4,29 @@
     SoftwareLicenseEditView,
     SoftwareLicenseDeleteView,
     SoftwareLicenseBulkDeleteView,
+    SoftwareLicenseBulkImportView,
 )
 from .software_product import (
     SoftwareProductListView,
     SoftwareProductView,
     SoftwareProductEditView,
     SoftwareProductDeleteView,
     SoftwareProductBulkDeleteView,
+    SoftwareProductBulkImportView,
 )
 from .software_product_installation import (
     SoftwareProductInstallationListView,
     SoftwareProductInstallationView,
     SoftwareProductInstallationEditView,
     SoftwareProductInstallationDeleteView,
     SoftwareProductInstallationBulkDeleteView,
+    SoftwareProductInstallationBulkImportView,
 )
 from .software_product_version import (
     SoftwareProductVersionListView,
     SoftwareProductVersionView,
     SoftwareProductVersionEditView,
     SoftwareProductVersionDeleteView,
     SoftwareProductVersionBulkDeleteView,
+    SoftwareProductVersionBulkImportView,
 )
diff --git a/netbox_slm/views/software_license.py b/netbox_slm/views/software_license.py
@@ -34,3 +34,9 @@ class SoftwareLicenseDeleteView(generic.ObjectDeleteView):
 class SoftwareLicenseBulkDeleteView(generic.BulkDeleteView):
     queryset = SoftwareLicense.objects.all()
     table = tables.SoftwareLicenseTable
+
+
+class SoftwareLicenseBulkImportView(generic.BulkImportView):
+    queryset = SoftwareLicense.objects.all()
+    table = tables.SoftwareLicenseTable
+    model_form = forms.SoftwareLicenseBulkImportForm
diff --git a/netbox_slm/views/software_product.py b/netbox_slm/views/software_product.py
@@ -38,3 +38,9 @@ class SoftwareProductDeleteView(generic.ObjectDeleteView):
 class SoftwareProductBulkDeleteView(generic.BulkDeleteView):
     queryset = SoftwareProduct.objects.all()
     table = tables.SoftwareProductTable
+
+
+class SoftwareProductBulkImportView(generic.BulkImportView):
+    queryset = SoftwareProduct.objects.all()
+    table = tables.SoftwareProductTable
+    model_form = forms.SoftwareProductBulkImportForm
diff --git a/netbox_slm/views/software_product_installation.py b/netbox_slm/views/software_product_installation.py
@@ -34,3 +34,9 @@ class SoftwareProductInstallationDeleteView(generic.ObjectDeleteView):
 class SoftwareProductInstallationBulkDeleteView(generic.BulkDeleteView):
     queryset = SoftwareProductInstallation.objects.all()
     table = tables.SoftwareProductInstallationTable
+
+
+class SoftwareProductInstallationBulkImportView(generic.BulkImportView):
+    queryset = SoftwareProductInstallation.objects.all()
+    table = tables.SoftwareProductInstallationTable
+    model_form = forms.SoftwareProductInstallationBulkImportForm
diff --git a/netbox_slm/views/software_product_version.py b/netbox_slm/views/software_product_version.py
@@ -38,3 +38,9 @@ class SoftwareProductVersionDeleteView(generic.ObjectDeleteView):
 class SoftwareProductVersionBulkDeleteView(generic.BulkDeleteView):
     queryset = SoftwareProductVersion.objects.all()
     table = tables.SoftwareProductVersionTable
+
+
+class SoftwareProductVersionBulkImportView(generic.BulkImportView):
+    queryset = SoftwareProductVersion.objects.all()
+    table = tables.SoftwareProductVersionTable
+    model_form = forms.SoftwareProductVersionBulkImportForm
