1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Files
netbox-community-netbox/netbox/tenancy/views.py

402 lines
13 KiB
Python
Raw Normal View History

2021-10-18 15:09:57 -04:00
from django.contrib.contenttypes.models import ContentType
from django.shortcuts import get_object_or_404
from circuits.models import Circuit
Closes: #7854 - Add VDC/Instances/etc (#10787) * Work on #7854 * Move to new URL scheme. * Fix PEP8 errors * Fix PEP8 errors * Add GraphQL and fix primary_ip missing * Fix PEP8 on GQL Type * Fix missing NestedSerializer. * Fix missing NestedSerializer & rename VDC to VDCs * Fix migration * Change Validation for identifier * Fix missing migration * Rebase to feature * Post-review changes * Remove VDC Type * Remove M2M Enforcement logic * Interface related changes * Add filter fields to filterset for Interface filter * Add form field to filterset form for Interface filter * Add VDC display to interface detail template * Remove VirtualDeviceContextTypeChoices * Accommodate recent changes in feature branch * Add tests Add missing search() * Update tests, and fix model form * Update test_api * Update test_api.InterfaceTest create_data * Fix issue with tests * Update interface serializer * Update serializer and tests * Update status to be required * Remove error message for constraint * Remove extraneous import * Re-ordered devices menu to place VDC below virtual chassis * Add helptext for `identifier` field * Fix breadcrumb link * Remove add interface link * Add missing tenant and status fields * Changes to tests as per Jeremy * Change for #9623 Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Update filterset form for status field * Remove Rename View * Change tabs to spaces * Update netbox/dcim/tables/devices.py Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Update netbox/dcim/tables/devices.py Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Fix tenant in bulk_edit * Apply suggestions from code review Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Add status field to table. * Re-order table fields. Co-authored-by: Jeremy Stretch <jstretch@ns1.com>
2022-11-11 06:55:49 -06:00
from dcim.models import Cable, Device, Location, Rack, RackReservation, Site, VirtualDeviceContext
from ipam.models import Aggregate, IPAddress, IPRange, Prefix, VLAN, VRF, ASN
2020-11-11 16:07:38 -05:00
from netbox.views import generic
2021-10-18 15:30:28 -04:00
from utilities.utils import count_related
from utilities.views import register_model_view
2019-10-07 08:29:32 +02:00
from virtualization.models import VirtualMachine, Cluster
from wireless.models import WirelessLAN, WirelessLink
2021-04-29 16:38:56 -04:00
from . import filtersets, forms, tables
2021-10-18 11:45:05 -04:00
from .models import *
2016-07-26 14:58:37 -04:00
#
# Tenant groups
#
2020-11-11 16:07:38 -05:00
class TenantGroupListView(generic.ObjectListView):
queryset = TenantGroup.objects.add_related_count(
TenantGroup.objects.all(),
Tenant,
'group',
'tenant_count',
cumulative=True
)
filterset = filtersets.TenantGroupFilterSet
filterset_form = forms.TenantGroupFilterForm
2016-07-26 14:58:37 -04:00
table = tables.TenantGroupTable
@register_model_view(TenantGroup)
class TenantGroupView(generic.ObjectView):
queryset = TenantGroup.objects.all()
def get_extra_context(self, request, instance):
tenants = Tenant.objects.restrict(request.user, 'view').filter(
group=instance
)
tenants_table = tables.TenantTable(tenants, user=request.user, exclude=('group',))
tenants_table.configure(request)
return {
'tenants_table': tenants_table,
}
@register_model_view(TenantGroup, 'edit')
2020-11-11 16:07:38 -05:00
class TenantGroupEditView(generic.ObjectEditView):
queryset = TenantGroup.objects.all()
form = forms.TenantGroupForm
2016-07-26 14:58:37 -04:00
@register_model_view(TenantGroup, 'delete')
2020-11-11 16:07:38 -05:00
class TenantGroupDeleteView(generic.ObjectDeleteView):
queryset = TenantGroup.objects.all()
2020-11-11 16:07:38 -05:00
class TenantGroupBulkImportView(generic.BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = TenantGroup.objects.all()
model_form = forms.TenantGroupImportForm
table = tables.TenantGroupTable
class TenantGroupBulkEditView(generic.BulkEditView):
queryset = TenantGroup.objects.add_related_count(
TenantGroup.objects.all(),
Tenant,
'group',
'tenant_count',
cumulative=True
)
2021-04-29 16:38:56 -04:00
filterset = filtersets.TenantGroupFilterSet
table = tables.TenantGroupTable
form = forms.TenantGroupBulkEditForm
2020-11-11 16:07:38 -05:00
class TenantGroupBulkDeleteView(generic.BulkDeleteView):
queryset = TenantGroup.objects.add_related_count(
TenantGroup.objects.all(),
Tenant,
'group',
'tenant_count',
cumulative=True
)
table = tables.TenantGroupTable
2016-07-26 14:58:37 -04:00
#
# Tenants
#
2020-11-11 16:07:38 -05:00
class TenantListView(generic.ObjectListView):
queryset = Tenant.objects.all()
2021-04-29 16:38:56 -04:00
filterset = filtersets.TenantFilterSet
filterset_form = forms.TenantFilterForm
2016-07-26 14:58:37 -04:00
table = tables.TenantTable
@register_model_view(Tenant)
2020-11-11 16:07:38 -05:00
class TenantView(generic.ObjectView):
queryset = Tenant.objects.all()
def get_extra_context(self, request, instance):
stats = {
'site_count': Site.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'rack_count': Rack.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'rackreservation_count': RackReservation.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'location_count': Location.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'device_count': Device.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
Closes: #7854 - Add VDC/Instances/etc (#10787) * Work on #7854 * Move to new URL scheme. * Fix PEP8 errors * Fix PEP8 errors * Add GraphQL and fix primary_ip missing * Fix PEP8 on GQL Type * Fix missing NestedSerializer. * Fix missing NestedSerializer & rename VDC to VDCs * Fix migration * Change Validation for identifier * Fix missing migration * Rebase to feature * Post-review changes * Remove VDC Type * Remove M2M Enforcement logic * Interface related changes * Add filter fields to filterset for Interface filter * Add form field to filterset form for Interface filter * Add VDC display to interface detail template * Remove VirtualDeviceContextTypeChoices * Accommodate recent changes in feature branch * Add tests Add missing search() * Update tests, and fix model form * Update test_api * Update test_api.InterfaceTest create_data * Fix issue with tests * Update interface serializer * Update serializer and tests * Update status to be required * Remove error message for constraint * Remove extraneous import * Re-ordered devices menu to place VDC below virtual chassis * Add helptext for `identifier` field * Fix breadcrumb link * Remove add interface link * Add missing tenant and status fields * Changes to tests as per Jeremy * Change for #9623 Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Update filterset form for status field * Remove Rename View * Change tabs to spaces * Update netbox/dcim/tables/devices.py Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Update netbox/dcim/tables/devices.py Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Fix tenant in bulk_edit * Apply suggestions from code review Co-authored-by: Jeremy Stretch <jstretch@ns1.com> * Add status field to table. * Re-order table fields. Co-authored-by: Jeremy Stretch <jstretch@ns1.com>
2022-11-11 06:55:49 -06:00
'vdc_count': VirtualDeviceContext.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'vrf_count': VRF.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'aggregate_count': Aggregate.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'prefix_count': Prefix.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'iprange_count': IPRange.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'ipaddress_count': IPAddress.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'vlan_count': VLAN.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'circuit_count': Circuit.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'virtualmachine_count': VirtualMachine.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'cluster_count': Cluster.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'cable_count': Cable.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'asn_count': ASN.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'wirelesslan_count': WirelessLAN.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
'wirelesslink_count': WirelessLink.objects.restrict(request.user, 'view').filter(tenant=instance).count(),
}
return {
'stats': stats,
}
2016-07-26 14:58:37 -04:00
@register_model_view(Tenant, 'edit')
2020-11-11 16:07:38 -05:00
class TenantEditView(generic.ObjectEditView):
queryset = Tenant.objects.all()
form = forms.TenantForm
2016-07-26 14:58:37 -04:00
@register_model_view(Tenant, 'delete')
2020-11-11 16:07:38 -05:00
class TenantDeleteView(generic.ObjectDeleteView):
queryset = Tenant.objects.all()
2016-07-26 14:58:37 -04:00
2020-11-11 16:07:38 -05:00
class TenantBulkImportView(generic.BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = Tenant.objects.all()
model_form = forms.TenantImportForm
2016-07-26 14:58:37 -04:00
table = tables.TenantTable
2020-11-11 16:07:38 -05:00
class TenantBulkEditView(generic.BulkEditView):
queryset = Tenant.objects.all()
2021-04-29 16:38:56 -04:00
filterset = filtersets.TenantFilterSet
table = tables.TenantTable
2016-07-26 14:58:37 -04:00
form = forms.TenantBulkEditForm
2020-11-11 16:07:38 -05:00
class TenantBulkDeleteView(generic.BulkDeleteView):
queryset = Tenant.objects.all()
2021-04-29 16:38:56 -04:00
filterset = filtersets.TenantFilterSet
table = tables.TenantTable
2021-10-18 11:45:05 -04:00
#
# Contact groups
#
class ContactGroupListView(generic.ObjectListView):
queryset = ContactGroup.objects.add_related_count(
ContactGroup.objects.all(),
Contact,
'group',
'contact_count',
cumulative=True
)
filterset = filtersets.ContactGroupFilterSet
filterset_form = forms.ContactGroupFilterForm
table = tables.ContactGroupTable
@register_model_view(ContactGroup)
2021-10-18 11:45:05 -04:00
class ContactGroupView(generic.ObjectView):
queryset = ContactGroup.objects.all()
def get_extra_context(self, request, instance):
child_groups = ContactGroup.objects.add_related_count(
ContactGroup.objects.all(),
Contact,
'group',
'contact_count',
cumulative=True
).restrict(request.user, 'view').filter(
parent__in=instance.get_descendants(include_self=True)
)
child_groups_table = tables.ContactGroupTable(child_groups)
child_groups_table.columns.hide('actions')
2021-10-18 11:45:05 -04:00
contacts = Contact.objects.restrict(request.user, 'view').filter(
group=instance
).annotate(
assignment_count=count_related(ContactAssignment, 'contact')
2021-10-18 11:45:05 -04:00
)
contacts_table = tables.ContactTable(contacts, user=request.user, exclude=('group',))
contacts_table.configure(request)
2021-10-18 11:45:05 -04:00
return {
'child_groups_table': child_groups_table,
2021-10-18 11:45:05 -04:00
'contacts_table': contacts_table,
}
@register_model_view(ContactGroup, 'edit')
2021-10-18 11:45:05 -04:00
class ContactGroupEditView(generic.ObjectEditView):
queryset = ContactGroup.objects.all()
form = forms.ContactGroupForm
2021-10-18 11:45:05 -04:00
@register_model_view(ContactGroup, 'delete')
2021-10-18 11:45:05 -04:00
class ContactGroupDeleteView(generic.ObjectDeleteView):
queryset = ContactGroup.objects.all()
class ContactGroupBulkImportView(generic.BulkImportView):
queryset = ContactGroup.objects.all()
model_form = forms.ContactGroupImportForm
2021-10-18 11:45:05 -04:00
table = tables.ContactGroupTable
class ContactGroupBulkEditView(generic.BulkEditView):
queryset = ContactGroup.objects.add_related_count(
ContactGroup.objects.all(),
Contact,
'group',
'contact_count',
cumulative=True
)
filterset = filtersets.ContactGroupFilterSet
table = tables.ContactGroupTable
form = forms.ContactGroupBulkEditForm
class ContactGroupBulkDeleteView(generic.BulkDeleteView):
queryset = ContactGroup.objects.add_related_count(
ContactGroup.objects.all(),
Contact,
'group',
'contact_count',
cumulative=True
)
table = tables.ContactGroupTable
#
# Contact roles
#
class ContactRoleListView(generic.ObjectListView):
queryset = ContactRole.objects.all()
filterset = filtersets.ContactRoleFilterSet
filterset_form = forms.ContactRoleFilterForm
table = tables.ContactRoleTable
@register_model_view(ContactRole)
2021-10-18 11:45:05 -04:00
class ContactRoleView(generic.ObjectView):
queryset = ContactRole.objects.all()
def get_extra_context(self, request, instance):
contact_assignments = ContactAssignment.objects.restrict(request.user, 'view').filter(
role=instance
)
contacts_table = tables.ContactAssignmentTable(contact_assignments, user=request.user)
2021-10-18 15:30:28 -04:00
contacts_table.columns.hide('role')
contacts_table.configure(request)
2021-10-18 11:45:05 -04:00
return {
'contacts_table': contacts_table,
2021-10-18 15:30:28 -04:00
'assignment_count': ContactAssignment.objects.filter(role=instance).count(),
2021-10-18 11:45:05 -04:00
}
@register_model_view(ContactRole, 'edit')
2021-10-18 11:45:05 -04:00
class ContactRoleEditView(generic.ObjectEditView):
queryset = ContactRole.objects.all()
form = forms.ContactRoleForm
2021-10-18 11:45:05 -04:00
@register_model_view(ContactRole, 'delete')
2021-10-18 11:45:05 -04:00
class ContactRoleDeleteView(generic.ObjectDeleteView):
queryset = ContactRole.objects.all()
class ContactRoleBulkImportView(generic.BulkImportView):
queryset = ContactRole.objects.all()
model_form = forms.ContactRoleImportForm
2021-10-18 11:45:05 -04:00
table = tables.ContactRoleTable
class ContactRoleBulkEditView(generic.BulkEditView):
queryset = ContactRole.objects.all()
filterset = filtersets.ContactRoleFilterSet
table = tables.ContactRoleTable
form = forms.ContactRoleBulkEditForm
class ContactRoleBulkDeleteView(generic.BulkDeleteView):
queryset = ContactRole.objects.all()
table = tables.ContactRoleTable
#
# Contacts
#
class ContactListView(generic.ObjectListView):
2021-10-18 15:30:28 -04:00
queryset = Contact.objects.annotate(
assignment_count=count_related(ContactAssignment, 'contact')
)
2021-10-18 11:45:05 -04:00
filterset = filtersets.ContactFilterSet
filterset_form = forms.ContactFilterForm
table = tables.ContactTable
@register_model_view(Contact)
2021-10-18 11:45:05 -04:00
class ContactView(generic.ObjectView):
queryset = Contact.objects.all()
2021-10-18 15:30:28 -04:00
def get_extra_context(self, request, instance):
contact_assignments = ContactAssignment.objects.restrict(request.user, 'view').filter(
contact=instance
)
assignments_table = tables.ContactAssignmentTable(contact_assignments, user=request.user)
assignments_table.columns.hide('contact')
assignments_table.configure(request)
2021-10-18 15:30:28 -04:00
return {
'assignments_table': assignments_table,
2021-10-18 15:30:28 -04:00
'assignment_count': ContactAssignment.objects.filter(contact=instance).count(),
}
2021-10-18 11:45:05 -04:00
@register_model_view(Contact, 'edit')
2021-10-18 11:45:05 -04:00
class ContactEditView(generic.ObjectEditView):
queryset = Contact.objects.all()
form = forms.ContactForm
2021-10-18 11:45:05 -04:00
@register_model_view(Contact, 'delete')
2021-10-18 11:45:05 -04:00
class ContactDeleteView(generic.ObjectDeleteView):
queryset = Contact.objects.all()
class ContactBulkImportView(generic.BulkImportView):
queryset = Contact.objects.all()
model_form = forms.ContactImportForm
2021-10-18 11:45:05 -04:00
table = tables.ContactTable
class ContactBulkEditView(generic.BulkEditView):
queryset = Contact.objects.annotate(
assignment_count=count_related(ContactAssignment, 'contact')
)
2021-10-18 11:45:05 -04:00
filterset = filtersets.ContactFilterSet
table = tables.ContactTable
form = forms.ContactBulkEditForm
class ContactBulkDeleteView(generic.BulkDeleteView):
queryset = Contact.objects.annotate(
assignment_count=count_related(ContactAssignment, 'contact')
)
2021-10-18 11:45:05 -04:00
filterset = filtersets.ContactFilterSet
table = tables.ContactTable
2021-10-18 15:09:57 -04:00
#
# Contact assignments
#
@register_model_view(ContactAssignment, 'edit')
2021-10-18 15:09:57 -04:00
class ContactAssignmentEditView(generic.ObjectEditView):
queryset = ContactAssignment.objects.all()
form = forms.ContactAssignmentForm
template_name = 'tenancy/contactassignment_edit.html'
2021-10-18 15:09:57 -04:00
2021-12-14 11:28:13 -05:00
def alter_object(self, instance, request, args, kwargs):
2021-10-18 15:09:57 -04:00
if not instance.pk:
# Assign the object based on URL kwargs
content_type = get_object_or_404(ContentType, pk=request.GET.get('content_type'))
2021-10-18 15:09:57 -04:00
instance.object = get_object_or_404(content_type.model_class(), pk=request.GET.get('object_id'))
return instance
2022-08-23 09:29:55 -07:00
def get_extra_addanother_params(self, request):
return {
'content_type': request.GET.get('content_type'),
'object_id': request.GET.get('object_id'),
}
2022-08-22 16:31:41 -07:00
2021-10-18 15:09:57 -04:00
@register_model_view(ContactAssignment, 'delete')
2021-10-18 15:09:57 -04:00
class ContactAssignmentDeleteView(generic.ObjectDeleteView):
queryset = ContactAssignment.objects.all()