/calebasse/agenda/forms.py - Annotate - Calebasse - Redmine Entr’ouvert

calebasse / calebasse / agenda / forms.py @ 3bb50103

1	0a82dcb8	Benjamin Dauvergne	# -- coding: utf-8 --
2
3			from datetime import datetime, timedelta
4
5			from django import forms
6
7	76974b6f	Benjamin Dauvergne	from ..dossiers.models import PatientRecord
8			from ..personnes.models import Worker
9			from ..ressources.models import ActType
10			from ..middleware.request import get_request
11	0a82dcb8	Benjamin Dauvergne
12	7b28c0b2	Benjamin Dauvergne	from ajax_select import make_ajax_field
13	37674910	Benjamin Dauvergne	from ajax_select.fields import AutoCompleteSelectField
14	76974b6f	Benjamin Dauvergne	from models import Event, EventWithAct, EventType
15	7b28c0b2	Benjamin Dauvergne
16	3e9b47e9	Benjamin Dauvergne	class BaseForm(forms.ModelForm):
17	86f4d2f2	Benjamin Dauvergne	date = forms.DateField(label=u'Date', localize=True)
18	0a82dcb8	Benjamin Dauvergne	time = forms.TimeField(label=u'Heure de début')
19	74ba9776	Benjamin Dauvergne	duration = forms.CharField(label=u'Durée',
20			help_text=u'en minutes; vous pouvez utiliser la roulette de votre souris.')
21	76974b6f	Benjamin Dauvergne	participants = make_ajax_field(EventWithAct, 'participants', 'worker-or-group', True)
22	3e9b47e9	Benjamin Dauvergne
23
24			class NewAppointmentForm(BaseForm):
25	76974b6f	Benjamin Dauvergne	patient = make_ajax_field(EventWithAct, 'patient', 'patientrecord', False)
26	7b28c0b2	Benjamin Dauvergne
27	0a82dcb8	Benjamin Dauvergne	class Meta:
28	76974b6f	Benjamin Dauvergne	model = EventWithAct
29	0a82dcb8	Benjamin Dauvergne	fields = (
30	5e9463c5	Benjamin Dauvergne	'start_datetime',
31	0a82dcb8	Benjamin Dauvergne	'date',
32			'time',
33			'duration',
34			'patient',
35			'participants',
36			'room',
37			'act_type',
38	b41e6f51	Mikaël Ates	'description',
39	4ce756c8	Benjamin Dauvergne	'recurrence_periodicity',
40	0a4c0520	Benjamin Dauvergne	'recurrence_end_date'
41	0a82dcb8	Benjamin Dauvergne	)
42	5e9463c5	Benjamin Dauvergne	widgets = {
43			'start_datetime': forms.HiddenInput,
44			}
45
46	0a82dcb8	Benjamin Dauvergne
47
48	e92e5a1b	Benjamin Dauvergne	def __init__(self, instance, service=None, **kwargs):
49	0a82dcb8	Benjamin Dauvergne	self.service = None
50	e92e5a1b	Benjamin Dauvergne	super(NewAppointmentForm, self).__init__(instance=instance, **kwargs)
51	7f9269be	Benjamin Dauvergne	self.fields['date'].css = 'datepicker'
52	f0a9b9ee	Mikaël Ates	self.fields['participants'].required = True
53	0a82dcb8	Benjamin Dauvergne	if service:
54			self.service = service
55			self.fields['participants'].queryset = \
56			Worker.objects.for_service(service)
57			self.fields['patient'].queryset = \
58			PatientRecord.objects.for_service(service)
59	b37eebfb	Jérôme Schneider	self.fields['act_type'].queryset = \
60	e1a80c08	Mikaël Ates	ActType.objects.for_service(service) \
61			.order_by('name')
62	0a82dcb8	Benjamin Dauvergne
63	dd45cd7d	Benjamin Dauvergne	def clean_duration(self):
64			duration = self.cleaned_data['duration']
65			try:
66			return int(duration)
67	76974b6f	Benjamin Dauvergne	except ValueError:
68			return 0
69	dd45cd7d	Benjamin Dauvergne
70	0a4c0520	Benjamin Dauvergne	def clean(self):
71			cleaned_data = super(NewAppointmentForm, self).clean()
72	4ce756c8	Benjamin Dauvergne	if not cleaned_data.get('recurrence_periodicity'):
73	0a4c0520	Benjamin Dauvergne	cleaned_data['recurrence_end_date'] = None
74	1e03331f	Jérôme Schneider	if cleaned_data.has_key('date') and cleaned_data.has_key('time'):
75			cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],
76			cleaned_data['time'])
77	0a4c0520	Benjamin Dauvergne	return cleaned_data
78
79	76974b6f	Benjamin Dauvergne	def save(self, commit=True):
80			appointment = super(NewAppointmentForm, self).save(commit=False)
81			appointment.start_datetime = datetime.combine(self.cleaned_data['date'],
82	0a82dcb8	Benjamin Dauvergne	self.cleaned_data['time'])
83	76974b6f	Benjamin Dauvergne	appointment.end_datetime = appointment.start_datetime + timedelta(
84	0a82dcb8	Benjamin Dauvergne	minutes=self.cleaned_data['duration'])
85	76974b6f	Benjamin Dauvergne	appointment.creator = get_request().user
86			appointment.title = appointment.patient.display_name
87			appointment.clean()
88			if commit:
89			appointment.save()
90	26c55fb5	Benjamin Dauvergne	appointment.services = [self.service]
91	76974b6f	Benjamin Dauvergne	return appointment
92	37f9d898	Jérôme Schneider
93
94	76974b6f	Benjamin Dauvergne	class UpdateAppointmentForm(NewAppointmentForm):
95	2ae5dd3d	Benjamin Dauvergne	class Meta(NewAppointmentForm.Meta):
96			fields = (
97			'start_datetime',
98			'date',
99			'time',
100			'duration',
101			'patient',
102			'participants',
103			'room',
104			'act_type',
105			)
106	3d469788	Jérôme Schneider
107	37f9d898	Jérôme Schneider
108	3e9b47e9	Benjamin Dauvergne	class NewEventForm(BaseForm):
109	b8608ae3	Frédéric Péters	title = forms.CharField(label=u"Complément de l'intitulé", max_length=32, required=False)
110	67707adb	Jérôme Schneider
111			class Meta:
112			model = Event
113			fields = (
114	5e9463c5	Benjamin Dauvergne	'start_datetime',
115	67707adb	Jérôme Schneider	'title',
116			'date',
117			'time',
118			'duration',
119			'room',
120	69a9009e	Jérôme Schneider	'participants',
121	0a4c0520	Benjamin Dauvergne	'event_type',
122	cc250927	Jérôme Schneider	'services',
123	b41e6f51	Mikaël Ates	'description',
124	4ce756c8	Benjamin Dauvergne	'recurrence_periodicity',
125	0a4c0520	Benjamin Dauvergne	'recurrence_end_date'
126	67707adb	Jérôme Schneider	)
127	5e9463c5	Benjamin Dauvergne	widgets = {
128			'start_datetime': forms.HiddenInput,
129	b41e6f51	Mikaël Ates	'services': forms.CheckboxSelectMultiple,
130	5e9463c5	Benjamin Dauvergne	}
131	67707adb	Jérôme Schneider
132	f9be3ef6	Benjamin Dauvergne	def __init__(self, instance, service=None, **kwargs):
133			self.service = service
134	67707adb	Jérôme Schneider	super(NewEventForm, self).__init__(instance=instance, **kwargs)
135			self.fields['date'].css = 'datepicker'
136	c8d74e1a	Jérôme Schneider	self.fields['event_type'].queryset = \
137	68e9f8c1	Frédéric Péters	EventType.objects.exclude(id=1).exclude(id=3).order_by('rank', 'label')
138	67707adb	Jérôme Schneider
139			def clean_duration(self):
140			duration = self.cleaned_data['duration']
141			try:
142			return int(duration)
143			except:
144			return None
145
146	76974b6f	Benjamin Dauvergne	def save(self, commit=True):
147			event = super(NewEventForm, self).save(commit=False)
148			event.start_datetime = datetime.combine(self.cleaned_data['date'],
149	67707adb	Jérôme Schneider	self.cleaned_data['time'])
150	76974b6f	Benjamin Dauvergne	event.end_datetime = event.start_datetime + timedelta(
151	67707adb	Jérôme Schneider	minutes=self.cleaned_data['duration'])
152	76974b6f	Benjamin Dauvergne	event.creator = get_request().user
153			event.clean()
154			if commit:
155			event.save()
156	26c55fb5	Benjamin Dauvergne	event.services = [self.service]
157	76974b6f	Benjamin Dauvergne	return event
158	d525e593	Jérôme Schneider
159	68e9f8c1	Frédéric Péters	def clean(self):
160			cleaned_data = super(NewEventForm, self).clean()
161	1e03331f	Jérôme Schneider	if cleaned_data.has_key('date') and cleaned_data.has_key('time'):
162			cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],
163			cleaned_data['time'])
164	4ce756c8	Benjamin Dauvergne	if not cleaned_data.get('recurrence_periodicity'):
165	0a4c0520	Benjamin Dauvergne	cleaned_data['recurrence_end_date'] = None
166	68e9f8c1	Frédéric Péters	event_type = cleaned_data.get('event_type')
167			if event_type and event_type.id == 4 and not cleaned_data.get('title'): # 'Autre'
168			self._errors['title'] = self.error_class([
169			u"Ce champ est obligatoire pour les événements de type « Autre »."])
170			return cleaned_data
171	2ae5dd3d	Benjamin Dauvergne
172
173			class UpdateEventForm(NewEventForm):
174			class Meta(NewEventForm.Meta):
175			fields = (
176			'start_datetime',
177			'title',
178			'date',
179			'time',
180			'duration',
181			'room',
182			'participants',
183			'event_type',
184			'services',
185			)
186	e46a1c3f	Benjamin Dauvergne
187			class PeriodicEventsSearchForm(forms.Form):
188	37674910	Benjamin Dauvergne	start_date = forms.DateField(localize=True)
189	e46a1c3f	Benjamin Dauvergne	end_date = forms.DateField(required=False, localize=True)
190
191	37674910	Benjamin Dauvergne	event_type = forms.MultipleChoiceField(
192			choices=(
193			(0, 'Rendez-vous patient'),
194	4ad85397	Benjamin Dauvergne	(1, 'Évènement')),
195	37674910	Benjamin Dauvergne	widget=forms.CheckboxSelectMultiple,
196			initial=[0,1])
197			no_end_date = forms.BooleanField(label=u'Sans date de fin', required=False)
198	392c5526	Benjamin Dauvergne	patient = AutoCompleteSelectField('patientrecord', required=False, help_text='')
199	6afd600b	Benjamin Dauvergne	worker = AutoCompleteSelectField('worker', required=False, help_text='')
200	37674910	Benjamin Dauvergne
201	e46a1c3f	Benjamin Dauvergne	def clean(self):
202			cleaned_data = super(PeriodicEventsSearchForm, self).clean()
203			if cleaned_data.get('start_date') and cleaned_data.get('end_date'):
204			if cleaned_data['start_date'] > cleaned_data['end_date']:
205			raise forms.ValidationError(u'La date de début doit être supérieure à la date de fin')
206			return cleaned_data

Project

General

Profile

Produits Entr'ouvert » Calebasse

calebasse / calebasse / agenda / forms.py @ 3bb50103