Closes #913: Added headers to object CSV exports

This commit is contained in:
Jeremy Stretch
2017-06-09 16:24:59 -04:00
parent 8a849ebeff
commit 08883d86ef
6 changed files with 44 additions and 2 deletions

View File

@@ -102,7 +102,9 @@ class ObjectListView(View):
.format(et.name))
# Fall back to built-in CSV export
elif 'export' in request.GET and hasattr(model, 'to_csv'):
output = '\n'.join([obj.to_csv() for obj in self.queryset])
headers = getattr(model, 'csv_headers', None)
output = ','.join(headers) + '\n' if headers else ''
output += '\n'.join([obj.to_csv() for obj in self.queryset])
response = HttpResponse(
output,
content_type='text/csv'