forms.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592
  1. from netaddr import IPNetwork
  2. from django import forms
  3. from django.db.models import Count
  4. from dcim.models import Site, Device, Interface
  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(forms.ModelForm, BootstrapMixin):
  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(forms.Form, BootstrapMixin):
  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(forms.ModelForm, BootstrapMixin):
  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(forms.Form, BootstrapMixin):
  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(forms.ModelForm, BootstrapMixin):
  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. data = self.cleaned_data['prefix']
  133. try:
  134. prefix = IPNetwork(data)
  135. except:
  136. raise
  137. if prefix.version == 4 and prefix.prefixlen == 32:
  138. raise forms.ValidationError("Cannot create host addresses (/32) as prefixes. These should be IPv4 "
  139. "addresses instead.")
  140. elif prefix.version == 6 and prefix.prefixlen == 128:
  141. raise forms.ValidationError("Cannot create host addresses (/128) as prefixes. These should be IPv6 "
  142. "addresses instead.")
  143. return data
  144. class PrefixFromCSVForm(forms.ModelForm):
  145. vrf = forms.ModelChoiceField(queryset=VRF.objects.all(), required=False, to_field_name='rd',
  146. error_messages={'invalid_choice': 'VRF not found.'})
  147. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  148. error_messages={'invalid_choice': 'Tenant not found.'})
  149. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False, to_field_name='name',
  150. error_messages={'invalid_choice': 'Site not found.'})
  151. vlan_group_name = forms.CharField(required=False)
  152. vlan_vid = forms.IntegerField(required=False)
  153. status_name = forms.ChoiceField(choices=[(s[1], s[0]) for s in PREFIX_STATUS_CHOICES])
  154. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False, to_field_name='name',
  155. error_messages={'invalid_choice': 'Invalid role.'})
  156. class Meta:
  157. model = Prefix
  158. fields = ['prefix', 'vrf', 'tenant', 'site', 'vlan_group_name', 'vlan_vid', 'status_name', 'role',
  159. 'description']
  160. def clean(self):
  161. super(PrefixFromCSVForm, self).clean()
  162. site = self.cleaned_data.get('site')
  163. vlan_group_name = self.cleaned_data.get('vlan_group_name')
  164. vlan_vid = self.cleaned_data.get('vlan_vid')
  165. # Validate VLAN
  166. vlan_group = None
  167. if vlan_group_name:
  168. try:
  169. vlan_group = VLANGroup.objects.get(site=site, name=vlan_group_name)
  170. except VLANGroup.DoesNotExist:
  171. self.add_error('vlan_group_name', "Invalid VLAN group ({} - {}).".format(site, vlan_group_name))
  172. if vlan_vid and vlan_group:
  173. try:
  174. self.instance.vlan = VLAN.objects.get(group=vlan_group, vid=vlan_vid)
  175. except VLAN.DoesNotExist:
  176. self.add_error('vlan_vid', "Invalid VLAN ID ({} - {}).".format(vlan_group, vlan_vid))
  177. elif vlan_vid and site:
  178. try:
  179. self.instance.vlan = VLAN.objects.get(site=site, vid=vlan_vid)
  180. except VLAN.MultipleObjectsReturned:
  181. self.add_error('vlan_vid', "Multiple VLANs found ({} - VID {})".format(site, vlan_vid))
  182. elif vlan_vid:
  183. self.add_error('vlan_vid', "Must specify site and/or VLAN group when assigning a VLAN.")
  184. def save(self, *args, **kwargs):
  185. m = super(PrefixFromCSVForm, self).save(commit=False)
  186. # Assign Prefix status by name
  187. m.status = dict(self.fields['status_name'].choices)[self.cleaned_data['status_name']]
  188. if kwargs.get('commit'):
  189. m.save()
  190. return m
  191. class PrefixImportForm(BulkImportForm, BootstrapMixin):
  192. csv = CSVDataField(csv_form=PrefixFromCSVForm)
  193. class PrefixBulkEditForm(forms.Form, BootstrapMixin):
  194. pk = forms.ModelMultipleChoiceField(queryset=Prefix.objects.all(), widget=forms.MultipleHiddenInput)
  195. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False)
  196. vrf = forms.TypedChoiceField(choices=bulkedit_vrf_choices, coerce=int, required=False, label='VRF')
  197. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  198. status = forms.ChoiceField(choices=FORM_PREFIX_STATUS_CHOICES, required=False)
  199. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False)
  200. description = forms.CharField(max_length=100, required=False)
  201. def prefix_vrf_choices():
  202. vrf_choices = VRF.objects.annotate(prefix_count=Count('prefixes'))
  203. return [(v.pk, u'{} ({})'.format(v.name, v.prefix_count)) for v in vrf_choices]
  204. def tenant_choices():
  205. tenant_choices = Tenant.objects.all()
  206. return [(t.slug, t.name) for t in tenant_choices]
  207. def prefix_site_choices():
  208. site_choices = Site.objects.annotate(prefix_count=Count('prefixes'))
  209. return [(s.slug, u'{} ({})'.format(s.name, s.prefix_count)) for s in site_choices]
  210. def prefix_status_choices():
  211. status_counts = {}
  212. for status in Prefix.objects.values('status').annotate(count=Count('status')).order_by('status'):
  213. status_counts[status['status']] = status['count']
  214. return [(s[0], u'{} ({})'.format(s[1], status_counts.get(s[0], 0))) for s in PREFIX_STATUS_CHOICES]
  215. def prefix_role_choices():
  216. role_choices = Role.objects.annotate(prefix_count=Count('prefixes'))
  217. return [(r.slug, u'{} ({})'.format(r.name, r.prefix_count)) for r in role_choices]
  218. class PrefixFilterForm(forms.Form, BootstrapMixin):
  219. parent = forms.CharField(required=False, label='Search Within', widget=forms.TextInput(attrs={
  220. 'placeholder': 'Network',
  221. }))
  222. vrf = forms.MultipleChoiceField(required=False, choices=prefix_vrf_choices, label='VRF',
  223. widget=forms.SelectMultiple(attrs={'size': 6}))
  224. tenant = forms.MultipleChoiceField(required=False, choices=tenant_choices, label='Tenant',
  225. widget=forms.SelectMultiple(attrs={'size': 6}))
  226. status = forms.MultipleChoiceField(required=False, choices=prefix_status_choices,
  227. widget=forms.SelectMultiple(attrs={'size': 6}))
  228. site = forms.MultipleChoiceField(required=False, choices=prefix_site_choices,
  229. widget=forms.SelectMultiple(attrs={'size': 6}))
  230. role = forms.MultipleChoiceField(required=False, choices=prefix_role_choices,
  231. widget=forms.SelectMultiple(attrs={'size': 6}))
  232. expand = forms.BooleanField(required=False, label='Expand prefix hierarchy')
  233. #
  234. # IP addresses
  235. #
  236. class IPAddressForm(forms.ModelForm, BootstrapMixin):
  237. nat_site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False, label='Site',
  238. widget=forms.Select(attrs={'filter-for': 'nat_device'}))
  239. nat_device = forms.ModelChoiceField(queryset=Device.objects.all(), required=False, label='Device',
  240. widget=APISelect(api_url='/api/dcim/devices/?site_id={{nat_site}}',
  241. attrs={'filter-for': 'nat_inside'}))
  242. livesearch = forms.CharField(required=False, label='IP Address', widget=Livesearch(
  243. query_key='q', query_url='ipam-api:ipaddress_list', field_to_update='nat_inside', obj_label='address')
  244. )
  245. nat_inside = forms.ModelChoiceField(queryset=IPAddress.objects.all(), required=False, label='NAT (Inside)',
  246. widget=APISelect(api_url='/api/ipam/ip-addresses/?device_id={{nat_device}}',
  247. display_field='address'))
  248. class Meta:
  249. model = IPAddress
  250. fields = ['address', 'vrf', 'tenant', 'nat_device', 'nat_inside', 'description']
  251. help_texts = {
  252. 'address': "IPv4 or IPv6 address and mask",
  253. 'vrf': "VRF (if applicable)",
  254. }
  255. def __init__(self, *args, **kwargs):
  256. super(IPAddressForm, self).__init__(*args, **kwargs)
  257. self.fields['vrf'].empty_label = 'Global'
  258. if self.instance.nat_inside:
  259. nat_inside = self.instance.nat_inside
  260. # If the IP is assigned to an interface, populate site/device fields accordingly
  261. if self.instance.nat_inside.interface:
  262. self.initial['nat_site'] = self.instance.nat_inside.interface.device.rack.site.pk
  263. self.initial['nat_device'] = self.instance.nat_inside.interface.device.pk
  264. self.fields['nat_device'].queryset = Device.objects.filter(
  265. rack__site=nat_inside.interface.device.rack.site)
  266. self.fields['nat_inside'].queryset = IPAddress.objects.filter(
  267. interface__device=nat_inside.interface.device)
  268. else:
  269. self.fields['nat_inside'].queryset = IPAddress.objects.filter(pk=nat_inside.pk)
  270. else:
  271. # Initialize nat_device choices if nat_site is set
  272. if self.is_bound and self.data.get('nat_site'):
  273. self.fields['nat_device'].queryset = Device.objects.filter(rack__site__pk=self.data['nat_site'])
  274. elif self.initial.get('nat_site'):
  275. self.fields['nat_device'].queryset = Device.objects.filter(rack__site=self.initial['nat_site'])
  276. else:
  277. self.fields['nat_device'].choices = []
  278. # Initialize nat_inside choices if nat_device is set
  279. if self.is_bound and self.data.get('nat_device'):
  280. self.fields['nat_inside'].queryset = IPAddress.objects.filter(
  281. interface__device__pk=self.data['nat_device'])
  282. elif self.initial.get('nat_device'):
  283. self.fields['nat_inside'].queryset = IPAddress.objects.filter(
  284. interface__device__pk=self.initial['nat_device'])
  285. else:
  286. self.fields['nat_inside'].choices = []
  287. class IPAddressFromCSVForm(forms.ModelForm):
  288. vrf = forms.ModelChoiceField(queryset=VRF.objects.all(), required=False, to_field_name='rd',
  289. error_messages={'invalid_choice': 'VRF not found.'})
  290. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  291. error_messages={'invalid_choice': 'Tenant not found.'})
  292. device = forms.ModelChoiceField(queryset=Device.objects.all(), required=False, to_field_name='name',
  293. error_messages={'invalid_choice': 'Device not found.'})
  294. interface_name = forms.CharField(required=False)
  295. is_primary = forms.BooleanField(required=False)
  296. class Meta:
  297. model = IPAddress
  298. fields = ['address', 'vrf', 'tenant', 'device', 'interface_name', 'is_primary', 'description']
  299. def clean(self):
  300. device = self.cleaned_data.get('device')
  301. interface_name = self.cleaned_data.get('interface_name')
  302. is_primary = self.cleaned_data.get('is_primary')
  303. # Validate interface
  304. if device and interface_name:
  305. try:
  306. Interface.objects.get(device=device, name=interface_name)
  307. except Interface.DoesNotExist:
  308. self.add_error('interface_name', "Invalid interface ({}) for {}".format(interface_name, device))
  309. elif device and not interface_name:
  310. self.add_error('interface_name', "Device set ({}) but interface missing".format(device))
  311. elif interface_name and not device:
  312. self.add_error('device', "Interface set ({}) but device missing or invalid".format(interface_name))
  313. # Validate is_primary
  314. if is_primary and not device:
  315. self.add_error('is_primary', "No device specified; cannot set as primary IP")
  316. def save(self, commit=True):
  317. # Set interface
  318. if self.cleaned_data['device'] and self.cleaned_data['interface_name']:
  319. self.instance.interface = Interface.objects.get(device=self.cleaned_data['device'],
  320. name=self.cleaned_data['interface_name'])
  321. # Set as primary for device
  322. if self.cleaned_data['is_primary']:
  323. if self.instance.address.version == 4:
  324. self.instance.primary_ip4_for = self.cleaned_data['device']
  325. elif self.instance.address.version == 6:
  326. self.instance.primary_ip6_for = self.cleaned_data['device']
  327. return super(IPAddressFromCSVForm, self).save(commit=commit)
  328. class IPAddressImportForm(BulkImportForm, BootstrapMixin):
  329. csv = CSVDataField(csv_form=IPAddressFromCSVForm)
  330. class IPAddressBulkEditForm(forms.Form, BootstrapMixin):
  331. pk = forms.ModelMultipleChoiceField(queryset=IPAddress.objects.all(), widget=forms.MultipleHiddenInput)
  332. vrf = forms.TypedChoiceField(choices=bulkedit_vrf_choices, coerce=int, required=False, label='VRF')
  333. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  334. description = forms.CharField(max_length=100, required=False)
  335. def ipaddress_family_choices():
  336. return [('', 'All'), (4, 'IPv4'), (6, 'IPv6')]
  337. def ipaddress_vrf_choices():
  338. vrf_choices = VRF.objects.annotate(ipaddress_count=Count('ip_addresses'))
  339. return [(v.pk, u'{} ({})'.format(v.name, v.ipaddress_count)) for v in vrf_choices]
  340. class IPAddressFilterForm(forms.Form, BootstrapMixin):
  341. parent = forms.CharField(required=False, label='Search Within', widget=forms.TextInput(attrs={
  342. 'placeholder': 'Prefix',
  343. }))
  344. family = forms.ChoiceField(required=False, choices=ipaddress_family_choices, label='Address Family')
  345. vrf = forms.MultipleChoiceField(required=False, choices=ipaddress_vrf_choices, label='VRF',
  346. widget=forms.SelectMultiple(attrs={'size': 6}))
  347. tenant = forms.MultipleChoiceField(required=False, choices=tenant_choices, label='Tenant',
  348. widget=forms.SelectMultiple(attrs={'size': 6}))
  349. #
  350. # VLAN groups
  351. #
  352. class VLANGroupForm(forms.ModelForm, BootstrapMixin):
  353. slug = SlugField()
  354. class Meta:
  355. model = VLANGroup
  356. fields = ['site', 'name', 'slug']
  357. def vlangroup_site_choices():
  358. site_choices = Site.objects.annotate(vlangroup_count=Count('vlan_groups'))
  359. return [(s.slug, u'{} ({})'.format(s.name, s.vlangroup_count)) for s in site_choices]
  360. class VLANGroupFilterForm(forms.Form, BootstrapMixin):
  361. site = forms.MultipleChoiceField(required=False, choices=vlangroup_site_choices,
  362. widget=forms.SelectMultiple(attrs={'size': 8}))
  363. #
  364. # VLANs
  365. #
  366. class VLANForm(forms.ModelForm, BootstrapMixin):
  367. group = forms.ModelChoiceField(queryset=VLANGroup.objects.all(), required=False, label='Group', widget=APISelect(
  368. api_url='/api/ipam/vlan-groups/?site_id={{site}}',
  369. ))
  370. class Meta:
  371. model = VLAN
  372. fields = ['site', 'group', 'vid', 'name', 'tenant', 'status', 'role', 'description']
  373. help_texts = {
  374. 'site': "The site at which this VLAN exists",
  375. 'group': "VLAN group (optional)",
  376. 'vid': "Configured VLAN ID",
  377. 'name': "Configured VLAN name",
  378. 'status': "Operational status of this VLAN",
  379. 'role': "The primary function of this VLAN",
  380. }
  381. widgets = {
  382. 'site': forms.Select(attrs={'filter-for': 'group'}),
  383. }
  384. def __init__(self, *args, **kwargs):
  385. super(VLANForm, self).__init__(*args, **kwargs)
  386. # Limit VLAN group choices
  387. if self.is_bound and self.data.get('site'):
  388. self.fields['group'].queryset = VLANGroup.objects.filter(site__pk=self.data['site'])
  389. elif self.initial.get('site'):
  390. self.fields['group'].queryset = VLANGroup.objects.filter(site=self.initial['site'])
  391. else:
  392. self.fields['group'].choices = []
  393. class VLANFromCSVForm(forms.ModelForm):
  394. site = forms.ModelChoiceField(queryset=Site.objects.all(), to_field_name='name',
  395. error_messages={'invalid_choice': 'Device not found.'})
  396. group = forms.ModelChoiceField(queryset=VLANGroup.objects.all(), required=False, to_field_name='name',
  397. error_messages={'invalid_choice': 'VLAN group not found.'})
  398. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  399. error_messages={'invalid_choice': 'Tenant not found.'})
  400. status_name = forms.ChoiceField(choices=[(s[1], s[0]) for s in VLAN_STATUS_CHOICES])
  401. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False, to_field_name='name',
  402. error_messages={'invalid_choice': 'Invalid role.'})
  403. class Meta:
  404. model = VLAN
  405. fields = ['site', 'group', 'vid', 'name', 'tenant', 'status_name', 'role', 'description']
  406. def save(self, *args, **kwargs):
  407. m = super(VLANFromCSVForm, self).save(commit=False)
  408. # Assign VLAN status by name
  409. m.status = dict(self.fields['status_name'].choices)[self.cleaned_data['status_name']]
  410. if kwargs.get('commit'):
  411. m.save()
  412. return m
  413. class VLANImportForm(BulkImportForm, BootstrapMixin):
  414. csv = CSVDataField(csv_form=VLANFromCSVForm)
  415. class VLANBulkEditForm(forms.Form, BootstrapMixin):
  416. pk = forms.ModelMultipleChoiceField(queryset=VLAN.objects.all(), widget=forms.MultipleHiddenInput)
  417. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False)
  418. group = forms.ModelChoiceField(queryset=VLANGroup.objects.all(), required=False)
  419. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  420. status = forms.ChoiceField(choices=FORM_VLAN_STATUS_CHOICES, required=False)
  421. role = forms.ModelChoiceField(queryset=Role.objects.all(), required=False)
  422. description = forms.CharField(max_length=100, required=False)
  423. def vlan_site_choices():
  424. site_choices = Site.objects.annotate(vlan_count=Count('vlans'))
  425. return [(s.slug, u'{} ({})'.format(s.name, s.vlan_count)) for s in site_choices]
  426. def vlan_group_choices():
  427. group_choices = VLANGroup.objects.select_related('site').annotate(vlan_count=Count('vlans'))
  428. return [(g.pk, u'{} ({})'.format(g, g.vlan_count)) for g in group_choices]
  429. def vlan_tenant_choices():
  430. tenant_choices = Tenant.objects.annotate(vrf_count=Count('vlans'))
  431. return [(t.slug, u'{} ({})'.format(t.name, t.vrf_count)) for t in tenant_choices]
  432. def vlan_status_choices():
  433. status_counts = {}
  434. for status in VLAN.objects.values('status').annotate(count=Count('status')).order_by('status'):
  435. status_counts[status['status']] = status['count']
  436. return [(s[0], u'{} ({})'.format(s[1], status_counts.get(s[0], 0))) for s in VLAN_STATUS_CHOICES]
  437. def vlan_role_choices():
  438. role_choices = Role.objects.annotate(vlan_count=Count('vlans'))
  439. return [(r.slug, u'{} ({})'.format(r.name, r.vlan_count)) for r in role_choices]
  440. class VLANFilterForm(forms.Form, BootstrapMixin):
  441. site = forms.MultipleChoiceField(required=False, choices=vlan_site_choices,
  442. widget=forms.SelectMultiple(attrs={'size': 8}))
  443. group_id = forms.MultipleChoiceField(required=False, choices=vlan_group_choices, label='VLAN Group',
  444. widget=forms.SelectMultiple(attrs={'size': 8}))
  445. tenant = forms.MultipleChoiceField(required=False, choices=vlan_tenant_choices,
  446. widget=forms.SelectMultiple(attrs={'size': 8}))
  447. status = forms.MultipleChoiceField(required=False, choices=vlan_status_choices)
  448. role = forms.MultipleChoiceField(required=False, choices=vlan_role_choices,
  449. widget=forms.SelectMultiple(attrs={'size': 8}))