Produits Entr'ouvert » Calebasse

# -*- coding: utf-8 -*-

from datetime import datetime, timedelta

from django import forms

from ..dossiers.models import PatientRecord

from ..personnes.models import Worker

from ..ressources.models import ActType

from ..middleware.request import get_request

from ajax_select import make_ajax_field

from ajax_select.fields import AutoCompleteSelectField, AutoCompleteSelectMultipleField

from models import Event, EventWithAct, EventType

class BaseForm(forms.ModelForm):

    date = forms.DateField(label=u'Date', localize=True)

    time = forms.TimeField(label=u'Heure de début')

    duration = forms.CharField(label=u'Durée',

            help_text=u'en minutes; vous pouvez utiliser la roulette de votre souris.')

    participants = make_ajax_field(EventWithAct, 'participants', 'worker-or-group', True)

class NewAppointmentForm(BaseForm):

    patient = AutoCompleteSelectMultipleField('patientrecord')

    class Meta:

        model = EventWithAct

        fields = (

                'start_datetime',

                'date',

                'time',

                'duration',

                'patient',

                'participants',

                'ressource',

                'act_type',

                'description',

                'recurrence_periodicity',

                'recurrence_end_date'

        )

        widgets = {

                'start_datetime': forms.HiddenInput,

        }

    def __init__(self, instance, service=None, **kwargs):

        self.service = None

        super(NewAppointmentForm, self).__init__(instance=instance, **kwargs)

        self.fields['date'].css = 'datepicker'

        self.fields['participants'].required = True

        if service:

            self.service = service

            self.fields['participants'].queryset = \

                    Worker.objects.for_service(service)

            self.fields['patient'].queryset = \

                    PatientRecord.objects.for_service(service)

            self.fields['act_type'].queryset = \

                    ActType.objects.for_service(service) \

                    .order_by('name')

    def clean_duration(self):

        duration = self.cleaned_data['duration']

        try:

            return int(duration)

        except ValueError:

            return 0

    def clean_patient(self):

        patients = self.cleaned_data['patient']

        if patients:

            return [patient for patient in PatientRecord.objects.filter(pk__in=patients)]

    def clean(self):

        cleaned_data = super(NewAppointmentForm, self).clean()

        if not cleaned_data.get('recurrence_periodicity'):

            cleaned_data['recurrence_end_date'] = None

        if cleaned_data.has_key('date') and cleaned_data.has_key('time'):

            cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],

                    cleaned_data['time'])

        if 'patient' in cleaned_data and isinstance(cleaned_data['patient'], list):

            # nasty trick to store the list of patients and pass the form

            # validation

            cleaned_data['patients'] = cleaned_data['patient']

            cleaned_data['patient'] = cleaned_data['patient'][0]

        return cleaned_data

    def save(self, commit=True):

        patients = self.cleaned_data.pop('patients')

        for patient in patients:

            appointment = forms.save_instance(self, self._meta.model(), commit=False)

            appointment.start_datetime = datetime.combine(self.cleaned_data['date'],

                                                          self.cleaned_data['time'])

            appointment.end_datetime = appointment.start_datetime + timedelta(

                minutes=self.cleaned_data['duration'])

            appointment.creator = get_request().user

            appointment.clean()

            if commit:

                appointment.patient = patient

                appointment.save()

                self.save_m2m()

                appointment.services = [self.service]

class UpdateAppointmentForm(NewAppointmentForm):

    patient = make_ajax_field(EventWithAct, 'patient', 'patientrecord', False)

    def clean_patient(self):

        return self.cleaned_data['patient']

    def save(self, commit=True):

        appointment = super(NewAppointmentForm, self).save(commit=False)

        appointment.start_datetime = datetime.combine(self.cleaned_data['date'],

                    self.cleaned_data['time'])

        appointment.end_datetime = appointment.start_datetime + timedelta(

                minutes=self.cleaned_data['duration'])

        appointment.creator = get_request().user

        appointment.clean()

        if commit:

            appointment.save()

            appointment.services = [self.service]

        return appointment

