mirror of
https://github.com/Southampton-RSG/breccia-mapper.git
synced 2026-03-03 03:17:07 +00:00
277 lines
8.5 KiB
Python
277 lines
8.5 KiB
Python
"""
|
|
Views for displaying or manipulating models in the 'people' app.
|
|
"""
|
|
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.db.models import Q
|
|
from django.urls import reverse
|
|
from django.utils import timezone
|
|
from django.views.generic import CreateView, DetailView, FormView, ListView, UpdateView
|
|
|
|
from rest_framework.views import APIView, Response
|
|
|
|
from . import forms, models, permissions, serializers
|
|
|
|
|
|
class PersonCreateView(CreateView):
|
|
"""
|
|
View to create a new instance of :class:`Person`.
|
|
|
|
If 'user' is passed as a URL parameter - link the new person to the current user.
|
|
"""
|
|
model = models.Person
|
|
template_name = 'people/person/create.html'
|
|
form_class = forms.PersonForm
|
|
|
|
def form_valid(self, form):
|
|
if 'user' in self.request.GET:
|
|
form.instance.user = self.request.user
|
|
|
|
return super().form_valid(form)
|
|
|
|
|
|
class PersonListView(ListView):
|
|
"""
|
|
View displaying a list of :class:`Person` objects - searchable.
|
|
"""
|
|
model = models.Person
|
|
template_name = 'people/person/list.html'
|
|
|
|
|
|
class ProfileView(permissions.UserIsLinkedPersonMixin, DetailView):
|
|
"""
|
|
View displaying the profile of a :class:`Person` - who may be a user.
|
|
"""
|
|
model = models.Person
|
|
template_name = 'people/person/detail.html'
|
|
|
|
def get_object(self, queryset=None) -> models.Person:
|
|
"""
|
|
Get the :class:`Person` object to be represented by this page.
|
|
|
|
If not determined from url get current user.
|
|
"""
|
|
try:
|
|
return super().get_object(queryset)
|
|
|
|
except AttributeError:
|
|
# pk was not provided in URL
|
|
return self.request.user.person
|
|
|
|
|
|
class PersonUpdateView(permissions.UserIsLinkedPersonMixin, UpdateView):
|
|
"""
|
|
View for updating a :class:`Person` record.
|
|
"""
|
|
model = models.Person
|
|
template_name = 'people/person/update.html'
|
|
form_class = forms.PersonForm
|
|
|
|
|
|
class RelationshipDetailView(permissions.UserIsLinkedPersonMixin, DetailView):
|
|
"""
|
|
View displaying details of a :class:`Relationship`.
|
|
"""
|
|
model = models.Relationship
|
|
template_name = 'people/relationship/detail.html'
|
|
related_person_field = 'source'
|
|
|
|
|
|
class RelationshipCreateView(permissions.UserIsLinkedPersonMixin, CreateView):
|
|
"""
|
|
View for creating a :class:`Relationship`.
|
|
|
|
Displays / processes a form containing the :class:`RelationshipQuestion`s.
|
|
"""
|
|
model = models.Relationship
|
|
template_name = 'people/relationship/create.html'
|
|
fields = [
|
|
'source',
|
|
'target',
|
|
]
|
|
|
|
def get_test_person(self) -> models.Person:
|
|
"""
|
|
Get the person instance which should be used for access control checks.
|
|
"""
|
|
if self.request.method == 'POST':
|
|
return models.Person.objects.get(pk=self.request.POST.get('source'))
|
|
|
|
return models.Person.objects.get(pk=self.kwargs.get('person_pk'))
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.person = models.Person.objects.get(pk=self.kwargs.get('person_pk'))
|
|
|
|
return super().get(request, *args, **kwargs)
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
self.person = models.Person.objects.get(pk=self.kwargs.get('person_pk'))
|
|
|
|
return super().post(request, *args, **kwargs)
|
|
|
|
def get_initial(self):
|
|
initial = super().get_initial()
|
|
|
|
initial['source'] = self.request.user.person
|
|
initial['target'] = self.person
|
|
|
|
return initial
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
context['person'] = self.person
|
|
|
|
return context
|
|
|
|
def get_success_url(self):
|
|
return reverse('people:relationship.update', kwargs={'pk': self.object.pk})
|
|
|
|
|
|
class RelationshipUpdateView(permissions.UserIsLinkedPersonMixin, CreateView):
|
|
"""
|
|
View for creating a :class:`Relationship`.
|
|
|
|
Displays / processes a form containing the :class:`RelationshipQuestion`s.
|
|
"""
|
|
model = models.RelationshipAnswerSet
|
|
template_name = 'people/relationship/update.html'
|
|
form_class = forms.RelationshipAnswerSetForm
|
|
|
|
def get_test_person(self) -> models.Person:
|
|
"""
|
|
Get the person instance which should be used for access control checks.
|
|
"""
|
|
relationship = models.Relationship.objects.get(pk=self.kwargs.get('relationship_pk'))
|
|
|
|
return relationship.source
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.relationship = models.Relationship.objects.get(pk=self.kwargs.get('relationship_pk'))
|
|
self.person = self.relationship.source
|
|
|
|
return super().get(request, *args, **kwargs)
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
self.relationship = models.Relationship.objects.get(pk=self.kwargs.get('relationship_pk'))
|
|
self.person = self.relationship.source
|
|
|
|
return super().post(request, *args, **kwargs)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
context['person'] = self.person
|
|
context['relationship'] = self.relationship
|
|
|
|
return context
|
|
|
|
def get_initial(self):
|
|
initial = super().get_initial()
|
|
|
|
initial['relationship'] = self.relationship
|
|
|
|
return initial
|
|
|
|
def form_valid(self, form):
|
|
"""
|
|
Mark any previous answer sets as replaced.
|
|
"""
|
|
previous_valid_answer_sets = self.relationship.answer_sets.filter(replaced_timestamp__isnull=True)
|
|
|
|
response = super().form_valid(form)
|
|
|
|
# Shouldn't be more than one after initial updates after migration
|
|
for answer_set in previous_valid_answer_sets:
|
|
answer_set.replaced_timestamp = timezone.now()
|
|
answer_set.save()
|
|
|
|
return response
|
|
|
|
|
|
class PersonApiView(APIView):
|
|
"""
|
|
List all :class:`Person` instances.
|
|
"""
|
|
def get(self, request, format=None):
|
|
"""
|
|
List all :class:`Person` instances.
|
|
"""
|
|
serializer = serializers.PersonSerializer(models.Person.objects.all(),
|
|
many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
class RelationshipApiView(APIView):
|
|
"""
|
|
List all :class:`Relationship` instances.
|
|
"""
|
|
def get(self, request, format=None):
|
|
"""
|
|
List all :class:`Relationship` instances.
|
|
"""
|
|
serializer = serializers.RelationshipSerializer(models.Relationship.objects.all(),
|
|
many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
class NetworkView(LoginRequiredMixin, FormView):
|
|
"""
|
|
View to display relationship network.
|
|
"""
|
|
template_name = 'people/network.html'
|
|
form_class = forms.NetworkFilterForm
|
|
|
|
def get_form_kwargs(self):
|
|
"""
|
|
Add GET params to form data.
|
|
"""
|
|
kwargs = super().get_form_kwargs()
|
|
|
|
if self.request.method == 'GET':
|
|
if 'data' in kwargs:
|
|
kwargs['data'].update(self.request.GET)
|
|
|
|
else:
|
|
kwargs['data'] = self.request.GET
|
|
|
|
return kwargs
|
|
|
|
def get_context_data(self, **kwargs):
|
|
"""
|
|
Add filtered QuerySets of :class:`Person` and :class:`Relationship` to the context.
|
|
"""
|
|
context = super().get_context_data(**kwargs)
|
|
form = context['form']
|
|
|
|
at_time = timezone.now()
|
|
|
|
relationship_set = models.Relationship.objects.all()
|
|
|
|
# Filter answers to relationship questions
|
|
for key, value in form.data.items():
|
|
if key.startswith('question_') and value:
|
|
question_id = key.replace('question_', '', 1)
|
|
answer = models.RelationshipQuestionChoice.objects.get(pk=value,
|
|
question__pk=question_id)
|
|
relationship_set = relationship_set.filter(
|
|
Q(answer_sets__replaced_timestamp__gt=at_time) | Q(answer_sets__replaced_timestamp__isnull=True),
|
|
answer_sets__timestamp__lte=at_time,
|
|
answer_sets__question_answers=answer
|
|
)
|
|
|
|
context['person_set'] = serializers.PersonSerializer(
|
|
models.Person.objects.all(),
|
|
many=True
|
|
).data
|
|
|
|
context['relationship_set'] = serializers.RelationshipSerializer(
|
|
relationship_set,
|
|
many=True
|
|
).data
|
|
|
|
return context
|
|
|
|
def form_valid(self, form):
|
|
return self.render_to_response(self.get_context_data())
|