forms.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916
  1. from django import forms
  2. from django.core.exceptions import ValidationError
  3. from taggit.forms import TagField
  4. from dcim.constants import IFACE_TYPE_VIRTUAL, IFACE_MODE_ACCESS, IFACE_MODE_TAGGED_ALL, IFACE_MODE_CHOICES
  5. from dcim.forms import INTERFACE_MODE_HELP_TEXT
  6. from dcim.models import Device, DeviceRole, Interface, Platform, Rack, Region, Site
  7. from extras.forms import AddRemoveTagsForm, CustomFieldBulkEditForm, CustomFieldForm, CustomFieldFilterForm
  8. from ipam.models import IPAddress, VLANGroup, VLAN
  9. from tenancy.forms import TenancyForm
  10. from tenancy.forms import TenancyFilterForm
  11. from tenancy.models import Tenant
  12. from utilities.forms import (
  13. add_blank_choice, APISelect, APISelectMultiple, BootstrapMixin, BulkEditForm, BulkEditNullBooleanSelect,
  14. ChainedFieldsMixin, ChainedModelChoiceField, ChainedModelMultipleChoiceField, CommentField, ComponentForm,
  15. ConfirmationForm, CSVChoiceField, ExpandableNameField, FilterChoiceField, JSONField, SlugField,
  16. SmallTextarea, StaticSelect2, StaticSelect2Multiple
  17. )
  18. from .constants import VM_STATUS_CHOICES
  19. from .models import Cluster, ClusterGroup, ClusterType, VirtualMachine
  20. VIFACE_TYPE_CHOICES = (
  21. (IFACE_TYPE_VIRTUAL, 'Virtual'),
  22. )
  23. #
  24. # Cluster types
  25. #
  26. class ClusterTypeForm(BootstrapMixin, forms.ModelForm):
  27. slug = SlugField()
  28. class Meta:
  29. model = ClusterType
  30. fields = [
  31. 'name', 'slug',
  32. ]
  33. class ClusterTypeCSVForm(forms.ModelForm):
  34. slug = SlugField()
  35. class Meta:
  36. model = ClusterType
  37. fields = ClusterType.csv_headers
  38. help_texts = {
  39. 'name': 'Name of cluster type',
  40. }
  41. #
  42. # Cluster groups
  43. #
  44. class ClusterGroupForm(BootstrapMixin, forms.ModelForm):
  45. slug = SlugField()
  46. class Meta:
  47. model = ClusterGroup
  48. fields = [
  49. 'name', 'slug',
  50. ]
  51. class ClusterGroupCSVForm(forms.ModelForm):
  52. slug = SlugField()
  53. class Meta:
  54. model = ClusterGroup
  55. fields = ClusterGroup.csv_headers
  56. help_texts = {
  57. 'name': 'Name of cluster group',
  58. }
  59. #
  60. # Clusters
  61. #
  62. class ClusterForm(BootstrapMixin, CustomFieldForm):
  63. comments = CommentField()
  64. tags = TagField(
  65. required=False
  66. )
  67. class Meta:
  68. model = Cluster
  69. fields = [
  70. 'name', 'type', 'group', 'site', 'comments', 'tags',
  71. ]
  72. widgets = {
  73. 'type': APISelect(
  74. api_url="/api/virtualization/cluster-types/"
  75. ),
  76. 'group': APISelect(
  77. api_url="/api/virtualization/cluster-groups/"
  78. ),
  79. 'site': APISelect(
  80. api_url="/api/dcim/sites/"
  81. ),
  82. }
  83. class ClusterCSVForm(forms.ModelForm):
  84. type = forms.ModelChoiceField(
  85. queryset=ClusterType.objects.all(),
  86. to_field_name='name',
  87. help_text='Name of cluster type',
  88. error_messages={
  89. 'invalid_choice': 'Invalid cluster type name.',
  90. }
  91. )
  92. group = forms.ModelChoiceField(
  93. queryset=ClusterGroup.objects.all(),
  94. to_field_name='name',
  95. required=False,
  96. help_text='Name of cluster group',
  97. error_messages={
  98. 'invalid_choice': 'Invalid cluster group name.',
  99. }
  100. )
  101. site = forms.ModelChoiceField(
  102. queryset=Site.objects.all(),
  103. to_field_name='name',
  104. required=False,
  105. help_text='Name of assigned site',
  106. error_messages={
  107. 'invalid_choice': 'Invalid site name.',
  108. }
  109. )
  110. class Meta:
  111. model = Cluster
  112. fields = Cluster.csv_headers
  113. class ClusterBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
  114. pk = forms.ModelMultipleChoiceField(
  115. queryset=Cluster.objects.all(),
  116. widget=forms.MultipleHiddenInput()
  117. )
  118. type = forms.ModelChoiceField(
  119. queryset=ClusterType.objects.all(),
  120. required=False,
  121. widget=APISelect(
  122. api_url="/api/virtualization/cluster-types/"
  123. )
  124. )
  125. group = forms.ModelChoiceField(
  126. queryset=ClusterGroup.objects.all(),
  127. required=False,
  128. widget=APISelect(
  129. api_url="/api/virtualization/cluster-groups/"
  130. )
  131. )
  132. site = forms.ModelChoiceField(
  133. queryset=Site.objects.all(),
  134. required=False,
  135. widget=APISelect(
  136. api_url="/api/dcim/sites/"
  137. )
  138. )
  139. comments = CommentField(
  140. widget=SmallTextarea()
  141. )
  142. class Meta:
  143. nullable_fields = [
  144. 'group', 'site', 'comments',
  145. ]
  146. class ClusterFilterForm(BootstrapMixin, CustomFieldFilterForm):
  147. model = Cluster
  148. q = forms.CharField(required=False, label='Search')
  149. type = FilterChoiceField(
  150. queryset=ClusterType.objects.all(),
  151. to_field_name='slug',
  152. required=False,
  153. widget=APISelectMultiple(
  154. api_url="/api/virtualization/cluster-types/",
  155. value_field='slug',
  156. )
  157. )
  158. group = FilterChoiceField(
  159. queryset=ClusterGroup.objects.all(),
  160. to_field_name='slug',
  161. null_label='-- None --',
  162. required=False,
  163. widget=APISelectMultiple(
  164. api_url="/api/virtualization/cluster-groups/",
  165. value_field='slug',
  166. null_option=True,
  167. )
  168. )
  169. site = FilterChoiceField(
  170. queryset=Site.objects.all(),
  171. to_field_name='slug',
  172. null_label='-- None --',
  173. required=False,
  174. widget=APISelectMultiple(
  175. api_url="/api/dcim/sites/",
  176. value_field='slug',
  177. null_option=True,
  178. )
  179. )
  180. class ClusterAddDevicesForm(BootstrapMixin, ChainedFieldsMixin, forms.Form):
  181. region = forms.ModelChoiceField(
  182. queryset=Region.objects.all(),
  183. required=False,
  184. widget=APISelect(
  185. api_url="/api/dcim/regions/",
  186. filter_for={
  187. "site": "region_id",
  188. },
  189. attrs={
  190. 'nullable': 'true',
  191. }
  192. )
  193. )
  194. site = ChainedModelChoiceField(
  195. queryset=Site.objects.all(),
  196. chains=(
  197. ('region', 'region'),
  198. ),
  199. required=False,
  200. widget=APISelect(
  201. api_url='/api/dcim/sites/',
  202. filter_for={
  203. "rack": "site_id",
  204. "devices": "site_id",
  205. }
  206. )
  207. )
  208. rack = ChainedModelChoiceField(
  209. queryset=Rack.objects.all(),
  210. chains=(
  211. ('site', 'site'),
  212. ),
  213. required=False,
  214. widget=APISelect(
  215. api_url='/api/dcim/racks/',
  216. filter_for={
  217. "devices": "rack_id"
  218. },
  219. attrs={
  220. 'nullable': 'true',
  221. }
  222. )
  223. )
  224. devices = ChainedModelMultipleChoiceField(
  225. queryset=Device.objects.filter(cluster__isnull=True),
  226. chains=(
  227. ('site', 'site'),
  228. ('rack', 'rack'),
  229. ),
  230. widget=APISelectMultiple(
  231. api_url='/api/dcim/devices/',
  232. display_field='display_name',
  233. disabled_indicator='cluster'
  234. )
  235. )
  236. class Meta:
  237. fields = [
  238. 'region', 'site', 'rack', 'devices',
  239. ]
  240. def __init__(self, cluster, *args, **kwargs):
  241. self.cluster = cluster
  242. super().__init__(*args, **kwargs)
  243. self.fields['devices'].choices = []
  244. def clean(self):
  245. super().clean()
  246. # If the Cluster is assigned to a Site, all Devices must be assigned to that Site.
  247. if self.cluster.site is not None:
  248. for device in self.cleaned_data.get('devices', []):
  249. if device.site != self.cluster.site:
  250. raise ValidationError({
  251. 'devices': "{} belongs to a different site ({}) than the cluster ({})".format(
  252. device, device.site, self.cluster.site
  253. )
  254. })
  255. class ClusterRemoveDevicesForm(ConfirmationForm):
  256. pk = forms.ModelMultipleChoiceField(
  257. queryset=Device.objects.all(),
  258. widget=forms.MultipleHiddenInput()
  259. )
  260. #
  261. # Virtual Machines
  262. #
  263. class VirtualMachineForm(BootstrapMixin, TenancyForm, CustomFieldForm):
  264. cluster_group = forms.ModelChoiceField(
  265. queryset=ClusterGroup.objects.all(),
  266. required=False,
  267. widget=APISelect(
  268. api_url='/api/virtualization/cluster-groups/',
  269. filter_for={
  270. "cluster": "group_id",
  271. },
  272. attrs={
  273. 'nullable': 'true',
  274. }
  275. )
  276. )
  277. cluster = ChainedModelChoiceField(
  278. queryset=Cluster.objects.all(),
  279. chains=(
  280. ('group', 'cluster_group'),
  281. ),
  282. widget=APISelect(
  283. api_url='/api/virtualization/clusters/'
  284. )
  285. )
  286. tags = TagField(
  287. required=False
  288. )
  289. local_context_data = JSONField(
  290. required=False,
  291. label=''
  292. )
  293. class Meta:
  294. model = VirtualMachine
  295. fields = [
  296. 'name', 'status', 'cluster_group', 'cluster', 'role', 'tenant_group', 'tenant', 'platform', 'primary_ip4',
  297. 'primary_ip6', 'vcpus', 'memory', 'disk', 'comments', 'tags', 'local_context_data',
  298. ]
  299. help_texts = {
  300. 'local_context_data': "Local config context data overwrites all sources contexts in the final rendered "
  301. "config context",
  302. }
  303. widgets = {
  304. "status": StaticSelect2(),
  305. "role": APISelect(
  306. api_url="/api/dcim/device-roles/",
  307. additional_query_params={
  308. "vm_role": "True"
  309. }
  310. ),
  311. 'primary_ip4': StaticSelect2(),
  312. 'primary_ip6': StaticSelect2(),
  313. 'platform': APISelect(
  314. api_url='/api/dcim/platforms/'
  315. )
  316. }
  317. def __init__(self, *args, **kwargs):
  318. # Initialize helper selector
  319. instance = kwargs.get('instance')
  320. if instance.pk and instance.cluster is not None:
  321. initial = kwargs.get('initial', {}).copy()
  322. initial['cluster_group'] = instance.cluster.group
  323. kwargs['initial'] = initial
  324. super().__init__(*args, **kwargs)
  325. if self.instance.pk:
  326. # Compile list of choices for primary IPv4 and IPv6 addresses
  327. for family in [4, 6]:
  328. ip_choices = [(None, '---------')]
  329. # Collect interface IPs
  330. interface_ips = IPAddress.objects.prefetch_related('interface').filter(
  331. family=family, interface__virtual_machine=self.instance
  332. )
  333. if interface_ips:
  334. ip_choices.append(
  335. ('Interface IPs', [
  336. (ip.id, '{} ({})'.format(ip.address, ip.interface)) for ip in interface_ips
  337. ])
  338. )
  339. # Collect NAT IPs
  340. nat_ips = IPAddress.objects.prefetch_related('nat_inside').filter(
  341. family=family, nat_inside__interface__virtual_machine=self.instance
  342. )
  343. if nat_ips:
  344. ip_choices.append(
  345. ('NAT IPs', [
  346. (ip.id, '{} ({})'.format(ip.address, ip.nat_inside.address)) for ip in nat_ips
  347. ])
  348. )
  349. self.fields['primary_ip{}'.format(family)].choices = ip_choices
  350. else:
  351. # An object that doesn't exist yet can't have any IPs assigned to it
  352. self.fields['primary_ip4'].choices = []
  353. self.fields['primary_ip4'].widget.attrs['readonly'] = True
  354. self.fields['primary_ip6'].choices = []
  355. self.fields['primary_ip6'].widget.attrs['readonly'] = True
  356. class VirtualMachineCSVForm(forms.ModelForm):
  357. status = CSVChoiceField(
  358. choices=VM_STATUS_CHOICES,
  359. required=False,
  360. help_text='Operational status of device'
  361. )
  362. cluster = forms.ModelChoiceField(
  363. queryset=Cluster.objects.all(),
  364. to_field_name='name',
  365. help_text='Name of parent cluster',
  366. error_messages={
  367. 'invalid_choice': 'Invalid cluster name.',
  368. }
  369. )
  370. role = forms.ModelChoiceField(
  371. queryset=DeviceRole.objects.filter(
  372. vm_role=True
  373. ),
  374. required=False,
  375. to_field_name='name',
  376. help_text='Name of functional role',
  377. error_messages={
  378. 'invalid_choice': 'Invalid role name.'
  379. }
  380. )
  381. tenant = forms.ModelChoiceField(
  382. queryset=Tenant.objects.all(),
  383. required=False,
  384. to_field_name='name',
  385. help_text='Name of assigned tenant',
  386. error_messages={
  387. 'invalid_choice': 'Tenant not found.'
  388. }
  389. )
  390. platform = forms.ModelChoiceField(
  391. queryset=Platform.objects.all(),
  392. required=False,
  393. to_field_name='name',
  394. help_text='Name of assigned platform',
  395. error_messages={
  396. 'invalid_choice': 'Invalid platform.',
  397. }
  398. )
  399. class Meta:
  400. model = VirtualMachine
  401. fields = VirtualMachine.csv_headers
  402. class VirtualMachineBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
  403. pk = forms.ModelMultipleChoiceField(
  404. queryset=VirtualMachine.objects.all(),
  405. widget=forms.MultipleHiddenInput()
  406. )
  407. status = forms.ChoiceField(
  408. choices=add_blank_choice(VM_STATUS_CHOICES),
  409. required=False,
  410. initial='',
  411. widget=StaticSelect2(),
  412. )
  413. cluster = forms.ModelChoiceField(
  414. queryset=Cluster.objects.all(),
  415. required=False,
  416. widget=APISelect(
  417. api_url='/api/virtualization/clusters/'
  418. )
  419. )
  420. role = forms.ModelChoiceField(
  421. queryset=DeviceRole.objects.filter(
  422. vm_role=True
  423. ),
  424. required=False,
  425. widget=APISelect(
  426. api_url="/api/dcim/device-roles/",
  427. additional_query_params={
  428. "vm_role": "True"
  429. }
  430. )
  431. )
  432. tenant = forms.ModelChoiceField(
  433. queryset=Tenant.objects.all(),
  434. required=False,
  435. widget=APISelect(
  436. api_url='/api/tenancy/tenants/'
  437. )
  438. )
  439. platform = forms.ModelChoiceField(
  440. queryset=Platform.objects.all(),
  441. required=False,
  442. widget=APISelect(
  443. api_url='/api/dcim/platforms/'
  444. )
  445. )
  446. vcpus = forms.IntegerField(
  447. required=False,
  448. label='vCPUs'
  449. )
  450. memory = forms.IntegerField(
  451. required=False,
  452. label='Memory (MB)'
  453. )
  454. disk = forms.IntegerField(
  455. required=False,
  456. label='Disk (GB)'
  457. )
  458. comments = CommentField(
  459. widget=SmallTextarea()
  460. )
  461. class Meta:
  462. nullable_fields = [
  463. 'role', 'tenant', 'platform', 'vcpus', 'memory', 'disk', 'comments',
  464. ]
  465. class VirtualMachineFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
  466. model = VirtualMachine
  467. field_order = [
  468. 'q', 'cluster_group', 'cluster_type', 'cluster_id', 'status', 'role', 'region', 'site', 'tenant_group',
  469. 'tenant', 'platform', 'mac_address',
  470. ]
  471. q = forms.CharField(
  472. required=False,
  473. label='Search'
  474. )
  475. cluster_group = FilterChoiceField(
  476. queryset=ClusterGroup.objects.all(),
  477. to_field_name='slug',
  478. null_label='-- None --',
  479. widget=APISelectMultiple(
  480. api_url='/api/virtualization/cluster-groups/',
  481. value_field="slug",
  482. null_option=True,
  483. )
  484. )
  485. cluster_type = FilterChoiceField(
  486. queryset=ClusterType.objects.all(),
  487. to_field_name='slug',
  488. null_label='-- None --',
  489. widget=APISelectMultiple(
  490. api_url='/api/virtualization/cluster-types/',
  491. value_field="slug",
  492. null_option=True,
  493. )
  494. )
  495. cluster_id = FilterChoiceField(
  496. queryset=Cluster.objects.all(),
  497. label='Cluster',
  498. widget=APISelectMultiple(
  499. api_url='/api/virtualization/clusters/',
  500. )
  501. )
  502. region = FilterChoiceField(
  503. queryset=Region.objects.all(),
  504. to_field_name='slug',
  505. required=False,
  506. widget=APISelectMultiple(
  507. api_url='/api/dcim/regions/',
  508. value_field="slug",
  509. null_option=True,
  510. )
  511. )
  512. site = FilterChoiceField(
  513. queryset=Site.objects.all(),
  514. to_field_name='slug',
  515. null_label='-- None --',
  516. widget=APISelectMultiple(
  517. api_url='/api/dcim/sites/',
  518. value_field="slug",
  519. null_option=True,
  520. )
  521. )
  522. role = FilterChoiceField(
  523. queryset=DeviceRole.objects.filter(vm_role=True),
  524. to_field_name='slug',
  525. null_label='-- None --',
  526. widget=APISelectMultiple(
  527. api_url='/api/dcim/device-roles/',
  528. value_field="slug",
  529. null_option=True,
  530. additional_query_params={
  531. 'vm_role': "True"
  532. }
  533. )
  534. )
  535. status = forms.MultipleChoiceField(
  536. choices=VM_STATUS_CHOICES,
  537. required=False,
  538. widget=StaticSelect2Multiple()
  539. )
  540. platform = FilterChoiceField(
  541. queryset=Platform.objects.all(),
  542. to_field_name='slug',
  543. null_label='-- None --',
  544. widget=APISelectMultiple(
  545. api_url='/api/dcim/platforms/',
  546. value_field="slug",
  547. null_option=True,
  548. )
  549. )
  550. mac_address = forms.CharField(
  551. required=False,
  552. label='MAC address'
  553. )
  554. #
  555. # VM interfaces
  556. #
  557. class InterfaceForm(BootstrapMixin, forms.ModelForm):
  558. untagged_vlan = forms.ModelChoiceField(
  559. queryset=VLAN.objects.all(),
  560. required=False,
  561. widget=APISelect(
  562. api_url="/api/ipam/vlans/",
  563. display_field='display_name',
  564. full=True
  565. )
  566. )
  567. tagged_vlans = forms.ModelMultipleChoiceField(
  568. queryset=VLAN.objects.all(),
  569. required=False,
  570. widget=APISelectMultiple(
  571. api_url="/api/ipam/vlans/",
  572. display_field='display_name',
  573. full=True
  574. )
  575. )
  576. tags = TagField(
  577. required=False
  578. )
  579. class Meta:
  580. model = Interface
  581. fields = [
  582. 'virtual_machine', 'name', 'type', 'enabled', 'mac_address', 'mtu', 'description', 'mode', 'tags',
  583. 'untagged_vlan', 'tagged_vlans',
  584. ]
  585. widgets = {
  586. 'virtual_machine': forms.HiddenInput(),
  587. 'type': forms.HiddenInput(),
  588. 'mode': StaticSelect2()
  589. }
  590. labels = {
  591. 'mode': '802.1Q Mode',
  592. }
  593. help_texts = {
  594. 'mode': INTERFACE_MODE_HELP_TEXT,
  595. }
  596. def __init__(self, *args, **kwargs):
  597. super().__init__(*args, **kwargs)
  598. # Limit VLan choices to those in: global vlans, global groups, the current site's group, the current site
  599. vlan_choices = []
  600. global_vlans = VLAN.objects.filter(site=None, group=None)
  601. vlan_choices.append(
  602. ('Global', [(vlan.pk, vlan) for vlan in global_vlans])
  603. )
  604. for group in VLANGroup.objects.filter(site=None):
  605. global_group_vlans = VLAN.objects.filter(group=group)
  606. vlan_choices.append(
  607. (group.name, [(vlan.pk, vlan) for vlan in global_group_vlans])
  608. )
  609. site = getattr(self.instance.device, 'site', None)
  610. if site is not None:
  611. # Add non-grouped site VLANs
  612. site_vlans = VLAN.objects.filter(site=site, group=None)
  613. vlan_choices.append((site.name, [(vlan.pk, vlan) for vlan in site_vlans]))
  614. # Add grouped site VLANs
  615. for group in VLANGroup.objects.filter(site=site):
  616. site_group_vlans = VLAN.objects.filter(group=group)
  617. vlan_choices.append((
  618. '{} / {}'.format(group.site.name, group.name),
  619. [(vlan.pk, vlan) for vlan in site_group_vlans]
  620. ))
  621. self.fields['untagged_vlan'].choices = [(None, '---------')] + vlan_choices
  622. self.fields['tagged_vlans'].choices = vlan_choices
  623. def clean(self):
  624. super().clean()
  625. # Validate VLAN assignments
  626. tagged_vlans = self.cleaned_data['tagged_vlans']
  627. # Untagged interfaces cannot be assigned tagged VLANs
  628. if self.cleaned_data['mode'] == IFACE_MODE_ACCESS and tagged_vlans:
  629. raise forms.ValidationError({
  630. 'mode': "An access interface cannot have tagged VLANs assigned."
  631. })
  632. # Remove all tagged VLAN assignments from "tagged all" interfaces
  633. elif self.cleaned_data['mode'] == IFACE_MODE_TAGGED_ALL:
  634. self.cleaned_data['tagged_vlans'] = []
  635. class InterfaceCreateForm(ComponentForm):
  636. name_pattern = ExpandableNameField(
  637. label='Name'
  638. )
  639. type = forms.ChoiceField(
  640. choices=VIFACE_TYPE_CHOICES,
  641. initial=IFACE_TYPE_VIRTUAL,
  642. widget=forms.HiddenInput()
  643. )
  644. enabled = forms.BooleanField(
  645. required=False
  646. )
  647. mtu = forms.IntegerField(
  648. required=False,
  649. min_value=1,
  650. max_value=32767,
  651. label='MTU'
  652. )
  653. mac_address = forms.CharField(
  654. required=False,
  655. label='MAC Address'
  656. )
  657. description = forms.CharField(
  658. max_length=100,
  659. required=False
  660. )
  661. mode = forms.ChoiceField(
  662. choices=add_blank_choice(IFACE_MODE_CHOICES),
  663. required=False,
  664. widget=StaticSelect2(),
  665. )
  666. untagged_vlan = forms.ModelChoiceField(
  667. queryset=VLAN.objects.all(),
  668. required=False,
  669. widget=APISelect(
  670. api_url="/api/ipam/vlans/",
  671. display_field='display_name',
  672. full=True
  673. )
  674. )
  675. tagged_vlans = forms.ModelMultipleChoiceField(
  676. queryset=VLAN.objects.all(),
  677. required=False,
  678. widget=APISelectMultiple(
  679. api_url="/api/ipam/vlans/",
  680. display_field='display_name',
  681. full=True
  682. )
  683. )
  684. tags = TagField(
  685. required=False
  686. )
  687. def __init__(self, *args, **kwargs):
  688. # Set interfaces enabled by default
  689. kwargs['initial'] = kwargs.get('initial', {}).copy()
  690. kwargs['initial'].update({'enabled': True})
  691. super().__init__(*args, **kwargs)
  692. # Limit VLan choices to those in: global vlans, global groups, the current site's group, the current site
  693. vlan_choices = []
  694. global_vlans = VLAN.objects.filter(site=None, group=None)
  695. vlan_choices.append(
  696. ('Global', [(vlan.pk, vlan) for vlan in global_vlans])
  697. )
  698. for group in VLANGroup.objects.filter(site=None):
  699. global_group_vlans = VLAN.objects.filter(group=group)
  700. vlan_choices.append(
  701. (group.name, [(vlan.pk, vlan) for vlan in global_group_vlans])
  702. )
  703. site = getattr(self.parent.cluster, 'site', None)
  704. if site is not None:
  705. # Add non-grouped site VLANs
  706. site_vlans = VLAN.objects.filter(site=site, group=None)
  707. vlan_choices.append((site.name, [(vlan.pk, vlan) for vlan in site_vlans]))
  708. # Add grouped site VLANs
  709. for group in VLANGroup.objects.filter(site=site):
  710. site_group_vlans = VLAN.objects.filter(group=group)
  711. vlan_choices.append((
  712. '{} / {}'.format(group.site.name, group.name),
  713. [(vlan.pk, vlan) for vlan in site_group_vlans]
  714. ))
  715. self.fields['untagged_vlan'].choices = [(None, '---------')] + vlan_choices
  716. self.fields['tagged_vlans'].choices = vlan_choices
  717. class InterfaceBulkEditForm(BootstrapMixin, BulkEditForm):
  718. pk = forms.ModelMultipleChoiceField(
  719. queryset=Interface.objects.all(),
  720. widget=forms.MultipleHiddenInput()
  721. )
  722. enabled = forms.NullBooleanField(
  723. required=False,
  724. widget=BulkEditNullBooleanSelect()
  725. )
  726. mtu = forms.IntegerField(
  727. required=False,
  728. min_value=1,
  729. max_value=32767,
  730. label='MTU'
  731. )
  732. description = forms.CharField(
  733. max_length=100,
  734. required=False
  735. )
  736. mode = forms.ChoiceField(
  737. choices=add_blank_choice(IFACE_MODE_CHOICES),
  738. required=False,
  739. widget=StaticSelect2()
  740. )
  741. untagged_vlan = forms.ModelChoiceField(
  742. queryset=VLAN.objects.all(),
  743. required=False,
  744. widget=APISelect(
  745. api_url="/api/ipam/vlans/",
  746. display_field='display_name',
  747. full=True
  748. )
  749. )
  750. tagged_vlans = forms.ModelMultipleChoiceField(
  751. queryset=VLAN.objects.all(),
  752. required=False,
  753. widget=APISelectMultiple(
  754. api_url="/api/ipam/vlans/",
  755. display_field='display_name',
  756. full=True
  757. )
  758. )
  759. class Meta:
  760. nullable_fields = [
  761. 'mtu', 'description',
  762. ]
  763. def __init__(self, *args, **kwargs):
  764. super().__init__(*args, **kwargs)
  765. # Limit VLan choices to those in: global vlans, global groups, the current site's group, the current site
  766. vlan_choices = []
  767. global_vlans = VLAN.objects.filter(site=None, group=None)
  768. vlan_choices.append(
  769. ('Global', [(vlan.pk, vlan) for vlan in global_vlans])
  770. )
  771. for group in VLANGroup.objects.filter(site=None):
  772. global_group_vlans = VLAN.objects.filter(group=group)
  773. vlan_choices.append(
  774. (group.name, [(vlan.pk, vlan) for vlan in global_group_vlans])
  775. )
  776. if self.parent_obj.cluster is not None:
  777. site = getattr(self.parent_obj.cluster, 'site', None)
  778. if site is not None:
  779. # Add non-grouped site VLANs
  780. site_vlans = VLAN.objects.filter(site=site, group=None)
  781. vlan_choices.append((site.name, [(vlan.pk, vlan) for vlan in site_vlans]))
  782. # Add grouped site VLANs
  783. for group in VLANGroup.objects.filter(site=site):
  784. site_group_vlans = VLAN.objects.filter(group=group)
  785. vlan_choices.append((
  786. '{} / {}'.format(group.site.name, group.name),
  787. [(vlan.pk, vlan) for vlan in site_group_vlans]
  788. ))
  789. self.fields['untagged_vlan'].choices = [(None, '---------')] + vlan_choices
  790. self.fields['tagged_vlans'].choices = vlan_choices
  791. #
  792. # Bulk VirtualMachine component creation
  793. #
  794. class VirtualMachineBulkAddComponentForm(BootstrapMixin, forms.Form):
  795. pk = forms.ModelMultipleChoiceField(
  796. queryset=VirtualMachine.objects.all(),
  797. widget=forms.MultipleHiddenInput()
  798. )
  799. name_pattern = ExpandableNameField(
  800. label='Name'
  801. )
  802. class VirtualMachineBulkAddInterfaceForm(VirtualMachineBulkAddComponentForm):
  803. type = forms.ChoiceField(
  804. choices=VIFACE_TYPE_CHOICES,
  805. initial=IFACE_TYPE_VIRTUAL,
  806. widget=forms.HiddenInput()
  807. )
  808. enabled = forms.BooleanField(
  809. required=False,
  810. initial=True
  811. )
  812. mtu = forms.IntegerField(
  813. required=False,
  814. min_value=1,
  815. max_value=32767,
  816. label='MTU'
  817. )
  818. description = forms.CharField(
  819. max_length=100,
  820. required=False
  821. )