forms.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587
  1. from django import forms
  2. from django.db.models import Count
  3. from dcim.models import Site, Device, Interface
  4. from extras.forms import CustomFieldForm, CustomFieldBulkEditForm
  5. from tenancy.forms import bulkedit_tenant_choices
  6. from tenancy.models import Tenant
  7. from utilities.forms import BootstrapMixin, APISelect, Livesearch, CSVDataField, BulkImportForm, SlugField
  8. from .models import (
  9. Aggregate, IPAddress, Prefix, PREFIX_STATUS_CHOICES, RIR, Role, VLAN, VLANGroup, VLAN_STATUS_CHOICES, VRF,
  10. )
  11. FORM_PREFIX_STATUS_CHOICES = (('', '---------'),) + PREFIX_STATUS_CHOICES
  12. FORM_VLAN_STATUS_CHOICES = (('', '---------'),) + VLAN_STATUS_CHOICES
  13. def bulkedit_vrf_choices():
  14. """
  15. Include an option to assign the object to the global table.
  16. """
  17. choices = [
  18. (None, '---------'),
  19. (0, 'Global'),
  20. ]
  21. choices += [(v.pk, v.name) for v in VRF.objects.all()]
  22. return choices
  23. #
  24. # VRFs
  25. #
  26. class VRFForm(BootstrapMixin, CustomFieldForm):
  27. class Meta:
  28. model = VRF
  29. fields = ['name', 'rd', 'tenant', 'enforce_unique', 'description']
  30. labels = {
  31. 'rd': "RD",
  32. }
  33. help_texts = {
  34. 'rd': "Route distinguisher in any format",
  35. }
  36. class VRFFromCSVForm(forms.ModelForm):
  37. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  38. error_messages={'invalid_choice': 'Tenant not found.'})
  39. class Meta:
  40. model = VRF
  41. fields = ['name', 'rd', 'tenant', 'enforce_unique', 'description']
  42. class VRFImportForm(BulkImportForm, BootstrapMixin):
  43. csv = CSVDataField(csv_form=VRFFromCSVForm)
  44. class VRFBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  45. pk = forms.ModelMultipleChoiceField(queryset=VRF.objects.all(), widget=forms.MultipleHiddenInput)
  46. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  47. description = forms.CharField(max_length=100, required=False)
  48. def vrf_tenant_choices():
  49. tenant_choices = Tenant.objects.annotate(vrf_count=Count('vrfs'))
  50. return [(t.slug, u'{} ({})'.format(t.name, t.vrf_count)) for t in tenant_choices]
  51. class VRFFilterForm(forms.Form, BootstrapMixin):
  52. tenant = forms.MultipleChoiceField(required=False, choices=vrf_tenant_choices,
  53. widget=forms.SelectMultiple(attrs={'size': 8}))
  54. #
  55. # RIRs
  56. #
  57. class RIRForm(forms.ModelForm, BootstrapMixin):
  58. slug = SlugField()
  59. class Meta:
  60. model = RIR
  61. fields = ['name', 'slug']
  62. #
  63. # Aggregates
  64. #
  65. class AggregateForm(BootstrapMixin, CustomFieldForm):
  66. class Meta:
  67. model = Aggregate
  68. fields = ['prefix', 'rir', 'date_added', 'description']
  69. help_texts = {
  70. 'prefix': "IPv4 or IPv6 network",
  71. 'rir': "Regional Internet Registry responsible for this prefix",
  72. 'date_added': "Format: YYYY-MM-DD",
  73. }
  74. class AggregateFromCSVForm(forms.ModelForm):
  75. rir = forms.ModelChoiceField(queryset=RIR.objects.all(), to_field_name='name',
  76. error_messages={'invalid_choice': 'RIR not found.'})
  77. class Meta:
  78. model = Aggregate
  79. fields = ['prefix', 'rir', 'date_added', 'description']
  80. class AggregateImportForm(BulkImportForm, BootstrapMixin):
  81. csv = CSVDataField(csv_form=AggregateFromCSVForm)
  82. class AggregateBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  83. pk = forms.ModelMultipleChoiceField(queryset=Aggregate.objects.all(), widget=forms.MultipleHiddenInput)
  84. rir = forms.ModelChoiceField(queryset=RIR.objects.all(), required=False, label='RIR')
  85. date_added = forms.DateField(required=False)
  86. description = forms.CharField(max_length=100, required=False)
  87. def aggregate_rir_choices():
  88. rir_choices = RIR.objects.annotate(aggregate_count=Count('aggregates'))
  89. return [(r.slug, u'{} ({})'.format(r.name, r.aggregate_count)) for r in rir_choices]
  90. class AggregateFilterForm(forms.Form, BootstrapMixin):
  91. rir = forms.MultipleChoiceField(required=False, choices=aggregate_rir_choices, label='RIR',
  92. widget=forms.SelectMultiple(attrs={'size': 8}))
  93. #
  94. # Roles
  95. #
  96. class RoleForm(forms.ModelForm, BootstrapMixin):
  97. slug = SlugField()
  98. class Meta:
  99. model = Role
  100. fields = ['name', 'slug']
  101. #
  102. # Prefixes
  103. #
  104. class PrefixForm(BootstrapMixin, CustomFieldForm):
  105. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False, label='Site',
  106. widget=forms.Select(attrs={'filter-for': 'vlan'}))
  107. vlan = forms.ModelChoiceField(queryset=VLAN.objects.all(), required=False, label='VLAN',
  108. widget=APISelect(api_url='/api/ipam/vlans/?site_id={{site}}',
  109. display_field='display_name'))
  110. class Meta:
  111. model = Prefix
  112. fields = ['prefix', 'vrf', 'tenant', 'site', 'vlan', 'status', 'role', 'description']
  113. help_texts = {
  114. 'prefix': "IPv4 or IPv6 network",
  115. 'vrf': "VRF (if applicable)",
  116. 'site': "The site to which this prefix is assigned (if applicable)",
  117. 'vlan': "The VLAN to which this prefix is assigned (if applicable)",
  118. 'status': "Operational status of this prefix",
  119. 'role': "The primary function of this prefix",
  120. }
  121. def __init__(self, *args, **kwargs):
  122. super(PrefixForm, self).__init__(*args, **kwargs)
  123. self.fields['vrf'].empty_label = 'Global'
  124. # Initialize field without choices to avoid pulling all VLANs from the database
  125. if self.is_bound and self.data.get('site'):
  126. self.fields['vlan'].queryset = VLAN.objects.filter(site__pk=self.data['site'])
  127. elif self.initial.get('site'):
  128. self.fields['vlan'].queryset = VLAN.objects.filter(site=self.initial['site'])
  129. else:
  130. self.fields['vlan'].choices = []
  131. def clean_prefix(self):
  132. prefix = self.cleaned_data['prefix']
  133. if prefix.version == 4 and prefix.prefixlen == 32:
  134. raise forms.ValidationError("Cannot create host addresses (/32) as prefixes. These should be IPv4 "
  135. "addresses instead.")
  136. elif prefix.version == 6 and prefix.prefixlen == 128:
  137. raise forms.ValidationError("Cannot create host addresses (/128) as prefixes. These should be IPv6 "
  138. "addresses instead.")
  139. return prefix
  140. class PrefixFromCSVForm(forms.ModelForm):
  141. vrf = forms.ModelChoiceField(queryset=VRF.objects.all(), required=False, to_field_name='rd',
  142. error_messages={'invalid_choice': 'VRF not found.'})
  143. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  144. error_messages={'invalid_choice': 'Tenant not found.'})
  145. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False, to_field_name='name',
  146. error_messages={'invalid_choice': 'Site not found.'})
  147. vlan_group_name = forms.CharField(required=False)
  148. vlan_vid = forms.IntegerField(required=False)
  149. status_name = forms.ChoiceField(choices=[(s[1], s[0]) for s in PREFIX_STATUS_CHOICES])
  150. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False, to_field_name='name',
  151. error_messages={'invalid_choice': 'Invalid role.'})
  152. class Meta:
  153. model = Prefix
  154. fields = ['prefix', 'vrf', 'tenant', 'site', 'vlan_group_name', 'vlan_vid', 'status_name', 'role',
  155. 'description']
  156. def clean(self):
  157. super(PrefixFromCSVForm, self).clean()
  158. site = self.cleaned_data.get('site')
  159. vlan_group_name = self.cleaned_data.get('vlan_group_name')
  160. vlan_vid = self.cleaned_data.get('vlan_vid')
  161. # Validate VLAN
  162. vlan_group = None
  163. if vlan_group_name:
  164. try:
  165. vlan_group = VLANGroup.objects.get(site=site, name=vlan_group_name)
  166. except VLANGroup.DoesNotExist:
  167. self.add_error('vlan_group_name', "Invalid VLAN group ({} - {}).".format(site, vlan_group_name))
  168. if vlan_vid and vlan_group:
  169. try:
  170. self.instance.vlan = VLAN.objects.get(group=vlan_group, vid=vlan_vid)
  171. except VLAN.DoesNotExist:
  172. self.add_error('vlan_vid', "Invalid VLAN ID ({} - {}).".format(vlan_group, vlan_vid))
  173. elif vlan_vid and site:
  174. try:
  175. self.instance.vlan = VLAN.objects.get(site=site, vid=vlan_vid)
  176. except VLAN.MultipleObjectsReturned:
  177. self.add_error('vlan_vid', "Multiple VLANs found ({} - VID {})".format(site, vlan_vid))
  178. elif vlan_vid:
  179. self.add_error('vlan_vid', "Must specify site and/or VLAN group when assigning a VLAN.")
  180. def save(self, *args, **kwargs):
  181. m = super(PrefixFromCSVForm, self).save(commit=False)
  182. # Assign Prefix status by name
  183. m.status = dict(self.fields['status_name'].choices)[self.cleaned_data['status_name']]
  184. if kwargs.get('commit'):
  185. m.save()
  186. return m
  187. class PrefixImportForm(BulkImportForm, BootstrapMixin):
  188. csv = CSVDataField(csv_form=PrefixFromCSVForm)
  189. class PrefixBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  190. pk = forms.ModelMultipleChoiceField(queryset=Prefix.objects.all(), widget=forms.MultipleHiddenInput)
  191. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False)
  192. vrf = forms.TypedChoiceField(choices=bulkedit_vrf_choices, coerce=int, required=False, label='VRF')
  193. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  194. status = forms.ChoiceField(choices=FORM_PREFIX_STATUS_CHOICES, required=False)
  195. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False)
  196. description = forms.CharField(max_length=100, required=False)
  197. def prefix_vrf_choices():
  198. vrf_choices = VRF.objects.annotate(prefix_count=Count('prefixes'))
  199. return [(v.pk, u'{} ({})'.format(v.name, v.prefix_count)) for v in vrf_choices]
  200. def tenant_choices():
  201. tenant_choices = Tenant.objects.all()
  202. return [(t.slug, t.name) for t in tenant_choices]
  203. def prefix_site_choices():
  204. site_choices = Site.objects.annotate(prefix_count=Count('prefixes'))
  205. return [(s.slug, u'{} ({})'.format(s.name, s.prefix_count)) for s in site_choices]
  206. def prefix_status_choices():
  207. status_counts = {}
  208. for status in Prefix.objects.values('status').annotate(count=Count('status')).order_by('status'):
  209. status_counts[status['status']] = status['count']
  210. return [(s[0], u'{} ({})'.format(s[1], status_counts.get(s[0], 0))) for s in PREFIX_STATUS_CHOICES]
  211. def prefix_role_choices():
  212. role_choices = Role.objects.annotate(prefix_count=Count('prefixes'))
  213. return [(r.slug, u'{} ({})'.format(r.name, r.prefix_count)) for r in role_choices]
  214. class PrefixFilterForm(forms.Form, BootstrapMixin):
  215. parent = forms.CharField(required=False, label='Search Within', widget=forms.TextInput(attrs={
  216. 'placeholder': 'Network',
  217. }))
  218. vrf = forms.MultipleChoiceField(required=False, choices=prefix_vrf_choices, label='VRF',
  219. widget=forms.SelectMultiple(attrs={'size': 6}))
  220. tenant = forms.MultipleChoiceField(required=False, choices=tenant_choices, label='Tenant',
  221. widget=forms.SelectMultiple(attrs={'size': 6}))
  222. status = forms.MultipleChoiceField(required=False, choices=prefix_status_choices,
  223. widget=forms.SelectMultiple(attrs={'size': 6}))
  224. site = forms.MultipleChoiceField(required=False, choices=prefix_site_choices,
  225. widget=forms.SelectMultiple(attrs={'size': 6}))
  226. role = forms.MultipleChoiceField(required=False, choices=prefix_role_choices,
  227. widget=forms.SelectMultiple(attrs={'size': 6}))
  228. expand = forms.BooleanField(required=False, label='Expand prefix hierarchy')
  229. #
  230. # IP addresses
  231. #
  232. class IPAddressForm(BootstrapMixin, CustomFieldForm):
  233. nat_site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False, label='Site',
  234. widget=forms.Select(attrs={'filter-for': 'nat_device'}))
  235. nat_device = forms.ModelChoiceField(queryset=Device.objects.all(), required=False, label='Device',
  236. widget=APISelect(api_url='/api/dcim/devices/?site_id={{nat_site}}',
  237. attrs={'filter-for': 'nat_inside'}))
  238. livesearch = forms.CharField(required=False, label='IP Address', widget=Livesearch(
  239. query_key='q', query_url='ipam-api:ipaddress_list', field_to_update='nat_inside', obj_label='address')
  240. )
  241. nat_inside = forms.ModelChoiceField(queryset=IPAddress.objects.all(), required=False, label='NAT (Inside)',
  242. widget=APISelect(api_url='/api/ipam/ip-addresses/?device_id={{nat_device}}',
  243. display_field='address'))
  244. class Meta:
  245. model = IPAddress
  246. fields = ['address', 'vrf', 'tenant', 'nat_device', 'nat_inside', 'description']
  247. help_texts = {
  248. 'address': "IPv4 or IPv6 address and mask",
  249. 'vrf': "VRF (if applicable)",
  250. }
  251. def __init__(self, *args, **kwargs):
  252. super(IPAddressForm, self).__init__(*args, **kwargs)
  253. self.fields['vrf'].empty_label = 'Global'
  254. if self.instance.nat_inside:
  255. nat_inside = self.instance.nat_inside
  256. # If the IP is assigned to an interface, populate site/device fields accordingly
  257. if self.instance.nat_inside.interface:
  258. self.initial['nat_site'] = self.instance.nat_inside.interface.device.rack.site.pk
  259. self.initial['nat_device'] = self.instance.nat_inside.interface.device.pk
  260. self.fields['nat_device'].queryset = Device.objects.filter(
  261. rack__site=nat_inside.interface.device.rack.site)
  262. self.fields['nat_inside'].queryset = IPAddress.objects.filter(
  263. interface__device=nat_inside.interface.device)
  264. else:
  265. self.fields['nat_inside'].queryset = IPAddress.objects.filter(pk=nat_inside.pk)
  266. else:
  267. # Initialize nat_device choices if nat_site is set
  268. if self.is_bound and self.data.get('nat_site'):
  269. self.fields['nat_device'].queryset = Device.objects.filter(rack__site__pk=self.data['nat_site'])
  270. elif self.initial.get('nat_site'):
  271. self.fields['nat_device'].queryset = Device.objects.filter(rack__site=self.initial['nat_site'])
  272. else:
  273. self.fields['nat_device'].choices = []
  274. # Initialize nat_inside choices if nat_device is set
  275. if self.is_bound and self.data.get('nat_device'):
  276. self.fields['nat_inside'].queryset = IPAddress.objects.filter(
  277. interface__device__pk=self.data['nat_device'])
  278. elif self.initial.get('nat_device'):
  279. self.fields['nat_inside'].queryset = IPAddress.objects.filter(
  280. interface__device__pk=self.initial['nat_device'])
  281. else:
  282. self.fields['nat_inside'].choices = []
  283. class IPAddressFromCSVForm(forms.ModelForm):
  284. vrf = forms.ModelChoiceField(queryset=VRF.objects.all(), required=False, to_field_name='rd',
  285. error_messages={'invalid_choice': 'VRF not found.'})
  286. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  287. error_messages={'invalid_choice': 'Tenant not found.'})
  288. device = forms.ModelChoiceField(queryset=Device.objects.all(), required=False, to_field_name='name',
  289. error_messages={'invalid_choice': 'Device not found.'})
  290. interface_name = forms.CharField(required=False)
  291. is_primary = forms.BooleanField(required=False)
  292. class Meta:
  293. model = IPAddress
  294. fields = ['address', 'vrf', 'tenant', 'device', 'interface_name', 'is_primary', 'description']
  295. def clean(self):
  296. device = self.cleaned_data.get('device')
  297. interface_name = self.cleaned_data.get('interface_name')
  298. is_primary = self.cleaned_data.get('is_primary')
  299. # Validate interface
  300. if device and interface_name:
  301. try:
  302. Interface.objects.get(device=device, name=interface_name)
  303. except Interface.DoesNotExist:
  304. self.add_error('interface_name', "Invalid interface ({}) for {}".format(interface_name, device))
  305. elif device and not interface_name:
  306. self.add_error('interface_name', "Device set ({}) but interface missing".format(device))
  307. elif interface_name and not device:
  308. self.add_error('device', "Interface set ({}) but device missing or invalid".format(interface_name))
  309. # Validate is_primary
  310. if is_primary and not device:
  311. self.add_error('is_primary', "No device specified; cannot set as primary IP")
  312. def save(self, commit=True):
  313. # Set interface
  314. if self.cleaned_data['device'] and self.cleaned_data['interface_name']:
  315. self.instance.interface = Interface.objects.get(device=self.cleaned_data['device'],
  316. name=self.cleaned_data['interface_name'])
  317. # Set as primary for device
  318. if self.cleaned_data['is_primary']:
  319. if self.instance.address.version == 4:
  320. self.instance.primary_ip4_for = self.cleaned_data['device']
  321. elif self.instance.address.version == 6:
  322. self.instance.primary_ip6_for = self.cleaned_data['device']
  323. return super(IPAddressFromCSVForm, self).save(commit=commit)
  324. class IPAddressImportForm(BulkImportForm, BootstrapMixin):
  325. csv = CSVDataField(csv_form=IPAddressFromCSVForm)
  326. class IPAddressBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  327. pk = forms.ModelMultipleChoiceField(queryset=IPAddress.objects.all(), widget=forms.MultipleHiddenInput)
  328. vrf = forms.TypedChoiceField(choices=bulkedit_vrf_choices, coerce=int, required=False, label='VRF')
  329. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  330. description = forms.CharField(max_length=100, required=False)
  331. def ipaddress_family_choices():
  332. return [('', 'All'), (4, 'IPv4'), (6, 'IPv6')]
  333. def ipaddress_vrf_choices():
  334. vrf_choices = VRF.objects.annotate(ipaddress_count=Count('ip_addresses'))
  335. return [(v.pk, u'{} ({})'.format(v.name, v.ipaddress_count)) for v in vrf_choices]
  336. class IPAddressFilterForm(forms.Form, BootstrapMixin):
  337. parent = forms.CharField(required=False, label='Search Within', widget=forms.TextInput(attrs={
  338. 'placeholder': 'Prefix',
  339. }))
  340. family = forms.ChoiceField(required=False, choices=ipaddress_family_choices, label='Address Family')
  341. vrf = forms.MultipleChoiceField(required=False, choices=ipaddress_vrf_choices, label='VRF',
  342. widget=forms.SelectMultiple(attrs={'size': 6}))
  343. tenant = forms.MultipleChoiceField(required=False, choices=tenant_choices, label='Tenant',
  344. widget=forms.SelectMultiple(attrs={'size': 6}))
  345. #
  346. # VLAN groups
  347. #
  348. class VLANGroupForm(forms.ModelForm, BootstrapMixin):
  349. slug = SlugField()
  350. class Meta:
  351. model = VLANGroup
  352. fields = ['site', 'name', 'slug']
  353. def vlangroup_site_choices():
  354. site_choices = Site.objects.annotate(vlangroup_count=Count('vlan_groups'))
  355. return [(s.slug, u'{} ({})'.format(s.name, s.vlangroup_count)) for s in site_choices]
  356. class VLANGroupFilterForm(forms.Form, BootstrapMixin):
  357. site = forms.MultipleChoiceField(required=False, choices=vlangroup_site_choices,
  358. widget=forms.SelectMultiple(attrs={'size': 8}))
  359. #
  360. # VLANs
  361. #
  362. class VLANForm(BootstrapMixin, CustomFieldForm):
  363. group = forms.ModelChoiceField(queryset=VLANGroup.objects.all(), required=False, label='Group', widget=APISelect(
  364. api_url='/api/ipam/vlan-groups/?site_id={{site}}',
  365. ))
  366. class Meta:
  367. model = VLAN
  368. fields = ['site', 'group', 'vid', 'name', 'tenant', 'status', 'role', 'description']
  369. help_texts = {
  370. 'site': "The site at which this VLAN exists",
  371. 'group': "VLAN group (optional)",
  372. 'vid': "Configured VLAN ID",
  373. 'name': "Configured VLAN name",
  374. 'status': "Operational status of this VLAN",
  375. 'role': "The primary function of this VLAN",
  376. }
  377. widgets = {
  378. 'site': forms.Select(attrs={'filter-for': 'group'}),
  379. }
  380. def __init__(self, *args, **kwargs):
  381. super(VLANForm, self).__init__(*args, **kwargs)
  382. # Limit VLAN group choices
  383. if self.is_bound and self.data.get('site'):
  384. self.fields['group'].queryset = VLANGroup.objects.filter(site__pk=self.data['site'])
  385. elif self.initial.get('site'):
  386. self.fields['group'].queryset = VLANGroup.objects.filter(site=self.initial['site'])
  387. else:
  388. self.fields['group'].choices = []
  389. class VLANFromCSVForm(forms.ModelForm):
  390. site = forms.ModelChoiceField(queryset=Site.objects.all(), to_field_name='name',
  391. error_messages={'invalid_choice': 'Device not found.'})
  392. group = forms.ModelChoiceField(queryset=VLANGroup.objects.all(), required=False, to_field_name='name',
  393. error_messages={'invalid_choice': 'VLAN group not found.'})
  394. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  395. error_messages={'invalid_choice': 'Tenant not found.'})
  396. status_name = forms.ChoiceField(choices=[(s[1], s[0]) for s in VLAN_STATUS_CHOICES])
  397. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False, to_field_name='name',
  398. error_messages={'invalid_choice': 'Invalid role.'})
  399. class Meta:
  400. model = VLAN
  401. fields = ['site', 'group', 'vid', 'name', 'tenant', 'status_name', 'role', 'description']
  402. def save(self, *args, **kwargs):
  403. m = super(VLANFromCSVForm, self).save(commit=False)
  404. # Assign VLAN status by name
  405. m.status = dict(self.fields['status_name'].choices)[self.cleaned_data['status_name']]
  406. if kwargs.get('commit'):
  407. m.save()
  408. return m
  409. class VLANImportForm(BulkImportForm, BootstrapMixin):
  410. csv = CSVDataField(csv_form=VLANFromCSVForm)
  411. class VLANBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  412. pk = forms.ModelMultipleChoiceField(queryset=VLAN.objects.all(), widget=forms.MultipleHiddenInput)
  413. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False)
  414. group = forms.ModelChoiceField(queryset=VLANGroup.objects.all(), required=False)
  415. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  416. status = forms.ChoiceField(choices=FORM_VLAN_STATUS_CHOICES, required=False)
  417. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False)
  418. description = forms.CharField(max_length=100, required=False)
  419. def vlan_site_choices():
  420. site_choices = Site.objects.annotate(vlan_count=Count('vlans'))
  421. return [(s.slug, u'{} ({})'.format(s.name, s.vlan_count)) for s in site_choices]
  422. def vlan_group_choices():
  423. group_choices = VLANGroup.objects.select_related('site').annotate(vlan_count=Count('vlans'))
  424. return [(g.pk, u'{} ({})'.format(g, g.vlan_count)) for g in group_choices]
  425. def vlan_tenant_choices():
  426. tenant_choices = Tenant.objects.annotate(vrf_count=Count('vlans'))
  427. return [(t.slug, u'{} ({})'.format(t.name, t.vrf_count)) for t in tenant_choices]
  428. def vlan_status_choices():
  429. status_counts = {}
  430. for status in VLAN.objects.values('status').annotate(count=Count('status')).order_by('status'):
  431. status_counts[status['status']] = status['count']
  432. return [(s[0], u'{} ({})'.format(s[1], status_counts.get(s[0], 0))) for s in VLAN_STATUS_CHOICES]
  433. def vlan_role_choices():
  434. role_choices = Role.objects.annotate(vlan_count=Count('vlans'))
  435. return [(r.slug, u'{} ({})'.format(r.name, r.vlan_count)) for r in role_choices]
  436. class VLANFilterForm(forms.Form, BootstrapMixin):
  437. site = forms.MultipleChoiceField(required=False, choices=vlan_site_choices,
  438. widget=forms.SelectMultiple(attrs={'size': 8}))
  439. group_id = forms.MultipleChoiceField(required=False, choices=vlan_group_choices, label='VLAN Group',
  440. widget=forms.SelectMultiple(attrs={'size': 8}))
  441. tenant = forms.MultipleChoiceField(required=False, choices=vlan_tenant_choices,
  442. widget=forms.SelectMultiple(attrs={'size': 8}))
  443. status = forms.MultipleChoiceField(required=False, choices=vlan_status_choices)
  444. role = forms.MultipleChoiceField(required=False, choices=vlan_role_choices,
  445. widget=forms.SelectMultiple(attrs={'size': 8}))