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 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 = make_ajax_field(EventWithAct, 'patient', 'patientrecord', False)

    class Meta:

        model = EventWithAct

        fields = (

                'start_datetime',

                'date',

                'time',

                'duration',

                'patient',

                'participants',

                'room',

                'act_type',

                '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)

    def clean_duration(self):

        duration = self.cleaned_data['duration']

        try:

            return int(duration)

        except ValueError:

            return 0

    def clean(self):

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

        if not cleaned_data.get('recurrence_periodicity'):

            cleaned_data['recurrence_end_date'] = None

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

                cleaned_data['time'])

        return cleaned_data

    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.title = appointment.patient.display_name

        appointment.clean()

        if commit:

            appointment.save()

            appointment.services = [self.service]

        return appointment

class UpdateAppointmentForm(NewAppointmentForm):

    class Meta(NewAppointmentForm.Meta):

        fields = (

                'start_datetime',

                'date',

                'time',

                'duration',

                'patient',

                'participants',

                'room',

                'act_type',

        )

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',

                'room',

                'participants',

                'event_type',

                'services',

                '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:

            return None

    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()

        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 UpdateEventForm(NewEventForm):

    class Meta(NewEventForm.Meta):

        fields = (

                'start_datetime',

                'title',

                'date',

                'time',

                'duration',

                'room',

                'participants',

                'event_type',

                'services',

        )