|
# -*- coding: utf-8 -*-
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from django import forms
|
|
|
|
from calebasse.dossiers.models import PatientRecord
|
|
from calebasse.personnes.models import Worker
|
|
from calebasse.actes.models import EventAct
|
|
from calebasse.agenda.models import Event
|
|
from calebasse.ressources.models import ActType
|
|
from calebasse.middleware.request import get_request
|
|
|
|
from ajax_select import make_ajax_field
|
|
|
|
class NewAppointmentForm(forms.ModelForm):
|
|
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(EventAct, 'participants', 'worker', True)
|
|
patient = make_ajax_field(EventAct, 'patient', 'patientrecord', False)
|
|
|
|
class Meta:
|
|
model = EventAct
|
|
fields = (
|
|
'date',
|
|
'time',
|
|
'duration',
|
|
'patient',
|
|
'participants',
|
|
'room',
|
|
'act_type',
|
|
)
|
|
|
|
|
|
def __init__(self, instance, service=None, **kwargs):
|
|
self.service = None
|
|
super(NewAppointmentForm, self).__init__(instance=instance, **kwargs)
|
|
self.fields['date'].css = 'datepicker'
|
|
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:
|
|
return None
|
|
|
|
def save(self, commit=False):
|
|
start_datetime = datetime.combine(self.cleaned_data['date'],
|
|
self.cleaned_data['time'])
|
|
end_datetime = start_datetime + timedelta(
|
|
minutes=self.cleaned_data['duration'])
|
|
patient = self.cleaned_data['patient']
|
|
creator = get_request().user
|
|
self.instance = EventAct.objects.create_patient_appointment(
|
|
creator=creator,
|
|
title=patient.display_name,
|
|
patient=patient,
|
|
participants=self.cleaned_data['participants'],
|
|
act_type=self.cleaned_data['act_type'],
|
|
service=self.service,
|
|
start_datetime=start_datetime,
|
|
end_datetime=end_datetime,
|
|
description='',
|
|
room=self.cleaned_data['room'],
|
|
note=None,)
|
|
return self.instance
|
|
|
|
class UpdateAppointmentForm(NewAppointmentForm):
|
|
|
|
def __init__(self, instance, service=None, occurrence=None, **kwargs):
|
|
super(UpdateAppointmentForm, self).__init__(instance=instance, service, **kwargs)
|
|
self.occurrence = occurrence
|
|
|
|
|
|
def save(self):
|
|
self.occurrence.start_time = datetime.combine(
|
|
self.cleaned_data['date'],
|
|
self.cleaned_data['time'])
|
|
self.occurrence.end_time = start_datetime + timedelta(
|
|
minutes=self.cleaned_data['duration'])
|
|
self.occurrence.save()
|
|
patient = self.cleaned_data['patient']
|
|
creator = get_request().user
|
|
self.instance.title = patient.display_name
|
|
return self.instance
|
|
|
|
|
|
class NewEventForm(forms.ModelForm):
|
|
|
|
date = forms.DateField(label=u'Date')
|
|
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(EventAct, 'participants', 'worker', True)
|
|
|
|
class Meta:
|
|
model = Event
|
|
fields = (
|
|
'title',
|
|
'date',
|
|
'time',
|
|
'duration',
|
|
'room',
|
|
'participants',
|
|
'services',
|
|
'event_type'
|
|
)
|
|
|
|
def __init__(self, instance, **kwargs):
|
|
super(NewEventForm, self).__init__(instance=instance, **kwargs)
|
|
self.fields['date'].css = 'datepicker'
|
|
|
|
def clean_duration(self):
|
|
duration = self.cleaned_data['duration']
|
|
try:
|
|
return int(duration)
|
|
except:
|
|
return None
|
|
|
|
def save(self, commit=False):
|
|
start_datetime = datetime.combine(self.cleaned_data['date'],
|
|
self.cleaned_data['time'])
|
|
end_datetime = start_datetime + timedelta(
|
|
minutes=self.cleaned_data['duration'])
|
|
self.instance = Event.objects.create_event(
|
|
title=self.cleaned_data['title'],
|
|
event_type=self.cleaned_data['event_type'],
|
|
participants=self.cleaned_data['participants'],
|
|
services=self.cleaned_data['services'],
|
|
start_datetime=start_datetime,
|
|
end_datetime=end_datetime,
|
|
description='',
|
|
room=self.cleaned_data['room'],
|
|
note=None,)
|
|
return self.instance
|
|
|