Project

General

Profile

Download (8.76 KB) Statistics
| Branch: | Tag: | Revision:
0a82dcb8 Benjamin Dauvergne
# -*- coding: utf-8 -*-

from datetime import datetime, timedelta

from django import forms

76974b6f Benjamin Dauvergne
from ..dossiers.models import PatientRecord
from ..personnes.models import Worker
from ..ressources.models import ActType
from ..middleware.request import get_request
0a82dcb8 Benjamin Dauvergne
7b28c0b2 Benjamin Dauvergne
from ajax_select import make_ajax_field
37674910 Benjamin Dauvergne
from ajax_select.fields import AutoCompleteSelectField
76974b6f Benjamin Dauvergne
from models import Event, EventWithAct, EventType
7b28c0b2 Benjamin Dauvergne
3e9b47e9 Benjamin Dauvergne
class BaseForm(forms.ModelForm):
86f4d2f2 Benjamin Dauvergne
date = forms.DateField(label=u'Date', localize=True)
0a82dcb8 Benjamin Dauvergne
time = forms.TimeField(label=u'Heure de début')
74ba9776 Benjamin Dauvergne
duration = forms.CharField(label=u'Durée',
help_text=u'en minutes; vous pouvez utiliser la roulette de votre souris.')
76974b6f Benjamin Dauvergne
participants = make_ajax_field(EventWithAct, 'participants', 'worker-or-group', True)
3e9b47e9 Benjamin Dauvergne

class NewAppointmentForm(BaseForm):
76974b6f Benjamin Dauvergne
patient = make_ajax_field(EventWithAct, 'patient', 'patientrecord', False)
7b28c0b2 Benjamin Dauvergne
0a82dcb8 Benjamin Dauvergne
class Meta:
76974b6f Benjamin Dauvergne
model = EventWithAct
0a82dcb8 Benjamin Dauvergne
fields = (
5e9463c5 Benjamin Dauvergne
'start_datetime',
0a82dcb8 Benjamin Dauvergne
'date',
'time',
'duration',
'patient',
'participants',
dd986559 Serghei MIHAI
'ressource',
0a82dcb8 Benjamin Dauvergne
'act_type',
b41e6f51 Mikaël Ates
'description',
4ce756c8 Benjamin Dauvergne
'recurrence_periodicity',
0a4c0520 Benjamin Dauvergne
'recurrence_end_date'
0a82dcb8 Benjamin Dauvergne
)
5e9463c5 Benjamin Dauvergne
widgets = {
'start_datetime': forms.HiddenInput,
}

0a82dcb8 Benjamin Dauvergne

e92e5a1b Benjamin Dauvergne
def __init__(self, instance, service=None, **kwargs):
0a82dcb8 Benjamin Dauvergne
self.service = None
e92e5a1b Benjamin Dauvergne
super(NewAppointmentForm, self).__init__(instance=instance, **kwargs)
7f9269be Benjamin Dauvergne
self.fields['date'].css = 'datepicker'
f0a9b9ee Mikaël Ates
self.fields['participants'].required = True
0a82dcb8 Benjamin Dauvergne
if service:
self.service = service
self.fields['participants'].queryset = \
Worker.objects.for_service(service)
self.fields['patient'].queryset = \
PatientRecord.objects.for_service(service)
b37eebfb Jérôme Schneider
self.fields['act_type'].queryset = \
e1a80c08 Mikaël Ates
ActType.objects.for_service(service) \
.order_by('name')
0a82dcb8 Benjamin Dauvergne
dd45cd7d Benjamin Dauvergne
def clean_duration(self):
duration = self.cleaned_data['duration']
try:
return int(duration)
76974b6f Benjamin Dauvergne
except ValueError:
return 0
dd45cd7d Benjamin Dauvergne
0a4c0520 Benjamin Dauvergne
def clean(self):
cleaned_data = super(NewAppointmentForm, self).clean()
4ce756c8 Benjamin Dauvergne
if not cleaned_data.get('recurrence_periodicity'):
0a4c0520 Benjamin Dauvergne
cleaned_data['recurrence_end_date'] = None
1e03331f Jérôme Schneider
if cleaned_data.has_key('date') and cleaned_data.has_key('time'):
cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],
cleaned_data['time'])
0a4c0520 Benjamin Dauvergne
return cleaned_data

