Merge pull request #76 from cruse1977/master

Update Images to use register_model_views

Author: cruse1977

netbox_floorplan/navigation.py

@@ -2,7 +2,7 @@
 Define the plugin menu buttons & the plugin navigation bar enteries.
 """
 
-from netbox.plugins import PluginMenuItem
+from netbox.plugins import PluginMenuItem, PluginMenuButton
 
 
 #
@@ -12,6 +12,13 @@
     PluginMenuItem(
         link="plugins:netbox_floorplan:floorplanimage_list",
         link_text="Floorplan Images",
+        buttons=(
+            PluginMenuButton(
+                link='plugins:netbox_floorplan:floorplanimage_add',
+                title='Add',
+                icon_class='mdi mdi-plus-thick',
+            ),
+        ),
     ),
 
 )

netbox_floorplan/urls.py

@@ -1,6 +1,7 @@
-from django.urls import path
+from django.urls import include, path
 from . import models, views
 from netbox.views.generic import ObjectChangeLogView
+from utilities.urls import get_model_urls
 
 urlpatterns = (
     path('floorplans/', views.FloorplanListView.as_view(), name='floorplan_list'),
@@ -10,10 +11,14 @@
     path('floorplans/<int:pk>/edit/', views.FloorplanMapEditView.as_view(), name='floorplan_edit'),
     path('floorplans/<int:pk>/delete/', views.FloorplanDeleteView.as_view(), name='floorplan_delete'),
     path('floorplans/<int:pk>/changelog/', ObjectChangeLogView.as_view(), name='floorplan_changelog', kwargs={'model': models.Floorplan}),
-    path('floorplans/floorplanimages/', views.FloorplanImageListView.as_view(), name='floorplanimage_list'),
-    path('floorplans/floorplanimages/add/', views.FloorplanImageEditView.as_view(), name='floorplanimage_add'),
-    path('floorplans/floorplanimages/<int:pk>/edit/', views.FloorplanImageEditView.as_view(), name='floorplanimage_edit'),
-    path('floorplans/floorplanimages/<int:pk>/delete/', views.FloorplanImageDeleteView.as_view(), name='floorplanimage_delete'),
-    path('floorplans/floorplanimages/<int:pk>/', views.FloorplanImageView.as_view(), name='floorplanimage'),
-    path('floorplans/floorplanimages/<int:pk>/changelog/', ObjectChangeLogView.as_view(), name='floorplanimage_changelog', kwargs={'model': models.FloorplanImage}),
+
+    # Community
+    path(
+        "floorplans/floorplanimages/",
+        include(get_model_urls("netbox_floorplan", "floorplanimage", detail=False)),
+    ),
+    path(
+        "floorplans/floorplanimages/<int:pk>/",
+        include(get_model_urls("netbox_floorplan", "floorplanimage")),
+    ),
 )

netbox_floorplan/views.py

@@ -137,20 +137,25 @@ def get(self, request):
         return super().get(request)
 
 
+@register_model_view(models.FloorplanImage)
 class FloorplanImageView(generic.ObjectView):
     queryset = models.FloorplanImage.objects.all()
 
 
+@register_model_view(models.FloorplanImage, "list", path="", detail=False)
 class FloorplanImageListView(generic.ObjectListView):
     queryset = models.FloorplanImage.objects.all()
     table = tables.FloorplanImageTable
 
 
+@register_model_view(models.FloorplanImage, "add", detail=False)
+@register_model_view(models.FloorplanImage, "edit")
 class FloorplanImageEditView(generic.ObjectEditView):
     queryset = models.FloorplanImage.objects.all()
     form = forms.FloorplanImageForm
     template_name = 'netbox_floorplan/floorplanimage_edit.html'
 
 
+@register_model_view(models.FloorplanImage, "delete")
 class FloorplanImageDeleteView(generic.ObjectDeleteView):
     queryset = models.FloorplanImage.objects.all()