Project

General

Profile

Download (6.38 KB) Statistics
| Branch: | Tag: | Revision:

calebasse / calebasse / agenda / forms.py @ e1a80c08

1
# -*- coding: utf-8 -*-
2

    
3
from datetime import datetime, timedelta
4

    
5
from django import forms
6

    
7
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

    
12
from ajax_select import make_ajax_field
13
from models import Event, EventWithAct, EventType
14

    
15
class NewAppointmentForm(forms.ModelForm):
16
    date = forms.DateField(label=u'Date', localize=True)
17
    time = forms.TimeField(label=u'Heure de début')
18
    duration = forms.CharField(label=u'Durée',
19
            help_text=u'en minutes; vous pouvez utiliser la roulette de votre souris.')
20

    
21
    participants = make_ajax_field(EventWithAct, 'participants', 'worker-or-group', True)
22
    patient = make_ajax_field(EventWithAct, 'patient', 'patientrecord', False)
23

    
24
    class Meta:
25
        model = EventWithAct
26
        fields = (
27
                'start_datetime',
28
                'date',
29
                'time',
30
                'duration',
31
                'patient',
32
                'participants',
33
                'room',
34
                'act_type',
35
                'description',
36
                'recurrence_periodicity',
37
                'recurrence_end_date'
38
        )
39
        widgets = {
40
                'start_datetime': forms.HiddenInput,
41
        }
42

    
43

    
44

    
45
    def __init__(self, instance, service=None, **kwargs):
46
        self.service = None
47
        super(NewAppointmentForm, self).__init__(instance=instance, **kwargs)
48
        self.fields['date'].css = 'datepicker'
49
        self.fields['participants'].required = True
50
        if service:
51
            self.service = service
52
            self.fields['participants'].queryset = \
53
                    Worker.objects.for_service(service)
54
            self.fields['patient'].queryset = \
55
                    PatientRecord.objects.for_service(service)
56
            self.fields['act_type'].queryset = \
57
                    ActType.objects.for_service(service) \
58
                    .order_by('name')
59

    
60
    def clean_duration(self):
61
        duration = self.cleaned_data['duration']
62
        try:
63
            return int(duration)
64
        except ValueError:
65
            return 0
66

    
67
    def clean(self):
68
        cleaned_data = super(NewAppointmentForm, self).clean()
69
        if not cleaned_data.get('recurrence_periodicity'):
70
            cleaned_data['recurrence_end_date'] = None
71
        cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],
72
                cleaned_data['time'])
73
        return cleaned_data
74

    
75
    def save(self, commit=True):
76
        appointment = super(NewAppointmentForm, self).save(commit=False)
77
        appointment.start_datetime = datetime.combine(self.cleaned_data['date'],
78
                    self.cleaned_data['time'])
79
        appointment.end_datetime = appointment.start_datetime + timedelta(
80
                minutes=self.cleaned_data['duration'])
81
        appointment.creator = get_request().user
82
        appointment.title = appointment.patient.display_name
83
        appointment.clean()
84
        if commit:
85
            appointment.save()
86
            appointment.services = [self.service]
87
        return appointment
88

    
89

    
90
class UpdateAppointmentForm(NewAppointmentForm):
91
    class Meta(NewAppointmentForm.Meta):
92
        fields = (
93
                'start_datetime',
94
                'date',
95
                'time',
96
                'duration',
97
                'patient',
98
                'participants',
99
                'room',
100
                'act_type',
101
        )
102

    
103

    
104
class NewEventForm(forms.ModelForm):
105

    
106
    title = forms.CharField(label=u"Complément de l'intitulé", max_length=32, required=False)
107
    date = forms.DateField(label=u'Date', localize=True)
108
    time = forms.TimeField(label=u'Heure de début')
109
    duration = forms.CharField(label=u'Durée',
110
            help_text=u'en minutes; vous pouvez utiliser la roulette de votre souris.')
111

    
112
    participants = make_ajax_field(Event, 'participants', 'worker-or-group', True)
113

    
114
    class Meta:
115
        model = Event
116
        fields = (
117
                'start_datetime',
118
                'title',
119
                'date',
120
                'time',
121
                'duration',
122
                'room',
123
                'participants',
124
                'event_type',
125
                'services',
126
                'description',
127
                'recurrence_periodicity',
128
                'recurrence_end_date'
129
        )
130
        widgets = {
131
                'start_datetime': forms.HiddenInput,
132
                'services': forms.CheckboxSelectMultiple,
133
        }
134

    
135
    def __init__(self, instance, service=None, **kwargs):
136
        self.service = service
137
        super(NewEventForm, self).__init__(instance=instance, **kwargs)
138
        self.fields['date'].css = 'datepicker'
139
        self.fields['event_type'].queryset = \
140
                    EventType.objects.exclude(id=1).exclude(id=3).order_by('rank', 'label')
141

    
142
    def clean_duration(self):
143
        duration = self.cleaned_data['duration']
144
        try:
145
            return int(duration)
146
        except:
147
            return None
148

    
149
    def save(self, commit=True):
150
        event = super(NewEventForm, self).save(commit=False)
151
        event.start_datetime = datetime.combine(self.cleaned_data['date'],
152
                    self.cleaned_data['time'])
153
        event.end_datetime = event.start_datetime + timedelta(
154
                minutes=self.cleaned_data['duration'])
155
        event.creator = get_request().user
156
        event.clean()
157
        if commit:
158
            event.save()
159
            event.services = [self.service]
160
        return event
161

    
162
    def clean(self):
163
        cleaned_data = super(NewEventForm, self).clean()
164
        cleaned_data['start_datetime'] = datetime.combine(cleaned_data['date'],
165
                cleaned_data['time'])
166
        if not cleaned_data.get('recurrence_periodicity'):
167
            cleaned_data['recurrence_end_date'] = None
168
        event_type = cleaned_data.get('event_type')
169
        if event_type and event_type.id == 4 and not cleaned_data.get('title'): # 'Autre'
170
            self._errors['title'] = self.error_class([
171
                            u"Ce champ est obligatoire pour les événements de type « Autre »."])
172
        return cleaned_data
173

    
174

    
175
class UpdateEventForm(NewEventForm):
176
    class Meta(NewEventForm.Meta):
177
        fields = (
178
                'start_datetime',
179
                'title',
180
                'date',
181
                'time',
182
                'duration',
183
                'room',
184
                'participants',
185
                'event_type',
186
                'services',
187
        )
(5-5/10)