fix: Use underscores in export column headers

Resolves #30
This commit is contained in:
James Graham
2020-04-02 15:42:48 +01:00
parent 6418c8c96a
commit 8f494e1be5

View File

@@ -48,7 +48,7 @@ class RelationshipSerializer(base.FlattenedModelSerializer):
# Add relationship questions to columns # Add relationship questions to columns
for question in models.RelationshipQuestion.objects.all(): for question in models.RelationshipQuestion.objects.all():
headers.append(question.slug) headers.append(question.slug.replace('-', '_'))
return headers return headers
@@ -58,7 +58,7 @@ class RelationshipSerializer(base.FlattenedModelSerializer):
try: try:
# Add relationship question answers to data # Add relationship question answers to data
for answer in instance.current_answers.question_answers.all(): for answer in instance.current_answers.question_answers.all():
rep[answer.question.slug] = answer.slug rep[answer.question.slug.replace('-', '_')] = answer.slug.replace('-', '_')
except AttributeError: except AttributeError:
pass pass