mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
b01bf6089c
Conflicts: netbox/dcim/forms.py netbox/dcim/views.py netbox/ipam/forms.py netbox/templates/_base.html netbox/utilities/views.py
116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
from django.contrib.auth.mixins import PermissionRequiredMixin
|
|
from django.db.models import Count, Q
|
|
from django.shortcuts import get_object_or_404, render
|
|
from django.urls import reverse
|
|
|
|
from circuits.models import Circuit
|
|
from dcim.models import Site, Rack, Device
|
|
from ipam.models import IPAddress, Prefix, VLAN, VRF
|
|
from utilities.views import (
|
|
BulkDeleteView, BulkEditView, BulkImportView, ObjectDeleteView, ObjectEditView, ObjectListView,
|
|
)
|
|
|
|
from .models import Tenant, TenantGroup
|
|
from . import filters, forms, tables
|
|
|
|
|
|
#
|
|
# Tenant groups
|
|
#
|
|
|
|
class TenantGroupListView(ObjectListView):
|
|
queryset = TenantGroup.objects.annotate(tenant_count=Count('tenants'))
|
|
table = tables.TenantGroupTable
|
|
template_name = 'tenancy/tenantgroup_list.html'
|
|
|
|
|
|
class TenantGroupEditView(PermissionRequiredMixin, ObjectEditView):
|
|
permission_required = 'tenancy.change_tenantgroup'
|
|
model = TenantGroup
|
|
form_class = forms.TenantGroupForm
|
|
|
|
def get_return_url(self, request, obj):
|
|
return reverse('tenancy:tenantgroup_list')
|
|
|
|
|
|
class TenantGroupBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
|
|
permission_required = 'tenancy.delete_tenantgroup'
|
|
cls = TenantGroup
|
|
default_return_url = 'tenancy:tenantgroup_list'
|
|
|
|
|
|
#
|
|
# Tenants
|
|
#
|
|
|
|
class TenantListView(ObjectListView):
|
|
queryset = Tenant.objects.select_related('group')
|
|
filter = filters.TenantFilter
|
|
filter_form = forms.TenantFilterForm
|
|
table = tables.TenantTable
|
|
template_name = 'tenancy/tenant_list.html'
|
|
|
|
|
|
def tenant(request, slug):
|
|
|
|
tenant = get_object_or_404(Tenant, slug=slug)
|
|
stats = {
|
|
'site_count': Site.objects.filter(tenant=tenant).count(),
|
|
'rack_count': Rack.objects.filter(tenant=tenant).count(),
|
|
'device_count': Device.objects.filter(tenant=tenant).count(),
|
|
'vrf_count': VRF.objects.filter(tenant=tenant).count(),
|
|
'prefix_count': Prefix.objects.filter(
|
|
Q(tenant=tenant) |
|
|
Q(tenant__isnull=True, vrf__tenant=tenant)
|
|
).count(),
|
|
'ipaddress_count': IPAddress.objects.filter(
|
|
Q(tenant=tenant) |
|
|
Q(tenant__isnull=True, vrf__tenant=tenant)
|
|
).count(),
|
|
'vlan_count': VLAN.objects.filter(tenant=tenant).count(),
|
|
'circuit_count': Circuit.objects.filter(tenant=tenant).count(),
|
|
}
|
|
|
|
return render(request, 'tenancy/tenant.html', {
|
|
'tenant': tenant,
|
|
'stats': stats,
|
|
})
|
|
|
|
|
|
class TenantEditView(PermissionRequiredMixin, ObjectEditView):
|
|
permission_required = 'tenancy.change_tenant'
|
|
model = Tenant
|
|
form_class = forms.TenantForm
|
|
template_name = 'tenancy/tenant_edit.html'
|
|
default_return_url = 'tenancy:tenant_list'
|
|
|
|
|
|
class TenantDeleteView(PermissionRequiredMixin, ObjectDeleteView):
|
|
permission_required = 'tenancy.delete_tenant'
|
|
model = Tenant
|
|
default_return_url = 'tenancy:tenant_list'
|
|
|
|
|
|
class TenantBulkImportView(PermissionRequiredMixin, BulkImportView):
|
|
permission_required = 'tenancy.add_tenant'
|
|
form = forms.TenantImportForm
|
|
table = tables.TenantTable
|
|
template_name = 'tenancy/tenant_import.html'
|
|
default_return_url = 'tenancy:tenant_list'
|
|
|
|
|
|
class TenantBulkEditView(PermissionRequiredMixin, BulkEditView):
|
|
permission_required = 'tenancy.change_tenant'
|
|
cls = Tenant
|
|
filter = filters.TenantFilter
|
|
form = forms.TenantBulkEditForm
|
|
template_name = 'tenancy/tenant_bulk_edit.html'
|
|
default_return_url = 'tenancy:tenant_list'
|
|
|
|
|
|
class TenantBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
|
|
permission_required = 'tenancy.delete_tenant'
|
|
cls = Tenant
|
|
filter = filters.TenantFilter
|
|
default_return_url = 'tenancy:tenant_list'
|