NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-17 20:46:30 -06:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Add query_params attribute to DynamicModelChoiceMixin

Browse Source
This commit is contained in:
Jeremy Stretch 2020-08-12 09:28:24 -04:00
parent a831e525da
commit f3fb85933f
3 changed files with 70 additions and 94 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
netbox/dcim/forms.py
View File

@ -1746,12 +1746,10 @@ class DeviceForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
platform = DynamicModelChoiceField( platform = DynamicModelChoiceField(
queryset=Platform.objects.all(), queryset=Platform.objects.all(),
required=False, required=False,
widget=APISelect( query_params={
additional_query_params={
"manufacturer_id": "null" "manufacturer_id": "null"
} }
) )
)
cluster_group = DynamicModelChoiceField( cluster_group = DynamicModelChoiceField(
queryset=ClusterGroup.objects.all(), queryset=ClusterGroup.objects.all(),
required=False, required=False,
@ -2680,24 +2678,20 @@ class InterfaceForm(InterfaceCommonForm, BootstrapMixin, forms.ModelForm):
required=False, required=False,
label='Untagged VLAN', label='Untagged VLAN',
display_field='display_name', display_field='display_name',
widget=APISelect( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelect(full=True)
) )
tagged_vlans = DynamicModelMultipleChoiceField( tagged_vlans = DynamicModelMultipleChoiceField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
label='Tagged VLANs', label='Tagged VLANs',
display_field='display_name', display_field='display_name',
widget=APISelectMultiple( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelectMultiple(full=True)
) )
tags = DynamicModelMultipleChoiceField( tags = DynamicModelMultipleChoiceField(
queryset=Tag.objects.all(), queryset=Tag.objects.all(),
@ -2781,23 +2775,19 @@ class InterfaceCreateForm(ComponentCreateForm, InterfaceCommonForm):
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelect( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelect(full=True)
) )
tagged_vlans = DynamicModelMultipleChoiceField( tagged_vlans = DynamicModelMultipleChoiceField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelectMultiple( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelectMultiple(full=True)
) )
field_order = ( field_order = (
'device', 'name_pattern', 'label_pattern', 'type', 'enabled', 'lag', 'mtu', 'mac_address', 'description', 'device', 'name_pattern', 'label_pattern', 'type', 'enabled', 'lag', 'mtu', 'mac_address', 'description',
@ -2850,23 +2840,19 @@ class InterfaceBulkEditForm(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelect( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelect(full=True)
) )
tagged_vlans = DynamicModelMultipleChoiceField( tagged_vlans = DynamicModelMultipleChoiceField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelectMultiple( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelectMultiple(full=True)
) )
class Meta: class Meta:

10
netbox/utilities/forms/fields.py
View File

@ -248,8 +248,9 @@ class DynamicModelChoiceMixin:
filter = django_filters.ModelChoiceFilter filter = django_filters.ModelChoiceFilter
widget = widgets.APISelect widget = widgets.APISelect
def __init__(self, *args, display_field='name', **kwargs): def __init__(self, *args, display_field='name', query_params=None, **kwargs):
self.display_field = display_field self.display_field = display_field
self.query_params = query_params or {}
# to_field_name is set by ModelChoiceField.__init__(), but we need to set it early for reference # to_field_name is set by ModelChoiceField.__init__(), but we need to set it early for reference
# by widget_attrs() # by widget_attrs()
@ -261,8 +262,15 @@ class DynamicModelChoiceMixin:
attrs = { attrs = {
'display-field': self.display_field, 'display-field': self.display_field,
} }
# Set value-field attribute if the field specifies to_field_name
if self.to_field_name: if self.to_field_name:
attrs['value-field'] = self.to_field_name attrs['value-field'] = self.to_field_name
# Attach any static query parameters
for key, value in self.query_params.items():
widget.add_additional_query_param(key, value)
return attrs return attrs
def get_bound_field(self, form, field_name): def get_bound_field(self, form, field_name):

52
netbox/virtualization/forms.py
View File

@ -297,12 +297,10 @@ class VirtualMachineForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
role = DynamicModelChoiceField( role = DynamicModelChoiceField(
queryset=DeviceRole.objects.all(), queryset=DeviceRole.objects.all(),
required=False, required=False,
widget=APISelect( query_params={
additional_query_params={
"vm_role": "True" "vm_role": "True"
} }
) )
)
platform = DynamicModelChoiceField( platform = DynamicModelChoiceField(
queryset=Platform.objects.all(), queryset=Platform.objects.all(),
required=False required=False
@ -438,12 +436,10 @@ class VirtualMachineBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldB
vm_role=True vm_role=True
), ),
required=False, required=False,
widget=APISelect( query_params={
additional_query_params={
"vm_role": "True" "vm_role": "True"
} }
) )
)
tenant = DynamicModelChoiceField( tenant = DynamicModelChoiceField(
queryset=Tenant.objects.all(), queryset=Tenant.objects.all(),
required=False required=False
@ -528,12 +524,10 @@ class VirtualMachineFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFil
queryset=DeviceRole.objects.filter(vm_role=True), queryset=DeviceRole.objects.filter(vm_role=True),
to_field_name='slug', to_field_name='slug',
required=False, required=False,
widget=APISelectMultiple( query_params={
null_option=True,
additional_query_params={
'vm_role': "True" 'vm_role': "True"
} },
) widget=APISelectMultiple(null_option=True)
) )
status = forms.MultipleChoiceField( status = forms.MultipleChoiceField(
choices=VirtualMachineStatusChoices, choices=VirtualMachineStatusChoices,
@ -564,23 +558,19 @@ class VMInterfaceForm(BootstrapMixin, forms.ModelForm):
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelect( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelect(full=True)
) )
tagged_vlans = DynamicModelMultipleChoiceField( tagged_vlans = DynamicModelMultipleChoiceField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelectMultiple( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelectMultiple(full=True)
) )
tags = DynamicModelMultipleChoiceField( tags = DynamicModelMultipleChoiceField(
queryset=Tag.objects.all(), queryset=Tag.objects.all(),
@ -668,23 +658,19 @@ class VMInterfaceCreateForm(BootstrapMixin, forms.Form):
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelect( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelect(full=True)
) )
tagged_vlans = DynamicModelMultipleChoiceField( tagged_vlans = DynamicModelMultipleChoiceField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelectMultiple( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelectMultiple(full=True)
) )
tags = DynamicModelMultipleChoiceField( tags = DynamicModelMultipleChoiceField(
queryset=Tag.objects.all(), queryset=Tag.objects.all(),
@ -762,23 +748,19 @@ class VMInterfaceBulkEditForm(BootstrapMixin, BulkEditForm):
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelect( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelect(full=True)
) )
tagged_vlans = DynamicModelMultipleChoiceField( tagged_vlans = DynamicModelMultipleChoiceField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
required=False, required=False,
display_field='display_name', display_field='display_name',
widget=APISelectMultiple( query_params={
full=True,
additional_query_params={
'site_id': 'null', 'site_id': 'null',
}, },
) widget=APISelectMultiple(full=True)
) )
class Meta: class Meta: