mirror of
https://github.com/Southampton-RSG/breccia-mapper.git
synced 2026-03-03 11:27:09 +00:00
5dcfbb6052
BREAKING CHANGE: Change format of Relationship CSV export BREAKING CHANGE: Use 'id' for id field in CSV exports Add RelationshipAnswerSet CSV export
80 lines
1.8 KiB
Python
80 lines
1.8 KiB
Python
import typing
|
|
|
|
from rest_framework import serializers
|
|
|
|
from people import models
|
|
|
|
from . import base
|
|
|
|
|
|
class SimplePersonSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.Person
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
]
|
|
|
|
|
|
class PersonSerializer(base.FlattenedModelSerializer):
|
|
class Meta:
|
|
model = models.Person
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
'core_member',
|
|
'gender',
|
|
'age_group',
|
|
'nationality',
|
|
'country_of_residence',
|
|
]
|
|
|
|
|
|
class RelationshipSerializer(base.FlattenedModelSerializer):
|
|
source = SimplePersonSerializer()
|
|
target = SimplePersonSerializer()
|
|
|
|
class Meta:
|
|
model = models.Relationship
|
|
fields = [
|
|
'id',
|
|
'source',
|
|
'target',
|
|
]
|
|
|
|
|
|
class RelationshipAnswerSetSerializer(base.FlattenedModelSerializer):
|
|
relationship = RelationshipSerializer()
|
|
|
|
class Meta:
|
|
model = models.RelationshipAnswerSet
|
|
fields = [
|
|
'id',
|
|
'relationship',
|
|
'timestamp',
|
|
'replaced_timestamp',
|
|
]
|
|
|
|
@property
|
|
def column_headers(self) -> typing.List[str]:
|
|
headers = super().column_headers
|
|
|
|
# Add relationship questions to columns
|
|
for question in models.RelationshipQuestion.objects.all():
|
|
headers.append(question.slug.replace('-', '_'))
|
|
|
|
return headers
|
|
|
|
def to_representation(self, instance):
|
|
rep = super().to_representation(instance)
|
|
|
|
try:
|
|
# Add relationship question answers to data
|
|
for answer in instance.question_answers.all():
|
|
rep[answer.question.slug.replace('-', '_')] = answer.slug.replace('-', '_')
|
|
|
|
except AttributeError:
|
|
pass
|
|
|
|
return rep
|