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/ipam/views.py

848 lines
28 KiB
Python
Raw Normal View History

2016-11-04 16:04:29 -04:00
import netaddr
from django.conf import settings
from django.db.models import Count, Prefetch
from django.db.models.expressions import RawSQL
from django.shortcuts import get_object_or_404, redirect, render
from django_tables2 import RequestConfig
2016-03-01 11:23:03 -05:00
from dcim.models import Device, Interface
2016-03-01 11:23:03 -05:00
from utilities.paginator import EnhancedPaginator
2016-05-18 16:20:30 -04:00
from utilities.views import (
BulkCreateView, BulkDeleteView, BulkEditView, BulkImportView, ObjectView, ObjectDeleteView, ObjectEditView,
2020-06-01 11:43:49 -04:00
ObjectListView,
2016-05-18 16:20:30 -04:00
)
2020-06-24 09:52:22 -04:00
from virtualization.models import VirtualMachine, VMInterface
2016-05-18 16:20:30 -04:00
from . import filters, forms, tables
2019-11-27 22:09:16 -05:00
from .choices import *
from .constants import *
from .models import Aggregate, IPAddress, Prefix, RIR, Role, Service, VLAN, VLANGroup, VRF
2020-06-26 11:56:30 -04:00
from .utils import add_available_ipaddresses, add_available_prefixes, add_available_vlans
2016-03-01 11:23:03 -05:00
#
# VRFs
#
class VRFListView(ObjectListView):
queryset = VRF.objects.prefetch_related('tenant')
filterset = filters.VRFFilterSet
filterset_form = forms.VRFFilterForm
2016-05-18 16:20:30 -04:00
table = tables.VRFTable
2016-03-01 11:23:03 -05:00
class VRFView(ObjectView):
queryset = VRF.objects.all()
2016-03-01 11:23:03 -05:00
def get(self, request, pk):
2016-03-01 11:23:03 -05:00
vrf = get_object_or_404(self.queryset, pk=pk)
2020-06-01 11:43:49 -04:00
prefix_count = Prefix.objects.restrict(request.user, 'view').filter(vrf=vrf).count()
return render(request, 'ipam/vrf.html', {
'vrf': vrf,
'prefix_count': prefix_count,
})
2016-03-01 11:23:03 -05:00
class VRFEditView(ObjectEditView):
queryset = VRF.objects.all()
model_form = forms.VRFForm
template_name = 'ipam/vrf_edit.html'
2016-03-01 11:23:03 -05:00
class VRFDeleteView(ObjectDeleteView):
queryset = VRF.objects.all()
2016-03-01 11:23:03 -05:00
class VRFBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = VRF.objects.all()
model_form = forms.VRFCSVForm
2016-05-18 16:20:30 -04:00
table = tables.VRFTable
2016-03-01 11:23:03 -05:00
class VRFBulkEditView(BulkEditView):
queryset = VRF.objects.prefetch_related('tenant')
filterset = filters.VRFFilterSet
table = tables.VRFTable
2016-05-18 16:20:30 -04:00
form = forms.VRFBulkEditForm
2016-03-01 11:23:03 -05:00
class VRFBulkDeleteView(BulkDeleteView):
queryset = VRF.objects.prefetch_related('tenant')
filterset = filters.VRFFilterSet
table = tables.VRFTable
2016-03-01 11:23:03 -05:00
2016-05-16 13:04:45 -04:00
#
# RIRs
#
class RIRListView(ObjectListView):
2016-05-16 13:04:45 -04:00
queryset = RIR.objects.annotate(aggregate_count=Count('aggregates'))
filterset = filters.RIRFilterSet
filterset_form = forms.RIRFilterForm
table = tables.RIRDetailTable
2016-05-16 13:04:45 -04:00
template_name = 'ipam/rir_list.html'
2016-11-11 12:45:24 -05:00
def alter_queryset(self, request):
if request.GET.get('family') == '6':
family = 6
denominator = 2 ** 64 # Count /64s for IPv6 rather than individual IPs
else:
family = 4
denominator = 1
2016-11-11 12:45:24 -05:00
rirs = []
for rir in self.queryset:
stats = {
'total': 0,
'active': 0,
'reserved': 0,
'deprecated': 0,
'available': 0,
}
aggregate_list = Aggregate.objects.restrict(request.user).filter(prefix__family=family, rir=rir)
2016-11-11 12:45:24 -05:00
for aggregate in aggregate_list:
queryset = Prefix.objects.restrict(request.user).filter(
prefix__net_contained_or_equal=str(aggregate.prefix)
)
2016-11-11 12:45:24 -05:00
# Find all consumed space for each prefix status (we ignore containers for this purpose).
active_prefixes = netaddr.cidr_merge(
2019-11-27 21:46:53 -05:00
[p.prefix for p in queryset.filter(status=PrefixStatusChoices.STATUS_ACTIVE)]
)
reserved_prefixes = netaddr.cidr_merge(
2019-11-27 21:46:53 -05:00
[p.prefix for p in queryset.filter(status=PrefixStatusChoices.STATUS_RESERVED)]
)
deprecated_prefixes = netaddr.cidr_merge(
2019-11-27 21:46:53 -05:00
[p.prefix for p in queryset.filter(status=PrefixStatusChoices.STATUS_DEPRECATED)]
)
2016-11-11 12:45:24 -05:00
# Find all available prefixes by subtracting each of the existing prefix sets from the aggregate prefix.
available_prefixes = (
2016-11-11 15:09:25 -05:00
netaddr.IPSet([aggregate.prefix]) -
netaddr.IPSet(active_prefixes) -
netaddr.IPSet(reserved_prefixes) -
netaddr.IPSet(deprecated_prefixes)
2016-11-11 12:45:24 -05:00
)
# Add the size of each metric to the RIR total.
stats['total'] += int(aggregate.prefix.size / denominator)
stats['active'] += int(netaddr.IPSet(active_prefixes).size / denominator)
stats['reserved'] += int(netaddr.IPSet(reserved_prefixes).size / denominator)
stats['deprecated'] += int(netaddr.IPSet(deprecated_prefixes).size / denominator)
stats['available'] += int(available_prefixes.size / denominator)
2016-11-11 12:45:24 -05:00
# Calculate the percentage of total space for each prefix status.
total = float(stats['total'])
stats['percentages'] = {
'active': float('{:.2f}'.format(stats['active'] / total * 100)) if total else 0,
'reserved': float('{:.2f}'.format(stats['reserved'] / total * 100)) if total else 0,
'deprecated': float('{:.2f}'.format(stats['deprecated'] / total * 100)) if total else 0,
}
stats['percentages']['available'] = (
2016-11-11 15:09:25 -05:00
100 -
stats['percentages']['active'] -
stats['percentages']['reserved'] -
stats['percentages']['deprecated']
2016-11-11 12:45:24 -05:00
)
rir.stats = stats
rirs.append(rir)
return rirs
2016-05-16 13:04:45 -04:00
class RIREditView(ObjectEditView):
queryset = RIR.objects.all()
model_form = forms.RIRForm
2016-05-16 13:04:45 -04:00
class RIRBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = RIR.objects.all()
model_form = forms.RIRCSVForm
table = tables.RIRTable
class RIRBulkDeleteView(BulkDeleteView):
queryset = RIR.objects.annotate(aggregate_count=Count('aggregates'))
filterset = filters.RIRFilterSet
table = tables.RIRTable
2016-05-16 13:04:45 -04:00
2016-03-01 11:23:03 -05:00
#
# Aggregates
#
class AggregateListView(ObjectListView):
queryset = Aggregate.objects.prefetch_related('rir').annotate(
child_count=RawSQL('SELECT COUNT(*) FROM ipam_prefix WHERE ipam_prefix.prefix <<= ipam_aggregate.prefix', ())
)
filterset = filters.AggregateFilterSet
filterset_form = forms.AggregateFilterForm
table = tables.AggregateDetailTable
template_name = 'ipam/aggregate_list.html'
2016-03-01 11:23:03 -05:00
def extra_context(self):
ipv4_total = 0
ipv6_total = 0
for aggregate in self.queryset:
if aggregate.prefix.version == 6:
# Report equivalent /64s for IPv6 to keep things sane
ipv6_total += int(aggregate.prefix.size / 2 ** 64)
else:
ipv4_total += aggregate.prefix.size
return {
'ipv4_total': ipv4_total,
'ipv6_total': ipv6_total,
}
2016-03-01 11:23:03 -05:00
class AggregateView(ObjectView):
queryset = Aggregate.objects.all()
def get(self, request, pk):
2016-03-01 11:23:03 -05:00
aggregate = get_object_or_404(self.queryset, pk=pk)
2016-03-01 11:23:03 -05:00
# Find all child prefixes contained by this aggregate
2020-06-01 11:43:49 -04:00
child_prefixes = Prefix.objects.restrict(request.user, 'view').filter(
prefix__net_contained_or_equal=str(aggregate.prefix)
).prefetch_related(
'site', 'role'
).annotate_depth(
limit=0
)
# Add available prefixes to the table if requested
2020-01-07 17:58:30 +00:00
if request.GET.get('show_available', 'true') == 'true':
child_prefixes = add_available_prefixes(aggregate.prefix, child_prefixes)
2016-03-01 11:23:03 -05:00
prefix_table = tables.PrefixDetailTable(child_prefixes)
if request.user.has_perm('ipam.change_prefix') or request.user.has_perm('ipam.delete_prefix'):
prefix_table.columns.show('pk')
paginate = {
2018-11-02 14:27:17 -04:00
'paginator_class': EnhancedPaginator,
'per_page': request.GET.get('per_page', settings.PAGINATE_COUNT)
}
RequestConfig(request, paginate).configure(prefix_table)
2016-03-01 11:23:03 -05:00
# Compile permissions list for rendering the object table
permissions = {
'add': request.user.has_perm('ipam.add_prefix'),
'change': request.user.has_perm('ipam.change_prefix'),
'delete': request.user.has_perm('ipam.delete_prefix'),
}
return render(request, 'ipam/aggregate.html', {
'aggregate': aggregate,
'prefix_table': prefix_table,
'permissions': permissions,
2020-01-07 17:58:30 +00:00
'show_available': request.GET.get('show_available', 'true') == 'true',
})
2016-03-01 11:23:03 -05:00
class AggregateEditView(ObjectEditView):
queryset = Aggregate.objects.all()
model_form = forms.AggregateForm
template_name = 'ipam/aggregate_edit.html'
2016-03-01 11:23:03 -05:00
class AggregateDeleteView(ObjectDeleteView):
queryset = Aggregate.objects.all()
2016-03-01 11:23:03 -05:00
class AggregateBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = Aggregate.objects.all()
model_form = forms.AggregateCSVForm
2016-05-18 16:20:30 -04:00
table = tables.AggregateTable
2016-03-01 11:23:03 -05:00
class AggregateBulkEditView(BulkEditView):
queryset = Aggregate.objects.prefetch_related('rir')
filterset = filters.AggregateFilterSet
table = tables.AggregateTable
2016-05-18 16:20:30 -04:00
form = forms.AggregateBulkEditForm
2016-03-01 11:23:03 -05:00
class AggregateBulkDeleteView(BulkDeleteView):
queryset = Aggregate.objects.prefetch_related('rir')
filterset = filters.AggregateFilterSet
table = tables.AggregateTable
2016-03-01 11:23:03 -05:00
2016-05-17 15:04:16 -04:00
#
# Prefix/VLAN roles
#
class RoleListView(ObjectListView):
2016-05-17 15:04:16 -04:00
queryset = Role.objects.all()
2016-05-18 16:20:30 -04:00
table = tables.RoleTable
2016-05-17 15:04:16 -04:00
class RoleEditView(ObjectEditView):
queryset = Role.objects.all()
model_form = forms.RoleForm
2016-05-17 15:04:16 -04:00
class RoleBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = Role.objects.all()
model_form = forms.RoleCSVForm
table = tables.RoleTable
class RoleBulkDeleteView(BulkDeleteView):
queryset = Role.objects.all()
table = tables.RoleTable
2016-05-17 15:04:16 -04:00
2016-03-01 11:23:03 -05:00
#
# Prefixes
#
class PrefixListView(ObjectListView):
queryset = Prefix.objects.prefetch_related('site', 'vrf__tenant', 'tenant', 'vlan', 'role')
filterset = filters.PrefixFilterSet
filterset_form = forms.PrefixFilterForm
table = tables.PrefixDetailTable
template_name = 'ipam/prefix_list.html'
2016-03-01 11:23:03 -05:00
2016-03-07 12:56:37 -05:00
def alter_queryset(self, request):
# Show only top-level prefixes by default (unless searching)
limit = None if request.GET.get('expand') or request.GET.get('q') else 0
2016-03-07 12:56:37 -05:00
return self.queryset.annotate_depth(limit=limit)
2016-03-01 11:23:03 -05:00
class PrefixView(ObjectView):
queryset = Prefix.objects.prefetch_related('vrf', 'site__region', 'tenant__group', 'vlan__group', 'role')
def get(self, request, pk):
prefix = get_object_or_404(self.queryset, pk=pk)
try:
2020-06-01 11:43:49 -04:00
aggregate = Aggregate.objects.restrict(request.user, 'view').get(
prefix__net_contains_or_equals=str(prefix.prefix)
)
except Aggregate.DoesNotExist:
aggregate = None
# Parent prefixes table
2020-06-01 11:43:49 -04:00
parent_prefixes = Prefix.objects.restrict(request.user, 'view').filter(
Q(vrf=prefix.vrf) | Q(vrf__isnull=True)
).filter(
prefix__net_contains=str(prefix.prefix)
).prefetch_related(
'site', 'role'
).annotate_depth()
parent_prefix_table = tables.PrefixTable(list(parent_prefixes), orderable=False)
parent_prefix_table.exclude = ('vrf',)
# Duplicate prefixes table
2020-06-01 11:43:49 -04:00
duplicate_prefixes = Prefix.objects.restrict(request.user, 'view').filter(
vrf=prefix.vrf, prefix=str(prefix.prefix)
).exclude(
pk=prefix.pk
).prefetch_related(
'site', 'role'
)
duplicate_prefix_table = tables.PrefixTable(list(duplicate_prefixes), orderable=False)
duplicate_prefix_table.exclude = ('vrf',)
return render(request, 'ipam/prefix.html', {
'prefix': prefix,
'aggregate': aggregate,
'parent_prefix_table': parent_prefix_table,
'duplicate_prefix_table': duplicate_prefix_table,
})
class PrefixPrefixesView(ObjectView):
queryset = Prefix.objects.all()
def get(self, request, pk):
prefix = get_object_or_404(self.queryset, pk=pk)
# Child prefixes table
2020-06-01 11:43:49 -04:00
child_prefixes = prefix.get_child_prefixes().restrict(request.user, 'view').prefetch_related(
'site', 'vlan', 'role',
).annotate_depth(limit=0)
# Add available prefixes to the table if requested
2020-01-07 17:58:30 +00:00
if child_prefixes and request.GET.get('show_available', 'true') == 'true':
child_prefixes = add_available_prefixes(prefix.prefix, child_prefixes)
prefix_table = tables.PrefixDetailTable(child_prefixes)
if request.user.has_perm('ipam.change_prefix') or request.user.has_perm('ipam.delete_prefix'):
prefix_table.columns.show('pk')
paginate = {
2018-11-02 14:27:17 -04:00
'paginator_class': EnhancedPaginator,
'per_page': request.GET.get('per_page', settings.PAGINATE_COUNT)
}
RequestConfig(request, paginate).configure(prefix_table)
# Compile permissions list for rendering the object table
permissions = {
'add': request.user.has_perm('ipam.add_prefix'),
'change': request.user.has_perm('ipam.change_prefix'),
'delete': request.user.has_perm('ipam.delete_prefix'),
}
return render(request, 'ipam/prefix_prefixes.html', {
'prefix': prefix,
'first_available_prefix': prefix.get_first_available_prefix(),
'prefix_table': prefix_table,
'permissions': permissions,
'bulk_querystring': 'vrf_id={}&within={}'.format(prefix.vrf.pk if prefix.vrf else '0', prefix.prefix),
'active_tab': 'prefixes',
2020-01-07 17:58:30 +00:00
'show_available': request.GET.get('show_available', 'true') == 'true',
})
class PrefixIPAddressesView(ObjectView):
queryset = Prefix.objects.all()
def get(self, request, pk):
prefix = get_object_or_404(self.queryset, pk=pk)
# Find all IPAddresses belonging to this Prefix
2020-06-01 11:43:49 -04:00
ipaddresses = prefix.get_child_ips().restrict(request.user, 'view').prefetch_related(
2020-06-22 14:46:25 -04:00
'vrf', 'primary_ip4_for', 'primary_ip6_for'
)
# Add available IP addresses to the table if requested
2020-01-07 17:58:30 +00:00
if request.GET.get('show_available', 'true') == 'true':
ipaddresses = add_available_ipaddresses(prefix.prefix, ipaddresses, prefix.is_pool)
ip_table = tables.IPAddressTable(ipaddresses)
if request.user.has_perm('ipam.change_ipaddress') or request.user.has_perm('ipam.delete_ipaddress'):
ip_table.columns.show('pk')
paginate = {
2018-11-02 14:27:17 -04:00
'paginator_class': EnhancedPaginator,
'per_page': request.GET.get('per_page', settings.PAGINATE_COUNT)
}
RequestConfig(request, paginate).configure(ip_table)
# Compile permissions list for rendering the object table
permissions = {
'add': request.user.has_perm('ipam.add_ipaddress'),
'change': request.user.has_perm('ipam.change_ipaddress'),
'delete': request.user.has_perm('ipam.delete_ipaddress'),
}
return render(request, 'ipam/prefix_ipaddresses.html', {
'prefix': prefix,
'first_available_ip': prefix.get_first_available_ip(),
'ip_table': ip_table,
'permissions': permissions,
'bulk_querystring': 'vrf_id={}&parent={}'.format(prefix.vrf.pk if prefix.vrf else '0', prefix.prefix),
'active_tab': 'ip-addresses',
2020-01-07 17:58:30 +00:00
'show_available': request.GET.get('show_available', 'true') == 'true',
})
2016-03-01 11:23:03 -05:00
class PrefixEditView(ObjectEditView):
queryset = Prefix.objects.all()
model_form = forms.PrefixForm
template_name = 'ipam/prefix_edit.html'
2016-03-01 11:23:03 -05:00
class PrefixDeleteView(ObjectDeleteView):
queryset = Prefix.objects.all()
template_name = 'ipam/prefix_delete.html'
2016-03-01 11:23:03 -05:00
class PrefixBulkImportView(BulkImportView):
queryset = Prefix.objects.all()
model_form = forms.PrefixCSVForm
2016-05-18 16:20:30 -04:00
table = tables.PrefixTable
2016-03-01 11:23:03 -05:00
class PrefixBulkEditView(BulkEditView):
queryset = Prefix.objects.prefetch_related('site', 'vrf__tenant', 'tenant', 'vlan', 'role')
filterset = filters.PrefixFilterSet
table = tables.PrefixTable
2016-05-18 16:20:30 -04:00
form = forms.PrefixBulkEditForm
2016-03-01 11:23:03 -05:00
class PrefixBulkDeleteView(BulkDeleteView):
queryset = Prefix.objects.prefetch_related('site', 'vrf__tenant', 'tenant', 'vlan', 'role')
filterset = filters.PrefixFilterSet
table = tables.PrefixTable
2016-03-01 11:23:03 -05:00
#
# IP addresses
#
class IPAddressListView(ObjectListView):
queryset = IPAddress.objects.prefetch_related(
2020-06-22 14:46:25 -04:00
'vrf__tenant', 'tenant', 'nat_inside'
)
filterset = filters.IPAddressFilterSet
filterset_form = forms.IPAddressFilterForm
table = tables.IPAddressDetailTable
2016-03-01 11:23:03 -05:00
class IPAddressView(ObjectView):
queryset = IPAddress.objects.prefetch_related('vrf__tenant', 'tenant')
def get(self, request, pk):
ipaddress = get_object_or_404(self.queryset, pk=pk)
2020-06-23 16:39:43 -04:00
# Parent prefixes table
parent_prefixes = Prefix.objects.restrict(request.user, 'view').filter(
vrf=ipaddress.vrf, prefix__net_contains=str(ipaddress.address.ip)
).prefetch_related(
'site', 'role'
)
parent_prefixes_table = tables.PrefixTable(list(parent_prefixes), orderable=False)
parent_prefixes_table.exclude = ('vrf',)
# Duplicate IPs table
duplicate_ips = IPAddress.objects.restrict(request.user, 'view').filter(
vrf=ipaddress.vrf, address=str(ipaddress.address)
).exclude(
pk=ipaddress.pk
).prefetch_related(
'nat_inside'
)
# Exclude anycast IPs if this IP is anycast
if ipaddress.role == IPAddressRoleChoices.ROLE_ANYCAST:
duplicate_ips = duplicate_ips.exclude(role=IPAddressRoleChoices.ROLE_ANYCAST)
duplicate_ips_table = tables.IPAddressTable(list(duplicate_ips), orderable=False)
# Related IP table
related_ips = IPAddress.objects.restrict(request.user, 'view').exclude(
address=str(ipaddress.address)
).filter(
vrf=ipaddress.vrf, address__net_contained_or_equal=str(ipaddress.address)
)
related_ips_table = tables.IPAddressTable(related_ips, orderable=False)
paginate = {
'paginator_class': EnhancedPaginator,
'per_page': request.GET.get('per_page', settings.PAGINATE_COUNT)
}
RequestConfig(request, paginate).configure(related_ips_table)
return render(request, 'ipam/ipaddress.html', {
'ipaddress': ipaddress,
2020-06-23 16:39:43 -04:00
'parent_prefixes_table': parent_prefixes_table,
'duplicate_ips_table': duplicate_ips_table,
'related_ips_table': related_ips_table,
})
2016-03-01 11:23:03 -05:00
class IPAddressEditView(ObjectEditView):
queryset = IPAddress.objects.all()
model_form = forms.IPAddressForm
template_name = 'ipam/ipaddress_edit.html'
2016-03-01 11:23:03 -05:00
def alter_obj(self, obj, request, url_args, url_kwargs):
2020-06-24 09:52:22 -04:00
if 'interface' in request.GET:
try:
2020-06-24 09:52:22 -04:00
obj.assigned_object = Interface.objects.get(pk=request.GET['interface'])
except (ValueError, Interface.DoesNotExist):
pass
2020-06-24 09:52:22 -04:00
elif 'vminterface' in request.GET:
try:
obj.assigned_object = VMInterface.objects.get(pk=request.GET['vminterface'])
except (ValueError, VMInterface.DoesNotExist):
pass
return obj
2016-03-01 11:23:03 -05:00
class IPAddressAssignView(ObjectView):
"""
Search for IPAddresses to be assigned to an Interface.
"""
queryset = IPAddress.objects.all()
def dispatch(self, request, *args, **kwargs):
# Redirect user if an interface has not been provided
if 'interface' not in request.GET:
return redirect('ipam:ipaddress_add')
return super().dispatch(request, *args, **kwargs)
def get(self, request):
form = forms.IPAddressAssignForm()
return render(request, 'ipam/ipaddress_assign.html', {
'form': form,
'return_url': request.GET.get('return_url', ''),
})
def post(self, request):
form = forms.IPAddressAssignForm(request.POST)
table = None
if form.is_valid():
2020-06-22 14:46:25 -04:00
addresses = self.queryset.prefetch_related('vrf', 'tenant')
2020-01-09 16:26:11 +00:00
# Limit to 100 results
addresses = filters.IPAddressFilterSet(request.POST, addresses).qs[:100]
2020-01-09 16:26:11 +00:00
table = tables.IPAddressAssignTable(addresses)
return render(request, 'ipam/ipaddress_assign.html', {
'form': form,
'table': table,
'return_url': request.GET.get('return_url', ''),
})
class IPAddressDeleteView(ObjectDeleteView):
queryset = IPAddress.objects.all()
2016-03-01 11:23:03 -05:00
class IPAddressBulkCreateView(BulkCreateView):
queryset = IPAddress.objects.all()
form = forms.IPAddressBulkCreateForm
model_form = forms.IPAddressBulkAddForm
pattern_target = 'address'
template_name = 'ipam/ipaddress_bulk_add.html'
class IPAddressBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = IPAddress.objects.all()
model_form = forms.IPAddressCSVForm
2016-05-18 16:20:30 -04:00
table = tables.IPAddressTable
2016-03-01 11:23:03 -05:00
class IPAddressBulkEditView(BulkEditView):
2020-06-22 14:46:25 -04:00
queryset = IPAddress.objects.prefetch_related('vrf__tenant', 'tenant')
filterset = filters.IPAddressFilterSet
table = tables.IPAddressTable
2016-05-18 16:20:30 -04:00
form = forms.IPAddressBulkEditForm
2016-03-01 11:23:03 -05:00
class IPAddressBulkDeleteView(BulkDeleteView):
2020-06-22 14:46:25 -04:00
queryset = IPAddress.objects.prefetch_related('vrf__tenant', 'tenant')
filterset = filters.IPAddressFilterSet
table = tables.IPAddressTable
2016-03-01 11:23:03 -05:00
2016-07-15 13:26:54 -04:00
#
# VLAN groups
#
class VLANGroupListView(ObjectListView):
queryset = VLANGroup.objects.prefetch_related('site').annotate(vlan_count=Count('vlans'))
filterset = filters.VLANGroupFilterSet
filterset_form = forms.VLANGroupFilterForm
2016-07-15 13:26:54 -04:00
table = tables.VLANGroupTable
class VLANGroupEditView(ObjectEditView):
queryset = VLANGroup.objects.all()
model_form = forms.VLANGroupForm
2016-07-15 13:26:54 -04:00
class VLANGroupBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = VLANGroup.objects.all()
model_form = forms.VLANGroupCSVForm
table = tables.VLANGroupTable
class VLANGroupBulkDeleteView(BulkDeleteView):
queryset = VLANGroup.objects.prefetch_related('site').annotate(vlan_count=Count('vlans'))
filterset = filters.VLANGroupFilterSet
table = tables.VLANGroupTable
2016-07-15 13:26:54 -04:00
class VLANGroupVLANsView(ObjectView):
queryset = VLANGroup.objects.all()
2019-04-11 17:27:38 -04:00
def get(self, request, pk):
vlan_group = get_object_or_404(self.queryset, pk=pk)
vlans = VLAN.objects.restrict(request.user, 'view').filter(group_id=pk).prefetch_related(
Prefetch('prefixes', queryset=Prefix.objects.restrict(request.user))
)
vlans = add_available_vlans(vlan_group, vlans)
vlan_table = tables.VLANDetailTable(vlans)
if request.user.has_perm('ipam.change_vlan') or request.user.has_perm('ipam.delete_vlan'):
vlan_table.columns.show('pk')
vlan_table.columns.hide('site')
vlan_table.columns.hide('group')
paginate = {
2018-11-02 14:27:17 -04:00
'paginator_class': EnhancedPaginator,
'per_page': request.GET.get('per_page', settings.PAGINATE_COUNT)
}
RequestConfig(request, paginate).configure(vlan_table)
# Compile permissions list for rendering the object table
permissions = {
'add': request.user.has_perm('ipam.add_vlan'),
'change': request.user.has_perm('ipam.change_vlan'),
'delete': request.user.has_perm('ipam.delete_vlan'),
}
return render(request, 'ipam/vlangroup_vlans.html', {
'vlan_group': vlan_group,
'first_available_vlan': vlan_group.get_next_available_vid(),
'vlan_table': vlan_table,
'permissions': permissions,
})
2016-03-01 11:23:03 -05:00
#
# VLANs
#
class VLANListView(ObjectListView):
queryset = VLAN.objects.prefetch_related('site', 'group', 'tenant', 'role').prefetch_related('prefixes')
filterset = filters.VLANFilterSet
filterset_form = forms.VLANFilterForm
table = tables.VLANDetailTable
2016-03-01 11:23:03 -05:00
class VLANView(ObjectView):
queryset = VLAN.objects.prefetch_related('site__region', 'tenant__group', 'role')
2016-03-01 11:23:03 -05:00
def get(self, request, pk):
2016-03-01 11:23:03 -05:00
vlan = get_object_or_404(self.queryset, pk=pk)
2020-06-01 11:43:49 -04:00
prefixes = Prefix.objects.restrict(request.user, 'view').filter(vlan=vlan).prefetch_related(
'vrf', 'site', 'role'
)
prefix_table = tables.PrefixTable(list(prefixes), orderable=False)
prefix_table.exclude = ('vlan',)
return render(request, 'ipam/vlan.html', {
'vlan': vlan,
'prefix_table': prefix_table,
})
2016-03-01 11:23:03 -05:00
class VLANMembersView(ObjectView):
queryset = VLAN.objects.all()
def get(self, request, pk):
vlan = get_object_or_404(self.queryset, pk=pk)
2020-06-01 11:43:49 -04:00
members = vlan.get_members().restrict(request.user, 'view').prefetch_related('device', 'virtual_machine')
members_table = tables.VLANMemberTable(members)
paginate = {
2018-11-02 14:27:17 -04:00
'paginator_class': EnhancedPaginator,
'per_page': request.GET.get('per_page', settings.PAGINATE_COUNT)
}
RequestConfig(request, paginate).configure(members_table)
return render(request, 'ipam/vlan_members.html', {
'vlan': vlan,
'members_table': members_table,
'active_tab': 'members',
})
class VLANEditView(ObjectEditView):
queryset = VLAN.objects.all()
model_form = forms.VLANForm
template_name = 'ipam/vlan_edit.html'
2016-03-01 11:23:03 -05:00
class VLANDeleteView(ObjectDeleteView):
queryset = VLAN.objects.all()
2016-03-01 11:23:03 -05:00
class VLANBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = VLAN.objects.all()
model_form = forms.VLANCSVForm
2016-05-18 16:20:30 -04:00
table = tables.VLANTable
2016-03-01 11:23:03 -05:00
class VLANBulkEditView(BulkEditView):
queryset = VLAN.objects.prefetch_related('site', 'group', 'tenant', 'role')
filterset = filters.VLANFilterSet
table = tables.VLANTable
2016-05-18 16:20:30 -04:00
form = forms.VLANBulkEditForm
2016-03-01 11:23:03 -05:00
class VLANBulkDeleteView(BulkDeleteView):
queryset = VLAN.objects.prefetch_related('site', 'group', 'tenant', 'role')
filterset = filters.VLANFilterSet
table = tables.VLANTable
#
# Services
#
class ServiceListView(ObjectListView):
queryset = Service.objects.prefetch_related('device', 'virtual_machine')
filterset = filters.ServiceFilterSet
filterset_form = forms.ServiceFilterForm
table = tables.ServiceTable
action_buttons = ('export',)
class ServiceView(ObjectView):
queryset = Service.objects.all()
def get(self, request, pk):
service = get_object_or_404(self.queryset, pk=pk)
return render(request, 'ipam/service.html', {
'service': service,
})
class ServiceEditView(ObjectEditView):
queryset = Service.objects.all()
model_form = forms.ServiceForm
template_name = 'ipam/service_edit.html'
def alter_obj(self, obj, request, url_args, url_kwargs):
if 'device' in url_kwargs:
obj.device = get_object_or_404(Device, pk=url_kwargs['device'])
2017-08-31 12:50:35 -04:00
elif 'virtualmachine' in url_kwargs:
obj.virtual_machine = get_object_or_404(VirtualMachine, pk=url_kwargs['virtualmachine'])
return obj
def get_return_url(self, request, service):
return service.parent.get_absolute_url()
class ServiceBulkImportView(BulkImportView):
2020-05-21 11:58:27 -04:00
queryset = Service.objects.all()
model_form = forms.ServiceCSVForm
table = tables.ServiceTable
class ServiceDeleteView(ObjectDeleteView):
queryset = Service.objects.all()
class ServiceBulkEditView(BulkEditView):
queryset = Service.objects.prefetch_related('device', 'virtual_machine')
filterset = filters.ServiceFilterSet
table = tables.ServiceTable
form = forms.ServiceBulkEditForm
class ServiceBulkDeleteView(BulkDeleteView):
queryset = Service.objects.prefetch_related('device', 'virtual_machine')
filterset = filters.ServiceFilterSet
table = tables.ServiceTable