mirror of
https://github.com/Southampton-RSG/breccia-mapper.git
synced 2026-03-03 11:27:09 +00:00
88 lines
2.2 KiB
Python
88 lines
2.2 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 is excluded from exports
|
|
# See https://github.com/Southampton-RSG/breccia-mapper/issues/35
|
|
]
|
|
|
|
|
|
class PersonSerializer(base.FlattenedModelSerializer):
|
|
class Meta:
|
|
model = models.Person
|
|
fields = [
|
|
'id',
|
|
# Name is excluded from exports
|
|
# See https://github.com/Southampton-RSG/breccia-mapper/issues/35
|
|
'gender',
|
|
'age_group',
|
|
'nationality',
|
|
'country_of_residence',
|
|
'organisation',
|
|
'organisation_started_date',
|
|
]
|
|
|
|
|
|
class RelationshipSerializer(base.FlattenedModelSerializer):
|
|
source = SimplePersonSerializer()
|
|
target = SimplePersonSerializer()
|
|
|
|
class Meta:
|
|
model = models.Relationship
|
|
fields = [
|
|
'id',
|
|
'source',
|
|
'target',
|
|
]
|
|
|
|
|
|
def underscore(slug: str) -> str:
|
|
"""Replace hyphens with underscores in text."""
|
|
return slug.replace('-', '_')
|
|
|
|
|
|
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(underscore(question.slug))
|
|
|
|
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[underscore(answer.question.slug)] = underscore(answer.slug)
|
|
|
|
except AttributeError:
|
|
pass
|
|
|
|
return rep
|