mirror of
https://github.com/Southampton-RSG/breccia-mapper.git
synced 2026-03-03 11:27:09 +00:00
9b3b759254
Answers to the relationship questions have been moved from the relationship to another model RelationshipAnswerSet. A new answer set is created each time a user answers the relationship questions. Resolves #16 Resolves #17 Resolves #18
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
"""
|
|
Forms for creating / updating models belonging to the 'people' app.
|
|
"""
|
|
from django import forms
|
|
|
|
from django_select2.forms import Select2Widget, Select2MultipleWidget
|
|
|
|
from . import models
|
|
|
|
|
|
class PersonForm(forms.ModelForm):
|
|
"""
|
|
Form for creating / updating an instance of :class:`Person`.
|
|
"""
|
|
class Meta:
|
|
model = models.Person
|
|
exclude = [
|
|
'user',
|
|
'relationship_targets',
|
|
]
|
|
widgets = {
|
|
'nationality': Select2Widget(),
|
|
'country_of_residence': Select2Widget(),
|
|
'themes': Select2MultipleWidget(),
|
|
}
|
|
|
|
|
|
class RelationshipAnswerSetForm(forms.ModelForm):
|
|
"""
|
|
Form to allow users to describe a relationship.
|
|
|
|
Dynamic fields inspired by https://jacobian.org/2010/feb/28/dynamic-form-generation/
|
|
"""
|
|
class Meta:
|
|
model = models.RelationshipAnswerSet
|
|
fields = [
|
|
'relationship',
|
|
]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
for question in models.RelationshipQuestion.objects.all():
|
|
# Get choices from model and add default 'not selected' option
|
|
choices = question.choices + [['', '---------']]
|
|
|
|
field = forms.ChoiceField(label=question,
|
|
choices=choices)
|
|
self.fields['question_{}'.format(question.pk)] = field
|
|
|
|
def save(self, commit=True) -> models.RelationshipAnswerSet:
|
|
# Save Relationship model
|
|
self.instance = super().save(commit=commit)
|
|
|
|
if commit:
|
|
# Save answers to relationship questions
|
|
for key, value in self.cleaned_data.items():
|
|
if key.startswith('question_'):
|
|
question_id = key.replace('question_', '', 1)
|
|
answer = models.RelationshipQuestionChoice.objects.get(pk=value,
|
|
question__pk=question_id)
|
|
self.instance.question_answers.add(answer)
|
|
|
|
return self.instance
|