448 lines
18 KiB
Python
448 lines
18 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from datetime import datetime, date, time as datetime_time
|
|
|
|
from django.db import models
|
|
from django.db.models import query
|
|
from django.contrib.auth.models import User
|
|
from django.template.defaultfilters import date as date_filter
|
|
from django import forms
|
|
from django.core.urlresolvers import reverse
|
|
|
|
from model_utils.managers import InheritanceManager
|
|
|
|
from calebasse.models import PhoneNumberField
|
|
from calebasse.ressources.models import Service, NamedAbstractModel
|
|
from calebasse.models import (BaseModelMixin, WeekRankField,
|
|
PhoneNumberField, ZipCodeField)
|
|
from calebasse.utils import weeks_since_epoch, weekday_ranks
|
|
|
|
from interval import Interval
|
|
|
|
from model_utils import Choices
|
|
from model_utils.managers import PassThroughManager
|
|
|
|
from ..middleware.request import get_request
|
|
|
|
class Role(NamedAbstractModel):
|
|
users = models.ManyToManyField(User,
|
|
verbose_name=u'Utilisateurs', blank=True)
|
|
|
|
class People(BaseModelMixin, models.Model):
|
|
GENDERS = Choices(
|
|
(1, 'Masculin'),
|
|
(2, 'Féminin'),
|
|
)
|
|
|
|
objects = InheritanceManager()
|
|
last_name = models.CharField(max_length=128, verbose_name=u'Nom',
|
|
db_index=True)
|
|
first_name = models.CharField(max_length=128, verbose_name=u'Prénom(s)',
|
|
blank=True, null=True)
|
|
display_name = models.CharField(max_length=256,
|
|
verbose_name=u'Nom complet', editable=False, db_index=True)
|
|
gender = models.IntegerField(verbose_name=u"Genre", choices=GENDERS,
|
|
max_length=1, blank=True, null=True)
|
|
email = models.EmailField(blank=True, null=True)
|
|
phone = PhoneNumberField(verbose_name=u"Téléphone", blank=True, null=True)
|
|
|
|
def save(self, **kwargs):
|
|
if self.first_name:
|
|
self.display_name = self.first_name + ' ' + self.last_name.upper()
|
|
else:
|
|
self.display_name = self.last_name.upper()
|
|
super(People, self).save(**kwargs)
|
|
|
|
def __unicode__(self):
|
|
return self.display_name
|
|
|
|
def get_initials(self):
|
|
initials = []
|
|
if self.first_name:
|
|
initials = [name[0].upper() for name in ' '.join(self.first_name.split('-')).split()]
|
|
initials += [name[0].upper() for name in ' '.join(self.last_name.split('-')).split()]
|
|
return ''.join(initials)
|
|
|
|
class Meta:
|
|
ordering = ['last_name', 'first_name']
|
|
|
|
class WorkerQuerySet(query.QuerySet):
|
|
def for_service(self, service, type=None):
|
|
if type:
|
|
return self.filter(enabled=True, services__in=[service], type=type)
|
|
else:
|
|
return self.filter(enabled=True, services__in=[service])
|
|
|
|
|
|
class Worker(People):
|
|
objects = PassThroughManager.for_queryset_class(WorkerQuerySet)()
|
|
|
|
initials = models.CharField(max_length=5, verbose_name=u'Initiales', default='', blank=True)
|
|
type = models.ForeignKey('ressources.WorkerType',
|
|
verbose_name=u'Type de personnel')
|
|
services = models.ManyToManyField('ressources.Service', blank=True, null=True)
|
|
enabled = models.BooleanField(verbose_name=u'Actif',
|
|
default=True)
|
|
old_camsp_id = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien ID au CAMSP', blank=True, null=True)
|
|
old_cmpp_id = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien ID au CMPP', blank=True, null=True)
|
|
old_sessad_dys_id = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien ID au SESSAD TED', blank=True, null=True)
|
|
old_sessad_ted_id = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien ID au SESSAD DYS', blank=True, null=True)
|
|
|
|
def save(self, **kwargs):
|
|
if not self.initials:
|
|
self.initials = self.get_initials()
|
|
super(Worker, self).save(**kwargs)
|
|
|
|
def is_active(self):
|
|
return self.enabled
|
|
|
|
def is_away(self):
|
|
if self.timetable_set.filter(weekday=date.today().weekday()).exists():
|
|
return False
|
|
return True
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('worker_update', kwargs={
|
|
'service': self.services.all()[0].slug.lower(),
|
|
'pk': self.pk })
|
|
|
|
class Meta:
|
|
verbose_name = u'Personnel'
|
|
verbose_name_plural = u'Personnels'
|
|
|
|
class ExternalWorker(People):
|
|
description = models.TextField(blank=True, null=True, default=None)
|
|
address = models.CharField(max_length=120,
|
|
verbose_name=u"Adresse", blank=True, null=True, default=None)
|
|
address_complement = models.CharField(max_length=120,
|
|
blank=True,
|
|
null=True,
|
|
default=None,
|
|
verbose_name=u"Complément d'adresse")
|
|
zip_code = ZipCodeField(verbose_name=u"Code postal",
|
|
blank=True, null=True, default=None)
|
|
city = models.CharField(max_length=80, verbose_name=u"Ville",
|
|
blank=True, null=True, default=None)
|
|
phone_work = PhoneNumberField(verbose_name=u"Téléphone du travail",
|
|
blank=True, null=True, default=None)
|
|
fax = models.CharField(max_length=30,
|
|
blank=True, null=True, default=None)
|
|
type = models.ForeignKey('ressources.WorkerType',
|
|
verbose_name=u'Spécialité', default=18)
|
|
old_id = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien ID', blank=True, null=True)
|
|
old_service = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien Service', blank=True, null=True)
|
|
class Meta:
|
|
verbose_name = u'Intervenant extérieur'
|
|
verbose_name_plural = u'Intervenants extérieurs'
|
|
|
|
class ExternalTherapist(People):
|
|
description = models.TextField(blank=True, null=True, default=None)
|
|
address = models.CharField(max_length=120,
|
|
verbose_name=u"Adresse", blank=True, null=True, default=None)
|
|
address_complement = models.CharField(max_length=120,
|
|
blank=True,
|
|
null=True,
|
|
default=None,
|
|
verbose_name=u"Complément d'adresse")
|
|
zip_code = ZipCodeField(verbose_name=u"Code postal",
|
|
blank=True, null=True, default=None)
|
|
city = models.CharField(max_length=80, verbose_name=u"Ville",
|
|
blank=True, null=True, default=None)
|
|
phone_work = PhoneNumberField(verbose_name=u"Téléphone du travail",
|
|
blank=True, null=True, default=None)
|
|
fax = models.CharField(max_length=30,
|
|
blank=True, null=True, default=None)
|
|
type = models.ForeignKey('ressources.WorkerType',
|
|
verbose_name=u'Spécialité', default=18)
|
|
old_id = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien ID', blank=True, null=True)
|
|
old_service = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien Service', blank=True, null=True)
|
|
old_id = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien ID', blank=True, null=True)
|
|
old_service = models.CharField(max_length=256,
|
|
verbose_name=u'Ancien Service', blank=True, null=True)
|
|
class Meta:
|
|
verbose_name = u'Médecin extérieur'
|
|
verbose_name_plural = u'Médecins extérieurs'
|
|
|
|
class UserWorker(BaseModelMixin, models.Model):
|
|
user = models.OneToOneField('auth.User')
|
|
worker = models.ForeignKey('Worker',
|
|
verbose_name=u'Personnel')
|
|
|
|
def __unicode__(self):
|
|
return u'Lien entre la personne {0} et l\'utilisateur {1}'.format(
|
|
self.worker, self.user)
|
|
|
|
class SchoolTeacher(People):
|
|
schools = models.ManyToManyField('ressources.School')
|
|
role = models.ForeignKey('ressources.SchoolTeacherRole')
|
|
|
|
class TimeTableQuerySet(query.QuerySet):
|
|
def current(self, today=None):
|
|
if today is None:
|
|
today = date.today()
|
|
return self.filter(models.Q(start_date__lte=today) | models.Q(start_date__isnull=True)) \
|
|
.filter(models.Q(end_date__gte=today) | models.Q(end_date__isnull=True))
|
|
|
|
def for_today(self, today=None):
|
|
if today is None:
|
|
today = date.today()
|
|
qs = self.current(today)
|
|
qs = qs.filter(weekday=today.weekday())
|
|
filters = []
|
|
# week periods
|
|
for week_period in range(1,5):
|
|
filters.append(models.Q(week_period=week_period,
|
|
week_offset=weeks_since_epoch(today) % week_period))
|
|
# week parity
|
|
parity = today.isocalendar()[1] % 2
|
|
filters.append(models.Q(week_parity=parity))
|
|
# week ranks
|
|
filters.append(models.Q(week_rank__in=weekday_ranks(today)))
|
|
qs = qs.filter(reduce(models.Q.__or__, filters))
|
|
return qs
|
|
|
|
PERIODICITIES = (
|
|
(1, u'Toutes les semaines'),
|
|
(2, u'Une semaine sur deux'),
|
|
(3, u'Une semaine sur trois'),
|
|
(4, u'Une semaine sur quatre'),
|
|
(5, u'Une semaine sur cinq'),
|
|
(6, u'La première semaine du mois'),
|
|
(7, u'La deuxième semaine du mois'),
|
|
(8, u'La troisième semaine du mois'),
|
|
(9, u'La quatrième semaine du mois'),
|
|
(10, u'La dernière semaine du mois'),
|
|
(11, u'Les semaines paires'),
|
|
(12, u'Les semaines impaires')
|
|
)
|
|
|
|
|
|
class TimeTable(BaseModelMixin, models.Model):
|
|
objects = PassThroughManager.for_queryset_class(TimeTableQuerySet)()
|
|
worker = models.ForeignKey(Worker,
|
|
verbose_name=u'Intervenant')
|
|
services = models.ManyToManyField('ressources.Service')
|
|
WEEKDAYS = Choices(*enumerate(('lundi', 'mardi', 'mercredi', 'jeudi', 'vendredi',
|
|
'samedi', 'dimanche')))
|
|
|
|
weekday = models.PositiveIntegerField(
|
|
verbose_name=u"Jour de la semaine",
|
|
choices=WEEKDAYS)
|
|
start_time = models.TimeField(
|
|
verbose_name=u'Heure de début')
|
|
end_time = models.TimeField(
|
|
verbose_name=u'Heure de fin')
|
|
start_date = models.DateField(
|
|
verbose_name=u'Début',
|
|
help_text=u'format: jj/mm/aaaa')
|
|
end_date = models.DateField(
|
|
verbose_name=u'Fin', blank=True, null=True,
|
|
help_text=u'format: jj/mm/aaaa')
|
|
|
|
periodicity = models.PositiveIntegerField(
|
|
choices=PERIODICITIES,
|
|
verbose_name=u"Périodicité",
|
|
default=1,
|
|
blank=True,
|
|
null=True)
|
|
|
|
PERIODS = (
|
|
(1, u'Toutes les semaines'),
|
|
(2, u'Une semaine sur deux'),
|
|
(3, u'Une semaine sur trois'),
|
|
(4, u'Une semaine sur quatre'),
|
|
(5, u'Une semaine sur cinq')
|
|
)
|
|
OFFSET = range(0,4)
|
|
week_offset = models.PositiveIntegerField(
|
|
choices=zip(OFFSET, OFFSET),
|
|
verbose_name=u"Décalage en semaines par rapport au 1/1/1970 pour le calcul de période",
|
|
default=0)
|
|
week_period = models.PositiveIntegerField(
|
|
choices=PERIODS,
|
|
verbose_name=u"Période en semaines",
|
|
blank=True, null=True)
|
|
|
|
PARITIES = (
|
|
(0, u'Les semaines paires'),
|
|
(1, u'Les semaines impaires')
|
|
)
|
|
week_parity = models.PositiveIntegerField(
|
|
choices=PARITIES,
|
|
verbose_name=u"Parité des semaines",
|
|
blank=True, null=True)
|
|
|
|
WEEK_RANKS = (
|
|
(0, u'La première semaine du mois'),
|
|
(1, u'La deuxième semaine du mois'),
|
|
(2, u'La troisième semaine du mois'),
|
|
(3, u'La quatrième semaine du mois'),
|
|
(4, u'La dernière semaine du mois')
|
|
)
|
|
|
|
week_rank = models.PositiveIntegerField(
|
|
verbose_name=u"Rang de la semaine dans le mois",
|
|
choices=WEEK_RANKS,
|
|
blank=True, null=True)
|
|
|
|
def clean(self):
|
|
if (self.week_period is None) + (self.week_parity is None) + \
|
|
(self.week_rank is None) != 2:
|
|
raise forms.ValidationError('Only one periodicity criteria can be used')
|
|
if self.week_period and self.start_date:
|
|
self.week_offset = weeks_since_epoch(self.start_date) % self.week_period
|
|
|
|
def __unicode__(self):
|
|
s = u'%s pour au %s le %s de %s à %s' % \
|
|
(self.worker, ', '.join(map(unicode, self.services.all())), self.weekday, self.start_time,
|
|
self.end_time)
|
|
if self.end_time:
|
|
s += u' à partir du %s' % self.start_date
|
|
else:
|
|
s += u' du %s au %s' % (self.start_data, self.end_date)
|
|
return s
|
|
|
|
class Meta:
|
|
verbose_name = u'Emploi du temps'
|
|
verbose_name_plural = u'Emplois du temps'
|
|
|
|
def to_interval(self, date):
|
|
return Interval(datetime.combine(date, self.start_time),
|
|
datetime.combine(date, self.end_time))
|
|
|
|
class HolidayQuerySet(query.QuerySet):
|
|
# To grab group holidays:
|
|
# No worker AND
|
|
# Either the holiday has no service, that means for all
|
|
# Or the user must be in the service of the holiday
|
|
def for_worker(self, worker):
|
|
filter_query = models.Q(worker=worker) \
|
|
| models.Q(worker__isnull=True,
|
|
services = None) \
|
|
| models.Q(worker__isnull=True,
|
|
services__in = worker.services.all())
|
|
return self.filter(filter_query).distinct()
|
|
|
|
def for_worker_id(self, worker_id):
|
|
worker = None
|
|
try:
|
|
worker = Worker.objects.get(pk=worker_id)
|
|
except:
|
|
return None
|
|
filter_query = models.Q(worker=worker) \
|
|
| models.Q(worker__isnull=True,
|
|
services = None) \
|
|
| models.Q(worker__isnull=True,
|
|
services__in = worker.services.all())
|
|
return self.filter(filter_query)
|
|
|
|
def for_type(self, holiday_type):
|
|
return self.filter(holiday_type = holiday_type)
|
|
|
|
def for_service(self, service):
|
|
return self.filter(worker__isnull = True) \
|
|
.filter(models.Q(services = service)
|
|
|models.Q(services__isnull = True))
|
|
|
|
def for_service_workers(self, service):
|
|
return self.filter(models.Q(worker__services = [service])
|
|
|models.Q(services__in = [service])
|
|
|models.Q(worker__isnull=True, services__isnull = True))
|
|
|
|
def future(self):
|
|
return self.filter(end_date__gte=date.today())
|
|
|
|
def today(self, today=None):
|
|
today = today or date.today()
|
|
return self.filter(start_date__lte=today,
|
|
end_date__gte=today)
|
|
|
|
def for_period(self, start_date, end_date):
|
|
return self.filter(start_date__lte=end_date, end_date__gte=start_date)
|
|
|
|
def for_timed_period(self, date, start_time, end_time):
|
|
filter_query = models.Q(start_date__lt=date, end_date__gt=date) \
|
|
| models.Q(start_date=date, start_time__isnull=True, end_date__gt=date) \
|
|
| models.Q(start_date=date, start_time__lt=end_time, end_date__gt=date) \
|
|
| models.Q(start_date__lt=date, end_date=date, end_time__isnull=True) \
|
|
| models.Q(start_date__lt=date, end_date=date, end_time__gt=start_time) \
|
|
| models.Q(start_date=date, end_date=date, start_time__isnull=True, end_time__isnull=True) \
|
|
| models.Q(start_date=date, end_date=date, start_time__isnull=True, end_time__gt=start_time) \
|
|
| models.Q(start_date=date, end_date=date, start_time__lt=end_time, end_time__isnull=True) \
|
|
| models.Q(start_date=date, end_date=date, start_time__lte=start_time, end_time__gt=start_time) \
|
|
| models.Q(start_date=date, end_date=date, start_time__lt=end_time, end_time__gte=end_time)
|
|
return self.filter(filter_query)
|
|
|
|
def time2french(time):
|
|
if time.minute:
|
|
return '{0}h{1}'.format(time.hour, time.minute)
|
|
return '{0}h'.format(time.hour)
|
|
|
|
class Holiday(BaseModelMixin, models.Model):
|
|
objects = PassThroughManager().for_queryset_class(HolidayQuerySet)()
|
|
|
|
holiday_type = models.ForeignKey('ressources.HolidayType',
|
|
verbose_name=u'Type de congé')
|
|
worker = models.ForeignKey(Worker, blank=True, null=True,
|
|
verbose_name=u"Personnel")
|
|
services = models.ManyToManyField(Service, blank = False, null = False,
|
|
verbose_name = u'Services')
|
|
start_date = models.DateField(verbose_name=u"Date de début",
|
|
help_text=u'format: jj/mm/aaaa')
|
|
end_date = models.DateField(verbose_name=u"Date de fin",
|
|
help_text=u'format: jj/mm/aaaa')
|
|
start_time = models.TimeField(verbose_name=u"Horaire de début", blank=True,
|
|
null=True)
|
|
end_time = models.TimeField(verbose_name=u"Horaire de fin", blank=True,
|
|
null=True)
|
|
comment = models.TextField(verbose_name=u'Commentaire', blank=True)
|
|
|
|
class Meta:
|
|
verbose_name = u'Congé'
|
|
verbose_name_plural = u'Congés'
|
|
ordering = ('start_date', 'start_time')
|
|
|
|
def is_current(self):
|
|
return self.start_date <= date.today() <= self.end_date
|
|
|
|
def for_all_services(self):
|
|
return self.services.count() == Service.objects.count()
|
|
|
|
def __unicode__(self):
|
|
ret = ''
|
|
if self.start_date == self.end_date:
|
|
ret = u'le {0}'.format(date_filter(self.start_date, 'j F Y'))
|
|
if self.start_time:
|
|
ret += u', à partir de {0}'.format(time2french(self.start_time))
|
|
if self.end_time:
|
|
ret += u", jusqu'à {0}".format(time2french(self.end_time))
|
|
else:
|
|
ret = u'du {0}'.format(date_filter(self.start_date, 'j F Y'))
|
|
if self.start_time:
|
|
ret += u' (à partir de {0})'.format(time2french(self.start_time))
|
|
ret += u' au {0}'.format(date_filter(self.end_date, 'j F Y'))
|
|
if self.end_time:
|
|
ret += u" (jusqu'à {0})".format(time2french(self.end_time))
|
|
return ret
|
|
|
|
def to_interval(self, date=None):
|
|
if date == self.start_date:
|
|
start_time = self.start_time or datetime_time(8, 0)
|
|
else:
|
|
start_time = datetime_time(8, 0)
|
|
if date == self.end_date:
|
|
end_time = self.end_time or datetime_time(20, 0)
|
|
else:
|
|
end_time = datetime_time(20, 0)
|
|
return Interval(datetime.combine(self.start_date, start_time),
|
|
datetime.combine(self.end_date, end_time))
|