class UpdatePeriodicAppointmentForm(UpdateAppointmentForm):

    patient = make_ajax_field(EventWithAct, 'patient', 'patientrecord', False)

    recurrence_periodicity = forms.ChoiceField(label=u"Périodicité",

            choices=Event.PERIODICITIES, required=True)

    def clean(self):

        cleaned_data = super(UpdatePeriodicAppointmentForm, self).clean()

        acts = self.instance.act_set.filter(is_billed=True).order_by('-date')

        if acts and cleaned_data.get('recurrence_end_date'):

            recurrence_end_date = cleaned_data['recurrence_end_date']

            if recurrence_end_date < acts[0].date:

                self._errors['recurrence_end_date'] = self.error_class([

                            u"La date doit être supérieur au dernier acte facturé de la récurrence"])

        return cleaned_data

class DisablePatientAppointmentForm(UpdateAppointmentForm):

    def __init__(self, instance, service=None, **kwargs):

        super(DisablePatientAppointmentForm, self).__init__(instance,

                service, **kwargs)

        if instance and instance.pk:

            self.fields['patient'].required = False

            self.fields['patient'].widget.attrs['disabled'] = 'disabled'

    def clean_patient(self):

        instance = getattr(self, 'instance', None)

        if instance:

            return instance.patient

        else:

            return self.cleaned_data.get('patient', None)

class NewEventForm(BaseForm):

    title = forms.CharField(label=u"Complément de l'intitulé", max_length=32, required=False)

    class Meta:

        model = Event

        fields = (

                'start_datetime',

                'title',

                'date',

                'time',

                'duration',

                'ressource',

                'participants',

                'event_type',

                'services',

                'description',

                'recurrence_periodicity',

                'recurrence_end_date',

        )

        widgets = {

                'start_datetime': forms.HiddenInput,

                'services': forms.CheckboxSelectMultiple,

        }

    def __init__(self, instance, service=None, **kwargs):

        self.service = service

        super(NewEventForm, self).__init__(instance=instance, **kwargs)

        self.fields['date'].css = 'datepicker'

        self.fields['event_type'].queryset = \

                    EventType.objects.exclude(id=1).exclude(id=3).order_by('rank', 'label')

    def clean_duration(self):

        duration = self.cleaned_data['duration']

        try:

            return int(duration)

        except:

            raise forms.ValidationError('Veuillez saisir un entier')

    def save(self, commit=True):

        event = super(NewEventForm, self).save(commit=False)

        event.start_datetime = datetime.combine(self.cleaned_data['date'],

                    self.cleaned_data['time'])

        event.end_datetime = event.start_datetime + timedelta(

                minutes=self.cleaned_data['duration'])

        event.creator = get_request().user

        event.clean()

        if commit:

            event.save()

            event.services = [self.service]

        return event

    def clean(self):

        cleaned_data = super(NewEventForm, self).clean()

        if cleaned_data.has_key('date') and cleaned_data.has_key('time'):

            cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],

                    cleaned_data['time'])

        if not cleaned_data.get('recurrence_periodicity'):

            cleaned_data['recurrence_end_date'] = None

        event_type = cleaned_data.get('event_type')

        if event_type and event_type.id == 4 and not cleaned_data.get('title'): # 'Autre'

            self._errors['title'] = self.error_class([

                            u"Ce champ est obligatoire pour les événements de type « Autre »."])

        return cleaned_data

class UpdatePeriodicEventForm(NewEventForm):

    recurrence_periodicity = forms.ChoiceField(label=u"Périodicité",

            choices=Event.PERIODICITIES, required=True)

class UpdateEventForm(NewEventForm):

    class Meta(NewEventForm.Meta):

        fields = (

                'start_datetime',

                'title',

                'date',

                'time',

                'duration',

                'ressource',

                'participants',

                'event_type',

                'services',

        )

class PeriodicEventsSearchForm(forms.Form):

    start_date = forms.DateField(localize=True, required = False)

    end_date = forms.DateField(required=False, localize=True)

    event_type = forms.MultipleChoiceField(

            choices=(

                (0, 'Rendez-vous patient'),

                (1, 'Évènement')),

            widget=forms.CheckboxSelectMultiple,

            initial=[0,1])

    no_end_date = forms.BooleanField(label=u'Sans date de fin', required=False)

    patient = AutoCompleteSelectField('patientrecord', required=False, help_text='')

    worker = AutoCompleteSelectField('worker', required=False, help_text='')

    def clean(self):

        cleaned_data = super(PeriodicEventsSearchForm, self).clean()

        if cleaned_data.get('start_date') and cleaned_data.get('end_date'):

            if cleaned_data['start_date'] > cleaned_data['end_date']:

                raise forms.ValidationError(u'La date de début doit être supérieure à la date de fin')

        return cleaned_data