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

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

@ -524,8 +524,14 @@ class CustomField(CloningMixin, ExportTemplatesMixin, ChangeLoggedModel):
# Object # Object
elif self.type == CustomFieldTypeChoices.TYPE_OBJECT: elif self.type == CustomFieldTypeChoices.TYPE_OBJECT:
model = self.related_object_type.model_class() model = self.related_object_type.model_class()
field_class = CSVModelChoiceField if for_csv_import else DynamicModelChoiceField if for_csv_import:
field = field_class( field = CSVModelChoiceField(
queryset=model.objects.all(),
required=required,
initial=initial,
)
else:
field = DynamicModelChoiceField(
queryset=model.objects.all(), queryset=model.objects.all(),
required=required, required=required,
initial=initial, initial=initial,
@ -535,8 +541,14 @@ class CustomField(CloningMixin, ExportTemplatesMixin, ChangeLoggedModel):
# Multiple objects # Multiple objects
elif self.type == CustomFieldTypeChoices.TYPE_MULTIOBJECT: elif self.type == CustomFieldTypeChoices.TYPE_MULTIOBJECT:
model = self.related_object_type.model_class() model = self.related_object_type.model_class()
field_class = CSVModelMultipleChoiceField if for_csv_import else DynamicModelMultipleChoiceField if for_csv_import:
field = field_class( field = CSVModelMultipleChoiceField(
queryset=model.objects.all(),
required=required,
initial=initial,
)
else:
field = DynamicModelMultipleChoiceField(
queryset=model.objects.all(), queryset=model.objects.all(),
required=required, required=required,
initial=initial, initial=initial,