NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-07 08:08:15 -06:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

17354 change to use kwargs

Browse Source
This commit is contained in:
Arthur Hanson 2024-09-04 13:00:31 -07:00
parent 3e51a5f9bf
commit 098d250552
1 changed files with 20 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
netbox/extras/models/customfields.py
View File

@ -524,36 +524,30 @@ class CustomField(CloningMixin, ExportTemplatesMixin, ChangeLoggedModel):
# Object
elif self.type == CustomFieldTypeChoices.TYPE_OBJECT:
model = self.related_object_type.model_class()
if for_csv_import:
field = CSVModelChoiceField(
queryset=model.objects.all(),
required=required,
initial=initial,
)
else:
field = DynamicModelChoiceField(
queryset=model.objects.all(),
required=required,
initial=initial,
query_params=self.related_object_filter
)
field_class = CSVModelChoiceField if for_csv_import else DynamicModelChoiceField
kwargs = {
'queryset': model.objects.all(),
'required': required,
'initial': initial,
}
if not for_csv_import:
kwargs['query_params'] = self.related_object_filter
field = field_class(**kwargs)
# Multiple objects
elif self.type == CustomFieldTypeChoices.TYPE_MULTIOBJECT:
model = self.related_object_type.model_class()
if for_csv_import:
field = CSVModelMultipleChoiceField(
queryset=model.objects.all(),
required=required,
initial=initial,
)
else:
field = DynamicModelMultipleChoiceField(
queryset=model.objects.all(),
required=required,
initial=initial,
query_params=self.related_object_filter
)
field_class = CSVModelMultipleChoiceField if for_csv_import else DynamicModelMultipleChoiceField
kwargs = {
'queryset': model.objects.all(),
'required': required,
'initial': initial,
}
if not for_csv_import:
kwargs['query_params'] = self.related_object_filter
field = field_class(**kwargs)
# Text
else: