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

17354 fix import with custom-field

Browse Source
This commit is contained in:
Arthur Hanson 2024-09-04 09:56:38 -07:00
parent 886d635524
commit 3e51a5f9bf
1 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -524,24 +524,36 @@ class CustomField(CloningMixin, ExportTemplatesMixin, ChangeLoggedModel):
# Object
elif self.type == CustomFieldTypeChoices.TYPE_OBJECT:
model = self.related_object_type.model_class()
field_class = CSVModelChoiceField if for_csv_import else DynamicModelChoiceField
field = field_class(
queryset=model.objects.all(),
required=required,
initial=initial,
query_params=self.related_object_filter
)
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
)
# Multiple objects
elif self.type == CustomFieldTypeChoices.TYPE_MULTIOBJECT:
model = self.related_object_type.model_class()
field_class = CSVModelMultipleChoiceField if for_csv_import else DynamicModelMultipleChoiceField
field = field_class(
queryset=model.objects.all(),
required=required,
initial=initial,
query_params=self.related_object_filter
)
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
)
# Text
else: