| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901 |
- from django import forms
- from django.contrib.contenttypes.models import ContentType
- from django.core.exceptions import ValidationError
- from django.utils.translation import gettext as _
- from dcim.choices import InterfaceModeChoices
- from dcim.constants import INTERFACE_MTU_MAX, INTERFACE_MTU_MIN
- from dcim.forms import InterfaceCommonForm, INTERFACE_MODE_HELP_TEXT
- from dcim.models import Device, DeviceRole, Platform, Rack, Region, Site, SiteGroup
- from extras.forms import (
- AddRemoveTagsForm, CustomFieldForm, CustomFieldBulkEditForm, CustomFieldModelCSVForm, CustomFieldModelForm,
- CustomFieldFilterForm,
- )
- from extras.models import Tag
- from ipam.models import IPAddress, VLAN, VLANGroup
- from tenancy.forms import TenancyFilterForm, TenancyForm
- from tenancy.models import Tenant
- from utilities.forms import (
- add_blank_choice, BootstrapMixin, BulkEditNullBooleanSelect, BulkRenameForm, CommentField, ConfirmationForm,
- CSVChoiceField, CSVModelChoiceField, DynamicModelChoiceField, DynamicModelMultipleChoiceField, ExpandableNameField,
- form_from_model, JSONField, SlugField, SmallTextarea, StaticSelect2, StaticSelect2Multiple, TagFilterField,
- BOOLEAN_WITH_BLANK_CHOICES,
- )
- from .choices import *
- from .models import Cluster, ClusterGroup, ClusterType, VirtualMachine, VMInterface
- #
- # Cluster types
- #
- class ClusterTypeForm(BootstrapMixin, CustomFieldModelForm):
- slug = SlugField()
- class Meta:
- model = ClusterType
- fields = [
- 'name', 'slug', 'description',
- ]
- class ClusterTypeCSVForm(CustomFieldModelCSVForm):
- slug = SlugField()
- class Meta:
- model = ClusterType
- fields = ClusterType.csv_headers
- class ClusterTypeBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=ClusterType.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- description = forms.CharField(
- max_length=200,
- required=False
- )
- class Meta:
- nullable_fields = ['description']
- #
- # Cluster groups
- #
- class ClusterGroupForm(BootstrapMixin, CustomFieldModelForm):
- slug = SlugField()
- class Meta:
- model = ClusterGroup
- fields = [
- 'name', 'slug', 'description',
- ]
- class ClusterGroupCSVForm(CustomFieldModelCSVForm):
- slug = SlugField()
- class Meta:
- model = ClusterGroup
- fields = ClusterGroup.csv_headers
- class ClusterGroupBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=ClusterGroup.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- description = forms.CharField(
- max_length=200,
- required=False
- )
- class Meta:
- nullable_fields = ['description']
- #
- # Clusters
- #
- class ClusterForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
- type = DynamicModelChoiceField(
- queryset=ClusterType.objects.all()
- )
- group = DynamicModelChoiceField(
- queryset=ClusterGroup.objects.all(),
- required=False
- )
- region = DynamicModelChoiceField(
- queryset=Region.objects.all(),
- required=False,
- initial_params={
- 'sites': '$site'
- }
- )
- site_group = DynamicModelChoiceField(
- queryset=SiteGroup.objects.all(),
- required=False,
- initial_params={
- 'sites': '$site'
- }
- )
- site = DynamicModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- query_params={
- 'region_id': '$region',
- 'group_id': '$site_group',
- }
- )
- comments = CommentField()
- tags = DynamicModelMultipleChoiceField(
- queryset=Tag.objects.all(),
- required=False
- )
- class Meta:
- model = Cluster
- fields = (
- 'name', 'type', 'group', 'tenant', 'region', 'site_group', 'site', 'comments', 'tags',
- )
- fieldsets = (
- ('Cluster', ('name', 'type', 'group', 'region', 'site_group', 'site', 'tags')),
- ('Tenancy', ('tenant_group', 'tenant')),
- )
- class ClusterCSVForm(CustomFieldModelCSVForm):
- type = CSVModelChoiceField(
- queryset=ClusterType.objects.all(),
- to_field_name='name',
- help_text='Type of cluster'
- )
- group = CSVModelChoiceField(
- queryset=ClusterGroup.objects.all(),
- to_field_name='name',
- required=False,
- help_text='Assigned cluster group'
- )
- site = CSVModelChoiceField(
- queryset=Site.objects.all(),
- to_field_name='name',
- required=False,
- help_text='Assigned site'
- )
- tenant = CSVModelChoiceField(
- queryset=Tenant.objects.all(),
- to_field_name='name',
- required=False,
- help_text='Assigned tenant'
- )
- class Meta:
- model = Cluster
- fields = Cluster.csv_headers
- class ClusterBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Cluster.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- type = DynamicModelChoiceField(
- queryset=ClusterType.objects.all(),
- required=False
- )
- group = DynamicModelChoiceField(
- queryset=ClusterGroup.objects.all(),
- required=False
- )
- tenant = DynamicModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False
- )
- region = DynamicModelChoiceField(
- queryset=Region.objects.all(),
- required=False,
- )
- site_group = DynamicModelChoiceField(
- queryset=SiteGroup.objects.all(),
- required=False,
- )
- site = DynamicModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- query_params={
- 'region_id': '$region',
- 'group_id': '$site_group',
- }
- )
- comments = CommentField(
- widget=SmallTextarea,
- label='Comments'
- )
- class Meta:
- nullable_fields = [
- 'group', 'site', 'comments', 'tenant',
- ]
- class ClusterFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
- model = Cluster
- field_order = [
- 'q', 'type_id', 'region_id', 'site_id', 'group_id', 'tenant_group_id', 'tenant_id',
- ]
- q = forms.CharField(
- required=False,
- label=_('Search')
- )
- type_id = DynamicModelMultipleChoiceField(
- queryset=ClusterType.objects.all(),
- required=False,
- label=_('Type')
- )
- region_id = DynamicModelMultipleChoiceField(
- queryset=Region.objects.all(),
- required=False,
- label=_('Region')
- )
- site_id = DynamicModelMultipleChoiceField(
- queryset=Site.objects.all(),
- required=False,
- null_option='None',
- query_params={
- 'region_id': '$region_id'
- },
- label=_('Site')
- )
- group_id = DynamicModelMultipleChoiceField(
- queryset=ClusterGroup.objects.all(),
- required=False,
- null_option='None',
- label=_('Group')
- )
- tag = TagFilterField(model)
- class ClusterAddDevicesForm(BootstrapMixin, forms.Form):
- region = DynamicModelChoiceField(
- queryset=Region.objects.all(),
- required=False,
- null_option='None'
- )
- site_group = DynamicModelChoiceField(
- queryset=SiteGroup.objects.all(),
- required=False,
- null_option='None'
- )
- site = DynamicModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- query_params={
- 'region_id': '$region',
- 'group_id': '$site_group',
- }
- )
- rack = DynamicModelChoiceField(
- queryset=Rack.objects.all(),
- required=False,
- null_option='None',
- query_params={
- 'site_id': '$site'
- }
- )
- devices = DynamicModelMultipleChoiceField(
- queryset=Device.objects.all(),
- query_params={
- 'site_id': '$site',
- 'rack_id': '$rack',
- 'cluster_id': 'null',
- }
- )
- class Meta:
- fields = [
- 'region', 'site', 'rack', 'devices',
- ]
- def __init__(self, cluster, *args, **kwargs):
- self.cluster = cluster
- super().__init__(*args, **kwargs)
- self.fields['devices'].choices = []
- def clean(self):
- super().clean()
- # If the Cluster is assigned to a Site, all Devices must be assigned to that Site.
- if self.cluster.site is not None:
- for device in self.cleaned_data.get('devices', []):
- if device.site != self.cluster.site:
- raise ValidationError({
- 'devices': "{} belongs to a different site ({}) than the cluster ({})".format(
- device, device.site, self.cluster.site
- )
- })
- class ClusterRemoveDevicesForm(ConfirmationForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Device.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- #
- # Virtual Machines
- #
- class VirtualMachineForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
- cluster_group = DynamicModelChoiceField(
- queryset=ClusterGroup.objects.all(),
- required=False,
- null_option='None',
- initial_params={
- 'clusters': '$cluster'
- }
- )
- cluster = DynamicModelChoiceField(
- queryset=Cluster.objects.all(),
- query_params={
- 'group_id': '$cluster_group'
- }
- )
- role = DynamicModelChoiceField(
- queryset=DeviceRole.objects.all(),
- required=False,
- query_params={
- "vm_role": "True"
- }
- )
- platform = DynamicModelChoiceField(
- queryset=Platform.objects.all(),
- required=False
- )
- local_context_data = JSONField(
- required=False,
- label=''
- )
- tags = DynamicModelMultipleChoiceField(
- queryset=Tag.objects.all(),
- required=False
- )
- class Meta:
- model = VirtualMachine
- fields = [
- 'name', 'status', 'cluster_group', 'cluster', 'role', 'tenant_group', 'tenant', 'platform', 'primary_ip4',
- 'primary_ip6', 'vcpus', 'memory', 'disk', 'comments', 'tags', 'local_context_data',
- ]
- fieldsets = (
- ('Virtual Machine', ('name', 'role', 'status', 'tags')),
- ('Cluster', ('cluster_group', 'cluster')),
- ('Tenancy', ('tenant_group', 'tenant')),
- ('Management', ('platform', 'primary_ip4', 'primary_ip6')),
- ('Resources', ('vcpus', 'memory', 'disk')),
- ('Config Context', ('local_context_data',)),
- )
- help_texts = {
- 'local_context_data': "Local config context data overwrites all sources contexts in the final rendered "
- "config context",
- }
- widgets = {
- "status": StaticSelect2(),
- 'primary_ip4': StaticSelect2(),
- 'primary_ip6': StaticSelect2(),
- }
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- if self.instance.pk:
- # Compile list of choices for primary IPv4 and IPv6 addresses
- for family in [4, 6]:
- ip_choices = [(None, '---------')]
- # Gather PKs of all interfaces belonging to this VM
- interface_ids = self.instance.interfaces.values_list('pk', flat=True)
- # Collect interface IPs
- interface_ips = IPAddress.objects.filter(
- address__family=family,
- assigned_object_type=ContentType.objects.get_for_model(VMInterface),
- assigned_object_id__in=interface_ids
- )
- if interface_ips:
- ip_list = [(ip.id, f'{ip.address} ({ip.assigned_object})') for ip in interface_ips]
- ip_choices.append(('Interface IPs', ip_list))
- # Collect NAT IPs
- nat_ips = IPAddress.objects.prefetch_related('nat_inside').filter(
- address__family=family,
- nat_inside__assigned_object_type=ContentType.objects.get_for_model(VMInterface),
- nat_inside__assigned_object_id__in=interface_ids
- )
- if nat_ips:
- ip_list = [(ip.id, f'{ip.address} (NAT)') for ip in nat_ips]
- ip_choices.append(('NAT IPs', ip_list))
- self.fields['primary_ip{}'.format(family)].choices = ip_choices
- else:
- # An object that doesn't exist yet can't have any IPs assigned to it
- self.fields['primary_ip4'].choices = []
- self.fields['primary_ip4'].widget.attrs['readonly'] = True
- self.fields['primary_ip6'].choices = []
- self.fields['primary_ip6'].widget.attrs['readonly'] = True
- class VirtualMachineCSVForm(CustomFieldModelCSVForm):
- status = CSVChoiceField(
- choices=VirtualMachineStatusChoices,
- required=False,
- help_text='Operational status of device'
- )
- cluster = CSVModelChoiceField(
- queryset=Cluster.objects.all(),
- to_field_name='name',
- help_text='Assigned cluster'
- )
- role = CSVModelChoiceField(
- queryset=DeviceRole.objects.filter(
- vm_role=True
- ),
- required=False,
- to_field_name='name',
- help_text='Functional role'
- )
- tenant = CSVModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Assigned tenant'
- )
- platform = CSVModelChoiceField(
- queryset=Platform.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Assigned platform'
- )
- class Meta:
- model = VirtualMachine
- fields = VirtualMachine.csv_headers
- class VirtualMachineBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=VirtualMachine.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- status = forms.ChoiceField(
- choices=add_blank_choice(VirtualMachineStatusChoices),
- required=False,
- initial='',
- widget=StaticSelect2(),
- )
- cluster = DynamicModelChoiceField(
- queryset=Cluster.objects.all(),
- required=False
- )
- role = DynamicModelChoiceField(
- queryset=DeviceRole.objects.filter(
- vm_role=True
- ),
- required=False,
- query_params={
- "vm_role": "True"
- }
- )
- tenant = DynamicModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False
- )
- platform = DynamicModelChoiceField(
- queryset=Platform.objects.all(),
- required=False
- )
- vcpus = forms.IntegerField(
- required=False,
- label='vCPUs'
- )
- memory = forms.IntegerField(
- required=False,
- label='Memory (MB)'
- )
- disk = forms.IntegerField(
- required=False,
- label='Disk (GB)'
- )
- comments = CommentField(
- widget=SmallTextarea,
- label='Comments'
- )
- class Meta:
- nullable_fields = [
- 'role', 'tenant', 'platform', 'vcpus', 'memory', 'disk', 'comments',
- ]
- class VirtualMachineFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
- model = VirtualMachine
- field_order = [
- 'q', 'cluster_group_id', 'cluster_type_id', 'cluster_id', 'status', 'role_id', 'region_id', 'site_group_id',
- 'site_id', 'tenant_group_id', 'tenant_id', 'platform_id', 'mac_address',
- ]
- q = forms.CharField(
- required=False,
- label=_('Search')
- )
- cluster_group_id = DynamicModelMultipleChoiceField(
- queryset=ClusterGroup.objects.all(),
- required=False,
- null_option='None',
- label=_('Cluster group')
- )
- cluster_type_id = DynamicModelMultipleChoiceField(
- queryset=ClusterType.objects.all(),
- required=False,
- null_option='None',
- label=_('Cluster type')
- )
- cluster_id = DynamicModelMultipleChoiceField(
- queryset=Cluster.objects.all(),
- required=False,
- label=_('Cluster')
- )
- region_id = DynamicModelMultipleChoiceField(
- queryset=Region.objects.all(),
- required=False,
- label=_('Region')
- )
- site_group_id = DynamicModelMultipleChoiceField(
- queryset=SiteGroup.objects.all(),
- required=False,
- label=_('Site group')
- )
- site_id = DynamicModelMultipleChoiceField(
- queryset=Site.objects.all(),
- required=False,
- null_option='None',
- query_params={
- 'region_id': '$region_id',
- 'group_id': '$site_group_id',
- },
- label=_('Site')
- )
- role_id = DynamicModelMultipleChoiceField(
- queryset=DeviceRole.objects.all(),
- required=False,
- null_option='None',
- query_params={
- 'vm_role': "True"
- },
- label=_('Role')
- )
- status = forms.MultipleChoiceField(
- choices=VirtualMachineStatusChoices,
- required=False,
- widget=StaticSelect2Multiple()
- )
- platform_id = DynamicModelMultipleChoiceField(
- queryset=Platform.objects.all(),
- required=False,
- null_option='None',
- label=_('Platform')
- )
- mac_address = forms.CharField(
- required=False,
- label='MAC address'
- )
- has_primary_ip = forms.NullBooleanField(
- required=False,
- label='Has a primary IP',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- tag = TagFilterField(model)
- #
- # VM interfaces
- #
- class VMInterfaceForm(BootstrapMixin, InterfaceCommonForm, CustomFieldModelForm):
- parent = DynamicModelChoiceField(
- queryset=VMInterface.objects.all(),
- required=False,
- label='Parent interface'
- )
- vlan_group = DynamicModelChoiceField(
- queryset=VLANGroup.objects.all(),
- required=False,
- label='VLAN group'
- )
- untagged_vlan = DynamicModelChoiceField(
- queryset=VLAN.objects.all(),
- required=False,
- label='Untagged VLAN',
- query_params={
- 'group_id': '$vlan_group',
- }
- )
- tagged_vlans = DynamicModelMultipleChoiceField(
- queryset=VLAN.objects.all(),
- required=False,
- label='Tagged VLANs',
- query_params={
- 'group_id': '$vlan_group',
- }
- )
- tags = DynamicModelMultipleChoiceField(
- queryset=Tag.objects.all(),
- required=False
- )
- class Meta:
- model = VMInterface
- fields = [
- 'virtual_machine', 'name', 'enabled', 'parent', 'mac_address', 'mtu', 'description', 'mode', 'tags',
- 'untagged_vlan', 'tagged_vlans',
- ]
- widgets = {
- 'virtual_machine': forms.HiddenInput(),
- 'mode': StaticSelect2()
- }
- labels = {
- 'mode': '802.1Q Mode',
- }
- help_texts = {
- 'mode': INTERFACE_MODE_HELP_TEXT,
- }
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- vm_id = self.initial.get('virtual_machine') or self.data.get('virtual_machine')
- # Restrict parent interface assignment by VM
- self.fields['parent'].widget.add_query_param('virtual_machine_id', vm_id)
- # Limit VLAN choices by virtual machine
- self.fields['untagged_vlan'].widget.add_query_param('available_on_virtualmachine', vm_id)
- self.fields['tagged_vlans'].widget.add_query_param('available_on_virtualmachine', vm_id)
- class VMInterfaceCreateForm(BootstrapMixin, CustomFieldForm, InterfaceCommonForm):
- model = VMInterface
- virtual_machine = DynamicModelChoiceField(
- queryset=VirtualMachine.objects.all()
- )
- name_pattern = ExpandableNameField(
- label='Name'
- )
- enabled = forms.BooleanField(
- required=False,
- initial=True
- )
- parent = DynamicModelChoiceField(
- queryset=VMInterface.objects.all(),
- required=False,
- display_field='display_name',
- query_params={
- 'virtual_machine_id': '$virtual_machine',
- }
- )
- mac_address = forms.CharField(
- required=False,
- label='MAC Address'
- )
- description = forms.CharField(
- max_length=200,
- required=False
- )
- mode = forms.ChoiceField(
- choices=add_blank_choice(InterfaceModeChoices),
- required=False,
- widget=StaticSelect2(),
- )
- untagged_vlan = DynamicModelChoiceField(
- queryset=VLAN.objects.all(),
- required=False
- )
- tagged_vlans = DynamicModelMultipleChoiceField(
- queryset=VLAN.objects.all(),
- required=False
- )
- tags = DynamicModelMultipleChoiceField(
- queryset=Tag.objects.all(),
- required=False
- )
- field_order = (
- 'virtual_machine', 'name_pattern', 'enabled', 'parent', 'mtu', 'mac_address', 'description', 'mode',
- 'untagged_vlan', 'tagged_vlans', 'tags'
- )
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- vm_id = self.initial.get('virtual_machine') or self.data.get('virtual_machine')
- # Limit VLAN choices by virtual machine
- self.fields['untagged_vlan'].widget.add_query_param('available_on_virtualmachine', vm_id)
- self.fields['tagged_vlans'].widget.add_query_param('available_on_virtualmachine', vm_id)
- class VMInterfaceCSVForm(CustomFieldModelCSVForm):
- virtual_machine = CSVModelChoiceField(
- queryset=VirtualMachine.objects.all(),
- to_field_name='name'
- )
- mode = CSVChoiceField(
- choices=InterfaceModeChoices,
- required=False,
- help_text='IEEE 802.1Q operational mode (for L2 interfaces)'
- )
- class Meta:
- model = VMInterface
- fields = VMInterface.csv_headers
- def clean_enabled(self):
- # Make sure enabled is True when it's not included in the uploaded data
- if 'enabled' not in self.data:
- return True
- else:
- return self.cleaned_data['enabled']
- class VMInterfaceBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=VMInterface.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- virtual_machine = forms.ModelChoiceField(
- queryset=VirtualMachine.objects.all(),
- required=False,
- disabled=True,
- widget=forms.HiddenInput()
- )
- parent = DynamicModelChoiceField(
- queryset=VMInterface.objects.all(),
- required=False,
- display_field='display_name'
- )
- enabled = forms.NullBooleanField(
- required=False,
- widget=BulkEditNullBooleanSelect()
- )
- mtu = forms.IntegerField(
- required=False,
- min_value=INTERFACE_MTU_MIN,
- max_value=INTERFACE_MTU_MAX,
- label='MTU'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- mode = forms.ChoiceField(
- choices=add_blank_choice(InterfaceModeChoices),
- required=False,
- widget=StaticSelect2()
- )
- untagged_vlan = DynamicModelChoiceField(
- queryset=VLAN.objects.all(),
- required=False
- )
- tagged_vlans = DynamicModelMultipleChoiceField(
- queryset=VLAN.objects.all(),
- required=False
- )
- class Meta:
- nullable_fields = [
- 'parent', 'mtu', 'description',
- ]
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- if 'virtual_machine' in self.initial:
- vm_id = self.initial.get('virtual_machine')
- # Restrict parent interface assignment by VM
- self.fields['parent'].widget.add_query_param('virtual_machine_id', vm_id)
- # Limit VLAN choices by virtual machine
- self.fields['untagged_vlan'].widget.add_query_param('available_on_virtualmachine', vm_id)
- self.fields['tagged_vlans'].widget.add_query_param('available_on_virtualmachine', vm_id)
- else:
- # See 5643
- if 'pk' in self.initial:
- site = None
- interfaces = VMInterface.objects.filter(pk__in=self.initial['pk']).prefetch_related(
- 'virtual_machine__cluster__site'
- )
- # Check interface sites. First interface should set site, further interfaces will either continue the
- # loop or reset back to no site and break the loop.
- for interface in interfaces:
- if site is None:
- site = interface.virtual_machine.cluster.site
- elif interface.virtual_machine.cluster.site is not site:
- site = None
- break
- if site is not None:
- self.fields['untagged_vlan'].widget.add_query_param('site_id', site.pk)
- self.fields['tagged_vlans'].widget.add_query_param('site_id', site.pk)
- class VMInterfaceBulkRenameForm(BulkRenameForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=VMInterface.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- class VMInterfaceFilterForm(BootstrapMixin, forms.Form):
- model = VMInterface
- q = forms.CharField(
- required=False,
- label=_('Search')
- )
- cluster_id = DynamicModelMultipleChoiceField(
- queryset=Cluster.objects.all(),
- required=False,
- label=_('Cluster')
- )
- virtual_machine_id = DynamicModelMultipleChoiceField(
- queryset=VirtualMachine.objects.all(),
- required=False,
- query_params={
- 'cluster_id': '$cluster_id'
- },
- label=_('Virtual machine')
- )
- enabled = forms.NullBooleanField(
- required=False,
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- mac_address = forms.CharField(
- required=False,
- label='MAC address'
- )
- tag = TagFilterField(model)
- #
- # Bulk VirtualMachine component creation
- #
- class VirtualMachineBulkAddComponentForm(BootstrapMixin, forms.Form):
- pk = forms.ModelMultipleChoiceField(
- queryset=VirtualMachine.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- name_pattern = ExpandableNameField(
- label='Name'
- )
- def clean_tags(self):
- # Because we're feeding TagField data (on the bulk edit form) to another TagField (on the model form), we
- # must first convert the list of tags to a string.
- return ','.join(self.cleaned_data.get('tags'))
- class VMInterfaceBulkCreateForm(
- form_from_model(VMInterface, ['enabled', 'mtu', 'description', 'tags']),
- VirtualMachineBulkAddComponentForm
- ):
- pass
|