1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

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

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))