# coding=utf-8
from django.core.validators import RegexValidator
from django.db import models

from constants import BOOL_CHOICES, SUBJECT_TYPE_CHOICES
from . import Appointment, Location, Visit


class StudySubject(models.Model):
    class Meta:
        app_label = 'web'

    def finish_all_visits(self):
        visits = Visit.objects.filter(subject=self, is_finished=False)
        for visit in visits:
            visit.is_finished = True
            visit.save()

    def finish_all_appointments(self):
        appointments = Appointment.objects.filter(visit__subject=self, status=Appointment.APPOINTMENT_STATUS_SCHEDULED)
        for appointment in appointments:
            appointment.status = Appointment.APPOINTMENT_STATUS_CANCELLED
            appointment.save()

    def mark_as_resigned(self):
        self.resigned = True
        self.finish_all_visits()
        self.finish_all_appointments()

    subject = models.ForeignKey("web.Subject",
                                verbose_name='Subject',
                                editable=False,
                                null=False,
                                )

    study = models.ForeignKey("web.Study",
                              verbose_name='Study',
                              editable=False,
                              null=False,
                              )

    postponed = models.BooleanField(choices=BOOL_CHOICES,
                                    verbose_name='Postponed',
                                    default=False
                                    )
    datetime_contact_reminder = models.DateTimeField(
        null=True,
        blank=True,
        verbose_name='Please make a contact on',
    )
    type = models.CharField(max_length=1,
                            choices=SUBJECT_TYPE_CHOICES.items(),
                            verbose_name='Type'
                            )

    default_location = models.ForeignKey(Location,
                                         verbose_name='Default appointment location',
                                         )

    flying_team = models.ForeignKey("web.FlyingTeam",
                                    verbose_name='Default flying team location (if applicable)',
                                    null=True, blank=True
                                    )

    screening_number = models.CharField(max_length=50,
                                        unique=True,
                                        verbose_name='Screening number', blank=False, null=False
                                        )
    nd_number = models.CharField(max_length=25,
                                 blank=True,
                                 verbose_name='ND number',
                                 validators=[
                                     RegexValidator('^(ND[0-9]{4}|)$',
                                                    message="ND number should look as follows: NDxxxx")]
                                 )
    mpower_id = models.CharField(max_length=20,
                                 blank=True,
                                 verbose_name='MPower ID'
                                 )
    comments = models.TextField(max_length=2000,
                                blank=True,
                                verbose_name='Comments'
                                )
    date_added = models.DateField(verbose_name='Added on',
                                  auto_now=True
                                  )
    referral = models.CharField(max_length=128,
                                null=True,
                                blank=True,
                                verbose_name='Referred by'
                                )
    diagnosis = models.CharField(max_length=128,
                                 null=True,
                                 blank=True,
                                 verbose_name='Diagnosis'
                                 )
    year_of_diagnosis = models.IntegerField(
        null=True,
        blank=True,
        verbose_name='Year of diagnosis (YYYY)'
    )

    information_sent = models.BooleanField(
        verbose_name='Information sent',
        default=False
    )
    pd_in_family = models.NullBooleanField(
        verbose_name='PD in family',
        default=None,
    )
    resigned = models.BooleanField(
        verbose_name='Resigned',
        default=False,
        editable=True
    )
    resign_reason = models.TextField(max_length=2000,
                                     blank=True,
                                     verbose_name='Resign reason'
                                     )

    def __str__(self):
        return "%s %s" % (self.subject.first_name, self.subject.last_name)

    def __unicode__(self):
        return "%s %s" % (self.subject.first_name, self.subject.last_name)