303 lines
12 KiB
Python
303 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
from datetime import datetime, timedelta, time
|
||
|
||
from django import forms
|
||
from django.db.models import Q
|
||
from django.utils.translation import ugettext_lazy as _
|
||
|
||
from ..dossiers.models import PatientRecord
|
||
from ..personnes.models import Worker
|
||
from ..actes.models import Act
|
||
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
|
||
self.fields['time'].required = False
|
||
self.fields['duration'].required = False
|
||
if service:
|
||
self.service = service
|
||
self.special_types = [str(act.id) for act in ActType.objects.filter(Q(name__iexact='courriel')
|
||
| Q(name__iexact='telephone')
|
||
| Q(name__iexact='téléphone'))]
|
||
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_time(self):
|
||
if self.cleaned_data['time']:
|
||
return self.cleaned_data['time']
|
||
|
||
act_type = self.data.get('act_type')
|
||
# act type is available as raw data from the post request
|
||
if act_type in self.special_types:
|
||
return time(8, 0)
|
||
raise forms.ValidationError(_(u'This field is required.'))
|
||
|
||
def clean_duration(self):
|
||
if self.cleaned_data['duration']:
|
||
duration = self.cleaned_data['duration']
|
||
try:
|
||
duration = int(duration)
|
||
if duration <= 0:
|
||
raise ValueError
|
||
return duration
|
||
except ValueError:
|
||
raise forms.ValidationError(u'Le champ doit contenir uniquement des chiffres')
|
||
|
||
act_type = self.data.get('act_type')
|
||
if act_type in self.special_types:
|
||
return 10
|
||
raise forms.ValidationError(_(u'This field is required.'))
|
||
|
||
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()
|
||
self.save_m2m()
|
||
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()
|
||
'''If one act not new in the reccurrence, we prevent to modify the
|
||
start date or the recurrence_periodicity since this could trigger act
|
||
deletion'''
|
||
start_datetime = cleaned_data.get('start_datetime')
|
||
if start_datetime and start_datetime != self.instance.start_datetime \
|
||
and self.instance.one_act_not_new():
|
||
self._errors['start_datetime'] = self.error_class([
|
||
u"La date de début ne peut-être modifiée car un acte de la récurrence est pointé"])
|
||
# FIXME
|
||
# recurrence_periodicity = cleaned_data.get('recurrence_periodicity')
|
||
# if recurrence_periodicity and recurrence_periodicity != str(self.instance.recurrence_periodicity) \
|
||
# and self.instance.one_act_not_new():
|
||
# self._errors['recurrence_periodicity'] = self.error_class([
|
||
# u"La récurrence ne peut-être modifiée car un acte de la récurrence est pointé"])
|
||
'''We check that the end date is posterior to the last act not new'''
|
||
recurrence_end_date = cleaned_data.get('recurrence_end_date')
|
||
if recurrence_end_date and recurrence_end_date != self.instance.recurrence_end_date \
|
||
and self.instance.one_act_not_new():
|
||
last = self.instance.last_act_not_new()
|
||
if last and last.date > recurrence_end_date:
|
||
self._errors['recurrence_end_date'] = self.error_class([
|
||
u"La date de fin doit être postérieure au dernier acte pointé de la récurrence (%s)" % last])
|
||
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 antérieure à la date de fin')
|
||
return cleaned_data
|