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

11000 improve yaml import

Browse Source
This commit is contained in:
Arthur 2022-12-01 10:08:46 -08:00
parent d0e0c2ff8b
commit d6e3ad098d
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
netbox/utilities/forms/forms.py
View File

@ -217,13 +217,24 @@ class ImportForm(BootstrapMixin, forms.Form):
}) })
def _clean_yaml(self, data): def _clean_yaml(self, data):
records = []
try: try:
return yaml.load_all(data, Loader=yaml.SafeLoader) for data in yaml.load_all(data, Loader=yaml.SafeLoader):
if type(data) == list:
records.extend(data)
elif type(data) == dict:
records.append(data)
else:
raise forms.ValidationError({
self.data_field: "Invalid YAML data: data must be dictionaries or lists of dictionaries"
})
except yaml.error.YAMLError as err: except yaml.error.YAMLError as err:
raise forms.ValidationError({ raise forms.ValidationError({
self.data_field: f"Invalid YAML data: {err}" self.data_field: f"Invalid YAML data: {err}"
}) })
return records
class FilterForm(BootstrapMixin, forms.Form): class FilterForm(BootstrapMixin, forms.Form):
""" """