76974b6f Benjamin Dauvergne
def save(self, commit=True):
appointment = super(NewAppointmentForm, self).save(commit=False)
appointment.start_datetime = datetime.combine(self.cleaned_data['date'],
0a82dcb8 Benjamin Dauvergne
self.cleaned_data['time'])
76974b6f Benjamin Dauvergne
appointment.end_datetime = appointment.start_datetime + timedelta(
0a82dcb8 Benjamin Dauvergne
minutes=self.cleaned_data['duration'])
76974b6f Benjamin Dauvergne
appointment.creator = get_request().user
appointment.clean()
if commit:
appointment.save()
26c55fb5 Benjamin Dauvergne
appointment.services = [self.service]
76974b6f Benjamin Dauvergne
return appointment
37f9d898 Jérôme Schneider
88abfb2a Jérôme Schneider
class UpdateAppointmentForm(NewAppointmentForm):
9ca08421 Jérôme Schneider
88abfb2a Jérôme Schneider
class Meta(NewAppointmentForm.Meta):
fields = (
'start_datetime',
'date',
'time',
'duration',
'patient',
'participants',
dd986559 Serghei MIHAI
'ressource',
88abfb2a Jérôme Schneider
'act_type',
)


88e57268 Jérôme Schneider
class UpdatePeriodicAppointmentForm(NewAppointmentForm):
9ca08421 Jérôme Schneider
recurrence_periodicity = forms.ChoiceField(label=u"Périodicité",
choices=Event.PERIODICITIES, required=True)
c082b14e Jérôme Schneider
c48dcd4c Jérôme Schneider
def clean(self):
cleaned_data = super(UpdatePeriodicAppointmentForm, self).clean()
7af04d52 Jérôme Schneider
acts = self.instance.act_set.filter(is_billed=True).order_by('-date')
c48dcd4c Jérôme Schneider
if acts and cleaned_data.get('recurrence_end_date'):
recurrence_end_date = cleaned_data['recurrence_end_date']
7af04d52 Jérôme Schneider
if recurrence_end_date < acts[0].date:
c48dcd4c Jérôme Schneider
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
7af04d52 Jérôme Schneider
88abfb2a Jérôme Schneider
class DisablePatientAppointmentForm(UpdateAppointmentForm):
7af04d52 Jérôme Schneider
c082b14e Jérôme Schneider
def __init__(self, instance, service=None, **kwargs):
7af04d52 Jérôme Schneider
super(DisablePatientAppointmentForm, self).__init__(instance,
c082b14e Jérôme Schneider
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)
37f9d898 Jérôme Schneider

