RomanNum3ral
/
netbox
mirror of https://github.com/netbox-community/netbox.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use model's `get_custom_fields`

This commit is contained in:
Saria Hajjar 2020-01-23 20:26:21 +00:00
parent 8f86244b4f
commit bed08a7b07
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
netbox/utilities/views.py
View File

@ -93,12 +93,11 @@ class ObjectListView(View):
# Start with the column headers
headers = self.queryset.model.csv_headers.copy()
# Add custom field headers
content_type = ContentType.objects.get_for_model(self.queryset.model)
for custom_field in CustomField.objects.filter(obj_type=content_type):
headers.append(custom_field.name)
custom_fields.append(custom_field.name)
# Add custom field headers, if any
if hasattr(self.queryset.model, 'get_custom_fields'):
for custom_field in self.queryset.model().get_custom_fields():
headers.append(custom_field.name)
custom_fields.append(custom_field.name)
csv_data.append(','.join(headers))