|
@@ -9,7 +9,7 @@ from utilities.forms import (
|
|
|
)
|
|
|
|
|
|
from .models import (
|
|
|
- Aggregate, IPAddress, Prefix, PREFIX_STATUS_CHOICES, RIR, Role, VLAN, VLAN_STATUS_CHOICES, VRF,
|
|
|
+ Aggregate, IPAddress, Prefix, PREFIX_STATUS_CHOICES, RIR, Role, VLAN, VLANGroup, VLAN_STATUS_CHOICES, VRF,
|
|
|
)
|
|
|
|
|
|
|
|
@@ -408,21 +408,66 @@ class IPAddressFilterForm(forms.Form, BootstrapMixin):
|
|
|
|
|
|
|
|
|
#
|
|
|
+# VLAN groups
|
|
|
+#
|
|
|
+
|
|
|
+class VLANGroupForm(forms.ModelForm, BootstrapMixin):
|
|
|
+ slug = SlugField()
|
|
|
+
|
|
|
+ class Meta:
|
|
|
+ model = VLANGroup
|
|
|
+ fields = ['site', 'name', 'slug']
|
|
|
+
|
|
|
+
|
|
|
+class VLANGroupBulkDeleteForm(ConfirmationForm):
|
|
|
+ pk = forms.ModelMultipleChoiceField(queryset=VLANGroup.objects.all(), widget=forms.MultipleHiddenInput)
|
|
|
+
|
|
|
+
|
|
|
+def vlangroup_site_choices():
|
|
|
+ site_choices = Site.objects.annotate(vlangroup_count=Count('vlan_groups'))
|
|
|
+ return [(s.slug, '{} ({})'.format(s.name, s.vlangroup_count)) for s in site_choices]
|
|
|
+
|
|
|
+
|
|
|
+class VLANGroupFilterForm(forms.Form, BootstrapMixin):
|
|
|
+ site = forms.MultipleChoiceField(required=False, choices=vlangroup_site_choices,
|
|
|
+ widget=forms.SelectMultiple(attrs={'size': 8}))
|
|
|
+
|
|
|
+
|
|
|
+#
|
|
|
# VLANs
|
|
|
#
|
|
|
|
|
|
class VLANForm(forms.ModelForm, BootstrapMixin):
|
|
|
+ group = forms.ModelChoiceField(queryset=VLANGroup.objects.all(), required=False, label='Group', widget=APISelect(
|
|
|
+ api_url='/api/ipam/vlan-groups/?site_id={{site}}',
|
|
|
+ ))
|
|
|
|
|
|
class Meta:
|
|
|
model = VLAN
|
|
|
- fields = ['site', 'vid', 'name', 'status', 'role']
|
|
|
+ fields = ['site', 'group', 'vid', 'name', 'status', 'role']
|
|
|
help_texts = {
|
|
|
'site': "The site at which this VLAN exists",
|
|
|
+ 'group': "VLAN group (optional)",
|
|
|
'vid': "Configured VLAN ID",
|
|
|
'name': "Configured VLAN name",
|
|
|
'status': "Operational status of this VLAN",
|
|
|
'role': "The primary function of this VLAN",
|
|
|
}
|
|
|
+ widgets = {
|
|
|
+ 'site': forms.Select(attrs={'filter-for': 'group'}),
|
|
|
+ }
|
|
|
+
|
|
|
+ def __init__(self, *args, **kwargs):
|
|
|
+
|
|
|
+ super(VLANForm, self).__init__(*args, **kwargs)
|
|
|
+
|
|
|
+ # Limit VLAN group choices
|
|
|
+ if self.is_bound and self.data.get('site'):
|
|
|
+ self.fields['group'].queryset = VLANGroup.objects.filter(site__pk=self.data['site'])
|
|
|
+ elif self.initial.get('site'):
|
|
|
+ self.fields['group'].queryset = VLANGroup.objects.filter(site=self.initial['site'])
|
|
|
+ else:
|
|
|
+ self.fields['group'].choices = []
|
|
|
|
|
|
|
|
|
class VLANFromCSVForm(forms.ModelForm):
|
|
@@ -465,6 +510,11 @@ def vlan_site_choices():
|
|
|
return [(s.slug, '{} ({})'.format(s.name, s.vlan_count)) for s in site_choices]
|
|
|
|
|
|
|
|
|
+def vlan_group_choices():
|
|
|
+ group_choices = VLANGroup.objects.select_related('site').annotate(vlan_count=Count('vlans'))
|
|
|
+ return [(g.pk, '{} ({})'.format(g, g.vlan_count)) for g in group_choices]
|
|
|
+
|
|
|
+
|
|
|
def vlan_status_choices():
|
|
|
status_counts = {}
|
|
|
for status in VLAN.objects.values('status').annotate(count=Count('status')).order_by('status'):
|
|
@@ -480,6 +530,8 @@ def vlan_role_choices():
|
|
|
class VLANFilterForm(forms.Form, BootstrapMixin):
|
|
|
site = forms.MultipleChoiceField(required=False, choices=vlan_site_choices,
|
|
|
widget=forms.SelectMultiple(attrs={'size': 8}))
|
|
|
+ group_id = forms.MultipleChoiceField(required=False, choices=vlan_group_choices, label='VLAN Group',
|
|
|
+ widget=forms.SelectMultiple(attrs={'size': 8}))
|
|
|
status = forms.MultipleChoiceField(required=False, choices=vlan_status_choices)
|
|
|
role = forms.MultipleChoiceField(required=False, choices=vlan_role_choices,
|
|
|
widget=forms.SelectMultiple(attrs={'size': 8}))
|