3e9b47e9 Benjamin Dauvergne
class NewEventForm(BaseForm):
b8608ae3 Frédéric Péters
title = forms.CharField(label=u"Complément de l'intitulé", max_length=32, required=False)
67707adb Jérôme Schneider
class Meta:
model = Event
fields = (
5e9463c5 Benjamin Dauvergne
'start_datetime',
67707adb Jérôme Schneider
'title',
'date',
'time',
'duration',
dd986559 Serghei MIHAI
'ressource',
69a9009e Jérôme Schneider
'participants',
0a4c0520 Benjamin Dauvergne
'event_type',
cc250927 Jérôme Schneider
'services',
b41e6f51 Mikaël Ates
'description',
4ce756c8 Benjamin Dauvergne
'recurrence_periodicity',
0a4c0520 Benjamin Dauvergne
'recurrence_end_date'
67707adb Jérôme Schneider
)
5e9463c5 Benjamin Dauvergne
widgets = {
'start_datetime': forms.HiddenInput,
b41e6f51 Mikaël Ates
'services': forms.CheckboxSelectMultiple,
5e9463c5 Benjamin Dauvergne
}
67707adb Jérôme Schneider
f9be3ef6 Benjamin Dauvergne
def __init__(self, instance, service=None, **kwargs):
self.service = service
67707adb Jérôme Schneider
super(NewEventForm, self).__init__(instance=instance, **kwargs)
self.fields['date'].css = 'datepicker'
c8d74e1a Jérôme Schneider
self.fields['event_type'].queryset = \
68e9f8c1 Frédéric Péters
EventType.objects.exclude(id=1).exclude(id=3).order_by('rank', 'label')
67707adb Jérôme Schneider
def clean_duration(self):
duration = self.cleaned_data['duration']
try:
return int(duration)
except:
d0220ea6 Jérôme Schneider
raise forms.ValidationError('Veuillez saisir un entier')
67707adb Jérôme Schneider
76974b6f Benjamin Dauvergne
def save(self, commit=True):
event = super(NewEventForm, self).save(commit=False)
event.start_datetime = datetime.combine(self.cleaned_data['date'],
67707adb Jérôme Schneider
self.cleaned_data['time'])
76974b6f Benjamin Dauvergne
event.end_datetime = event.start_datetime + timedelta(
67707adb Jérôme Schneider
minutes=self.cleaned_data['duration'])
76974b6f Benjamin Dauvergne
event.creator = get_request().user
event.clean()
if commit:
event.save()
26c55fb5 Benjamin Dauvergne
event.services = [self.service]
76974b6f Benjamin Dauvergne
return event
d525e593 Jérôme Schneider
68e9f8c1 Frédéric Péters
def clean(self):
cleaned_data = super(NewEventForm, self).clean()
1e03331f Jérôme Schneider
if cleaned_data.has_key('date') and cleaned_data.has_key('time'):
cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],
cleaned_data['time'])
4ce756c8 Benjamin Dauvergne
if not cleaned_data.get('recurrence_periodicity'):
0a4c0520 Benjamin Dauvergne
cleaned_data['recurrence_end_date'] = None
68e9f8c1 Frédéric Péters
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
2ae5dd3d Benjamin Dauvergne
9ca08421 Jérôme Schneider
class UpdatePeriodicEventForm(NewEventForm):
recurrence_periodicity = forms.ChoiceField(label=u"Périodicité",
choices=Event.PERIODICITIES, required=True)
2ae5dd3d Benjamin Dauvergne
class UpdateEventForm(NewEventForm):
class Meta(NewEventForm.Meta):
fields = (
'start_datetime',
'title',
'date',
'time',
'duration',
dd986559 Serghei MIHAI
'ressource',
2ae5dd3d Benjamin Dauvergne
'participants',
'event_type',
'services',
)
e46a1c3f Benjamin Dauvergne
class PeriodicEventsSearchForm(forms.Form):
4cb5c31f Serghei MIHAI
start_date = forms.DateField(localize=True, required = False)
e46a1c3f Benjamin Dauvergne
end_date = forms.DateField(required=False, localize=True)

37674910 Benjamin Dauvergne
event_type = forms.MultipleChoiceField(
choices=(
(0, 'Rendez-vous patient'),
4ad85397 Benjamin Dauvergne
(1, 'Évènement')),
37674910 Benjamin Dauvergne
widget=forms.CheckboxSelectMultiple,
initial=[0,1])
no_end_date = forms.BooleanField(label=u'Sans date de fin', required=False)
392c5526 Benjamin Dauvergne
patient = AutoCompleteSelectField('patientrecord', required=False, help_text='')
6afd600b Benjamin Dauvergne
worker = AutoCompleteSelectField('worker', required=False, help_text='')
37674910 Benjamin Dauvergne
e46a1c3f Benjamin Dauvergne
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