Produits Entr'ouvert » Calebasse

# -*- coding: utf-8 -*-

from datetime import datetime, date

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

import reversion

from calebasse.ressources.models import Service, NamedAbstractModel

from calebasse.models import BaseModelMixin, WeekRankField

from calebasse.utils import weeks_since_epoch, weekday_ranks

from interval import Interval

from model_utils import Choices

from model_utils.managers import PassThroughManager

class Role(NamedAbstractModel):

    users = models.ManyToManyField(User,

                verbose_name=u'Utilisateurs', blank=True)

class People(BaseModelMixin, models.Model):

    GENDERS =  Choices(

            (1, 'homme'),

            (2, 'femme'),

    )

    last_name = models.CharField(max_length=128, verbose_name=u'Nom')

    first_name = models.CharField(max_length=128, verbose_name=u'Prénom(s)')

    display_name = models.CharField(max_length=256,

            verbose_name=u'Nom complet', editable=False)

    gender = models.IntegerField(verbose_name=u"Genre", choices=GENDERS,

            max_length=1, default=0, blank=True)

    def save(self, **kwargs):

        self.display_name = self.first_name + ' ' + self.last_name

        super(People, self).save(**kwargs)

    def __unicode__(self):

        return self.display_name

    class Meta:

        ordering = ['last_name', 'first_name']

class WorkerQuerySet(query.QuerySet):

    def for_service(self, service, type=None):

        if type:

            return self.filter(services__in=[service]).filter(type=type)

        else:

            return self.filter(services__in=[service])

class Worker(People):

    objects = PassThroughManager.for_queryset_class(WorkerQuerySet)()

    # TODO : use manytomany here ?

    type = models.ForeignKey('ressources.WorkerType',

            verbose_name=u'Type de personnel')

    services = models.ManyToManyField('ressources.Service')

    def is_away(self):

        if self.timetable_set.filter(weekday=date.today().weekday()).exists():

            return False

        return True

    @models.permalink

    def get_absolute_url(self):

        return ('worker_update', (), {

            'service': self.services.all()[0].name.lower(),

            'pk': self.pk })

    class Meta:

        verbose_name = u'Personnel'

        verbose_name_plural = u'Personnels'

reversion.register(Worker, follow=['people_ptr'])

reversion.register(User)

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)

reversion.register(UserWorker)

class SchoolTeacher(People):

    schools = models.ManyToManyField('ressources.School')

    role = models.ForeignKey('ressources.SchoolTeacherRole')

reversion.register(SchoolTeacher, follow=['user_ptr'])

class TimeTableQuerySet(query.QuerySet):

    def current(self, today=None):

        if today is None:

            today = date.today()

        return self.filter(start_date__lte=today, end_date__gte=today)

    def for_today(self, today=None):

        if today is None:

            today = date.today()

        qs = self.current(today)

        qs = self.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]-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

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')

    PERIODS = (

            (1, u'Toutes les semaines'),

            (2, u'Une semaine sur deux'),

            (3, u'Une semaine sur trois'),

            (4, 'Une semaine sur quatre'),

            (4, '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",

            default=1,

            blank=True,

            null=True)

    PARITIES = (

            (0, 'Les semaines paires'),

            (1, 'Les semaines impaires')

    )

    week_parity = models.PositiveIntegerField(

            choices=PARITIES,

            verbose_name=u"Parité des semaines",

            default=None,

            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 = WeekRankField(

            verbose_name=u"Rang de la semaine dans le mois",

            choices=WEEK_RANKS,

            default=None)

    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):

    def for_worker(self, worker):

        filter_query = models.Q(worker=worker) \

              | models.Q(worker__isnull=True,

                           service=worker.services.all())

        return self.filter(filter_query)

    def for_service(self, service):

        return self.filter(worker__isnull=True) \

                   .filter(models.Q(service=service)

                          |models.Q(service__isnull=True)) \

    def for_service_workers(self, service):

        return self.filter(worker__services=service)

    def future(self):

        return self.filter(end_date__gte=date.today())

    def today(self):

        today = 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 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)()

    worker = models.ForeignKey(Worker, blank=True, null=True,

            verbose_name=u"Personnel")

    service = models.ForeignKey(Service, blank=True, null=True,

            verbose_name=u"Service")

    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 __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 and self.end_time:

                ret += u', de {0} à {1}'.format(time2french(self.start_time),

                time2french(self.end_time))

        else:

            ret = u'du {0} au {1}'.format(

                    date_filter(self.start_date, 'j F'),

                    date_filter(self.end_date, 'j F Y'))

        return ret

class ExternalDoctor(People):

    class Meta:

        verbose_name = u'Médecin extérieur'

        verbose_name_plural = u'Médecins extérieurs'

class ExternalIntervener(People):

    class Meta:

        verbose_name = u'Intervenant extérieur'

        verbose_name_plural = u'Intervenants extérieurs'