mirror of
https://github.com/Southampton-RSG/breccia-mapper.git
synced 2026-03-03 19:37:06 +00:00
112 lines
2.7 KiB
Python
112 lines
2.7 KiB
Python
import typing
|
|
|
|
from people import models
|
|
|
|
from . import base
|
|
|
|
|
|
class PersonSerializer(base.FlattenedModelSerializer):
|
|
class Meta:
|
|
model = models.Person
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
]
|
|
|
|
|
|
class PersonAnswerSetSerializer(base.FlattenedModelSerializer):
|
|
person = PersonSerializer()
|
|
|
|
class Meta:
|
|
model = models.PersonAnswerSet
|
|
fields = [
|
|
'id',
|
|
'person',
|
|
'timestamp',
|
|
'replaced_timestamp',
|
|
'nationality',
|
|
'country_of_residence',
|
|
'organisation',
|
|
'organisation_started_date',
|
|
'job_title',
|
|
'latitude',
|
|
'longitude',
|
|
]
|
|
|
|
@property
|
|
def column_headers(self) -> typing.List[str]:
|
|
headers = super().column_headers
|
|
|
|
# Add questions to columns
|
|
for question in models.PersonQuestion.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
|
|
|
|
|
|
class RelationshipSerializer(base.FlattenedModelSerializer):
|
|
source = PersonSerializer()
|
|
target = PersonSerializer()
|
|
|
|
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
|