-
Piotr Gawron authored
Feature/improve management of visits from subject list Closes #273 See merge request NCER-PD/scheduling-system!188
Piotr Gawron authoredFeature/improve management of visits from subject list Closes #273 See merge request NCER-PD/scheduling-system!188
Code owners
Assign users and groups as approvers for specific file changes. Learn more.
visit.py 5.43 KiB
# coding=utf-8
import datetime
from dateutil.relativedelta import relativedelta
from django.db import models
from django.db.models.signals import post_save
from django.dispatch import receiver
from django.db import transaction
from web.models.constants import BOOL_CHOICES, SUBJECT_TYPE_CHOICES_CONTROL
from web.models import Study
import logging
logger = logging.getLogger(__name__)
class Visit(models.Model):
class Meta:
app_label = 'web'
subject = models.ForeignKey("web.StudySubject", on_delete=models.CASCADE, verbose_name='Subject'
)
datetime_begin = models.DateTimeField(
verbose_name='Visit starts at'
)
datetime_end = models.DateTimeField(
verbose_name='Visit ends at'
) # Deadline before which all appointments need to be scheduled
is_finished = models.BooleanField(
verbose_name='Has ended',
default=False
)
post_mail_sent = models.BooleanField(choices=BOOL_CHOICES,
verbose_name='Post mail sent',
default=False
)
appointment_types = models.ManyToManyField("web.AppointmentType",
verbose_name='Requested appointments',
blank=True,
)
# this value is automatically computed by signal handled by
# update_visit_number method
visit_number = models.IntegerField(
verbose_name='Visit number',
default=1
)
def __unicode__(self):
return "%s %s" % (self.subject.subject.first_name, self.subject.subject.last_name)
def __str__(self):
return "%s %s" % (self.subject.subject.first_name, self.subject.subject.last_name)
def mark_as_finished(self):
self.is_finished = True
self.save()
create_follow_up = True
if self.subject.subject.dead:
create_follow_up = False
elif self.subject.resigned:
create_follow_up = False
elif self.subject.excluded:
create_follow_up = False
elif not self.subject.study.auto_create_follow_up:
create_follow_up = False
if create_follow_up:
visit_started = Visit.objects.filter(
subject=self.subject).filter(visit_number=1)[0].datetime_begin
follow_up_number = Visit.objects.filter(
subject=self.subject).count() + 1
study = self.subject.study
if self.subject.type == SUBJECT_TYPE_CHOICES_CONTROL:
args = {study.default_delta_time_for_follow_up_units: study.default_delta_time_for_control_follow_up}
else:
args = {study.default_delta_time_for_follow_up_units: study.default_delta_time_for_patient_follow_up}
time_to_next_visit = relativedelta(**args) * (follow_up_number - 1) #calculated from first visit
logger.warn('new visit: {} {} {}'.format(args, relativedelta(**args), time_to_next_visit))
Visit.objects.create(
subject=self.subject,
datetime_begin=visit_started + time_to_next_visit,
datetime_end=visit_started + time_to_next_visit + relativedelta(months=study.default_visit_duration_in_months)
)
@receiver(post_save, sender=Visit)
def check_visit_number(sender, instance, created, **kwargs):
# no other solution to ensure the visit_number is in cronological order than to sort the whole list if there are future visits
visit = instance
if visit.subject is not None: #not sure if select_for_update has an effect, the tests work as well without it
#new visit, sort only future visit respect to the new one
if created:
visits_before = Visit.objects.select_for_update().filter(subject=visit.subject).filter(datetime_begin__lt=visit.datetime_begin).count()
# we need to sort the future visits respect to the new one, if any
visits = Visit.objects.select_for_update().filter(subject=visit.subject).filter(datetime_begin__gte=visit.datetime_begin).order_by('datetime_begin','datetime_end')
with transaction.atomic(): #not sure if it has an effect, the tests work as well without it
for i, v in enumerate(visits):
expected_visit_number = (visits_before + i + 1)
if v.visit_number != expected_visit_number:
Visit.objects.filter(id=v.id).update(visit_number=expected_visit_number) # does not rise post_save, we avoid recursion
if v.id == visit.id: #if the iteration visit is the same that the instance that produced the signal call
#this ensures that the upper saved object is also updated, otherwise, refresh_from_db should be called
visit.visit_number = v.visit_number
else:
#if visits are modified, then, check everything
visits = Visit.objects.select_for_update().filter(subject=visit.subject).order_by('datetime_begin','datetime_end')
with transaction.atomic():
for i, v in enumerate(visits):
expected_visit_number = (i+1)
if v.visit_number != expected_visit_number: #update only those with wrong numbers
Visit.objects.filter(id=v.id).update(visit_number=expected_visit_number)