Closes #10608: Register all core model views using register_model_view()

netbox/tenancy/views.py

@@ -1,5 +1,4 @@
 from django.contrib.contenttypes.models import ContentType
-from django.http import QueryDict
 from django.shortcuts import get_object_or_404
 
 from circuits.models import Circuit
@@ -7,6 +6,7 @@ from dcim.models import Cable, Device, Location, Rack, RackReservation, Site
 from ipam.models import Aggregate, IPAddress, IPRange, Prefix, VLAN, VRF, ASN
 from netbox.views import generic
 from utilities.utils import count_related
+from utilities.views import register_model_view
 from virtualization.models import VirtualMachine, Cluster
 from wireless.models import WirelessLAN, WirelessLink
 from . import filtersets, forms, tables
@@ -30,6 +30,7 @@ class TenantGroupListView(generic.ObjectListView):
     table = tables.TenantGroupTable
 
 
+@register_model_view(TenantGroup)
 class TenantGroupView(generic.ObjectView):
     queryset = TenantGroup.objects.all()
 
@@ -45,11 +46,13 @@ class TenantGroupView(generic.ObjectView):
         }
 
 
+@register_model_view(TenantGroup, 'edit')
 class TenantGroupEditView(generic.ObjectEditView):
     queryset = TenantGroup.objects.all()
     form = forms.TenantGroupForm
 
 
+@register_model_view(TenantGroup, 'delete')
 class TenantGroupDeleteView(generic.ObjectDeleteView):
     queryset = TenantGroup.objects.all()
 
@@ -95,6 +98,7 @@ class TenantListView(generic.ObjectListView):
     table = tables.TenantTable
 
 
+@register_model_view(Tenant)
 class TenantView(generic.ObjectView):
     queryset = Tenant.objects.all()
 
@@ -125,11 +129,13 @@ class TenantView(generic.ObjectView):
         }
 
 
+@register_model_view(Tenant, 'edit')
 class TenantEditView(generic.ObjectEditView):
     queryset = Tenant.objects.all()
     form = forms.TenantForm
 
 
+@register_model_view(Tenant, 'delete')
 class TenantDeleteView(generic.ObjectDeleteView):
     queryset = Tenant.objects.all()
 
@@ -170,6 +176,7 @@ class ContactGroupListView(generic.ObjectListView):
     table = tables.ContactGroupTable
 
 
+@register_model_view(ContactGroup)
 class ContactGroupView(generic.ObjectView):
     queryset = ContactGroup.objects.all()
 
@@ -198,11 +205,13 @@ class ContactGroupView(generic.ObjectView):
         }
 
 
+@register_model_view(ContactGroup, 'edit')
 class ContactGroupEditView(generic.ObjectEditView):
     queryset = ContactGroup.objects.all()
     form = forms.ContactGroupForm
 
 
+@register_model_view(ContactGroup, 'delete')
 class ContactGroupDeleteView(generic.ObjectDeleteView):
     queryset = ContactGroup.objects.all()
 
@@ -248,6 +257,7 @@ class ContactRoleListView(generic.ObjectListView):
     table = tables.ContactRoleTable
 
 
+@register_model_view(ContactRole)
 class ContactRoleView(generic.ObjectView):
     queryset = ContactRole.objects.all()
 
@@ -265,11 +275,13 @@ class ContactRoleView(generic.ObjectView):
         }
 
 
+@register_model_view(ContactRole, 'edit')
 class ContactRoleEditView(generic.ObjectEditView):
     queryset = ContactRole.objects.all()
     form = forms.ContactRoleForm
 
 
+@register_model_view(ContactRole, 'delete')
 class ContactRoleDeleteView(generic.ObjectDeleteView):
     queryset = ContactRole.objects.all()
 
@@ -305,6 +317,7 @@ class ContactListView(generic.ObjectListView):
     table = tables.ContactTable
 
 
+@register_model_view(Contact)
 class ContactView(generic.ObjectView):
     queryset = Contact.objects.all()
 
@@ -322,11 +335,13 @@ class ContactView(generic.ObjectView):
         }
 
 
+@register_model_view(Contact, 'edit')
 class ContactEditView(generic.ObjectEditView):
     queryset = Contact.objects.all()
     form = forms.ContactForm
 
 
+@register_model_view(Contact, 'delete')
 class ContactDeleteView(generic.ObjectDeleteView):
     queryset = Contact.objects.all()
 
@@ -354,6 +369,7 @@ class ContactBulkDeleteView(generic.BulkDeleteView):
 # Contact assignments
 #
 
+@register_model_view(ContactAssignment, 'edit')
 class ContactAssignmentEditView(generic.ObjectEditView):
     queryset = ContactAssignment.objects.all()
     form = forms.ContactAssignmentForm
@@ -373,5 +389,6 @@ class ContactAssignmentEditView(generic.ObjectEditView):
         }
 
 
+@register_model_view(ContactAssignment, 'delete')
 class ContactAssignmentDeleteView(generic.ObjectDeleteView):
     queryset = ContactAssignment.objects.all()