2017-05-24 11:33:11 -04:00
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
2016-07-26 14:58:37 -04:00
|
|
|
from django.contrib.auth.mixins import PermissionRequiredMixin
|
2016-08-15 16:11:17 -04:00
|
|
|
from django.db.models import Count, Q
|
2016-07-26 14:58:37 -04:00
|
|
|
from django.shortcuts import get_object_or_404, render
|
2017-04-05 14:40:25 -04:00
|
|
|
from django.urls import reverse
|
2017-05-18 17:00:57 -04:00
|
|
|
from django.views.generic import View
|
2016-07-26 14:58:37 -04:00
|
|
|
|
2016-08-03 14:24:09 -04:00
|
|
|
from circuits.models import Circuit
|
|
|
|
from dcim.models import Site, Rack, Device
|
|
|
|
from ipam.models import IPAddress, Prefix, VLAN, VRF
|
2016-07-26 14:58:37 -04:00
|
|
|
from utilities.views import (
|
2017-06-05 15:53:41 -04:00
|
|
|
BulkDeleteView, BulkEditView, BulkImportView, ObjectDeleteView, ObjectEditView, ObjectListView,
|
2016-07-26 14:58:37 -04:00
|
|
|
)
|
2017-01-23 22:44:29 +01:00
|
|
|
from .models import Tenant, TenantGroup
|
2016-07-26 14:58:37 -04:00
|
|
|
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'
|
|
|
|
|
|
|
|
|
2017-06-13 16:41:57 -04:00
|
|
|
class TenantGroupCreateView(PermissionRequiredMixin, ObjectEditView):
|
|
|
|
permission_required = 'tenancy.add_tenantgroup'
|
2016-07-26 14:58:37 -04:00
|
|
|
model = TenantGroup
|
|
|
|
form_class = forms.TenantGroupForm
|
2017-01-18 14:34:17 -05:00
|
|
|
|
2017-04-13 13:11:23 -04:00
|
|
|
def get_return_url(self, request, obj):
|
2017-01-18 14:34:17 -05:00
|
|
|
return reverse('tenancy:tenantgroup_list')
|
2016-07-26 14:58:37 -04:00
|
|
|
|
|
|
|
|
2017-06-13 16:41:57 -04:00
|
|
|
class TenantGroupEditView(TenantGroupCreateView):
|
|
|
|
permission_required = 'tenancy.change_tenantgroup'
|
|
|
|
|
|
|
|
|
2016-07-26 14:58:37 -04:00
|
|
|
class TenantGroupBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
|
|
|
|
permission_required = 'tenancy.delete_tenantgroup'
|
|
|
|
cls = TenantGroup
|
2017-07-13 17:39:28 -04:00
|
|
|
queryset = TenantGroup.objects.annotate(tenant_count=Count('tenants'))
|
|
|
|
table = tables.TenantGroupTable
|
2017-01-23 14:07:26 -05:00
|
|
|
default_return_url = 'tenancy:tenantgroup_list'
|
2016-07-26 14:58:37 -04:00
|
|
|
|
|
|
|
|
|
|
|
#
|
|
|
|
# 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'
|
|
|
|
|
|
|
|
|
2017-05-18 17:00:57 -04:00
|
|
|
class TenantView(View):
|
|
|
|
|
|
|
|
def get(self, 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,
|
|
|
|
})
|
2016-07-26 14:58:37 -04:00
|
|
|
|
|
|
|
|
2017-06-13 16:41:57 -04:00
|
|
|
class TenantCreateView(PermissionRequiredMixin, ObjectEditView):
|
|
|
|
permission_required = 'tenancy.add_tenant'
|
2016-07-26 14:58:37 -04:00
|
|
|
model = Tenant
|
|
|
|
form_class = forms.TenantForm
|
|
|
|
template_name = 'tenancy/tenant_edit.html'
|
2017-01-23 14:07:26 -05:00
|
|
|
default_return_url = 'tenancy:tenant_list'
|
2016-07-26 14:58:37 -04:00
|
|
|
|
|
|
|
|
2017-06-13 16:41:57 -04:00
|
|
|
class TenantEditView(TenantCreateView):
|
|
|
|
permission_required = 'tenancy.change_tenant'
|
|
|
|
|
|
|
|
|
2016-07-26 14:58:37 -04:00
|
|
|
class TenantDeleteView(PermissionRequiredMixin, ObjectDeleteView):
|
|
|
|
permission_required = 'tenancy.delete_tenant'
|
|
|
|
model = Tenant
|
2017-01-18 13:30:19 -05:00
|
|
|
default_return_url = 'tenancy:tenant_list'
|
2016-07-26 14:58:37 -04:00
|
|
|
|
|
|
|
|
2017-06-05 15:53:41 -04:00
|
|
|
class TenantBulkImportView(PermissionRequiredMixin, BulkImportView):
|
2016-07-26 14:58:37 -04:00
|
|
|
permission_required = 'tenancy.add_tenant'
|
2017-05-31 17:40:11 -04:00
|
|
|
model_form = forms.TenantCSVForm
|
2016-07-26 14:58:37 -04:00
|
|
|
table = tables.TenantTable
|
2017-01-23 14:07:26 -05:00
|
|
|
default_return_url = 'tenancy:tenant_list'
|
2016-07-26 14:58:37 -04:00
|
|
|
|
|
|
|
|
|
|
|
class TenantBulkEditView(PermissionRequiredMixin, BulkEditView):
|
|
|
|
permission_required = 'tenancy.change_tenant'
|
|
|
|
cls = Tenant
|
2017-07-13 17:39:28 -04:00
|
|
|
queryset = Tenant.objects.select_related('group')
|
2017-01-20 16:42:11 -05:00
|
|
|
filter = filters.TenantFilter
|
2017-07-13 16:31:47 -04:00
|
|
|
table = tables.TenantTable
|
2016-07-26 14:58:37 -04:00
|
|
|
form = forms.TenantBulkEditForm
|
2017-01-23 14:07:26 -05:00
|
|
|
default_return_url = 'tenancy:tenant_list'
|
2016-07-26 14:58:37 -04:00
|
|
|
|
|
|
|
|
|
|
|
class TenantBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
|
|
|
|
permission_required = 'tenancy.delete_tenant'
|
|
|
|
cls = Tenant
|
2017-07-13 17:39:28 -04:00
|
|
|
queryset = Tenant.objects.select_related('group')
|
2017-01-20 16:42:11 -05:00
|
|
|
filter = filters.TenantFilter
|
2017-07-13 17:39:28 -04:00
|
|
|
table = tables.TenantTable
|
2017-01-23 14:07:26 -05:00
|
|
|
default_return_url = 'tenancy:tenant_list'
|