5637

[auf_rh_dae.git] / project / dae / catalogues.py

# -*- encoding: utf-8 -*-

import datetime

from django.db.models import Q
from project import groups
from project.rh import models as rh


class Responsable(object):
    q = ""

    def get_query(self, q, request):
        if len(q) < 4:
            return rh.Poste.objects.none()

        self.q = q
        filtre = (
            Q(poste__nom__icontains=q) |
            Q(poste__type_poste__nom__icontains=q) |
            Q(employe__nom__icontains=q) |
            Q(employe__prenom__icontains=q)
            )

        dossiers = rh.Dossier.objects.\
                select_related('poste', 'poste__type_poste').\
                actifs().\
                filter(filtre)
        q = q.lower()
        postes = list(set([d.poste for d in dossiers if q in
            unicode(d.employe).lower() or \
            q in d.poste.nom.lower() or \
            q in d.poste.type_poste.nom.lower()]))
        return postes

    def format_result(self, poste):
        q = self.q
        filtre = Q(poste=poste) & (
            Q(poste__nom__icontains=q) | Q(employe__nom__icontains=q) |
            Q(employe__prenom__icontains=q)
        )
        dossiers = rh.Dossier.objects.actifs().filter(filtre)

        nom_poste = poste.nom

        if len(dossiers) == 1:
            dossier = dossiers[0]
            employe = dossier.employe
        else:
            dossiers = poste.rh_dossiers.all().order_by('-date_debut')
            if len(dossiers) > 0:
                employe = unicode(dossiers[0].employe)
            else:
                employe = ""
        return "[%s] %s (%s) (%s)" % (
            poste.implantation.id, nom_poste, poste.id, employe
        )

    def format_item(self, poste):
        """
        the display of a currently selected object in the area below the
        search box. html is OK
        """
        return self.format_result(poste)

    def get_objects(self, ids):
        """
        given a list of ids, return the objects ordered as you would like
        them on the admin page.  this is for displaying the currently
        selected items (in the case of a ManyToMany field)
        """
        return rh.Poste.objects.filter(pk__in=ids)


class Dossier(object):

    def get_query(self, q, request):
        employe = groups.get_employe_from_user(request.user)
        prefixe_implantation = 'poste__implantation'

        q_recherche = (
            (Q(poste__nom__icontains=q) |
             Q(poste__type_poste__nom__icontains=q) |
             Q(employe__nom__icontains=q) |
             Q(employe__prenom__icontains=q))
            )

        q_place = Q(**{
            prefixe_implantation + '__zone_administrative':
            employe.implantation.zone_administrative
        })


        user_groupes = [g.name for g in request.user.groups.all()]
        if groups.DRH_NIVEAU_1 in user_groupes:
            q_filtre = q_recherche
        else:
            q_filtre = q_place & q_recherche
        return rh.Dossier.objects.ma_region_ou_service(request.user).filter(q_filtre).distinct()

    def format_result(self, dossier):
        return dossier.__unicode__()

    def format_item(self, dossier):
        """
        the display of a currently selected object in the area below the
        search box. html is OK
        """
        return self.format_result(dossier)

    def get_objects(self, ids):
        """
        given a list of ids, return the objects ordered as you would like
        them on the admin page.  this is for displaying the currently
        selected items (in the case of a ManyToMany field)
        """
        return rh.Dossier.objects.filter(pk__in=ids)


class Poste(object):

    def get_query(self, q, request):
        employe = groups.get_employe_from_user(request.user)
        prefixe_implantation = 'poste__implantation'

        q_recherche = (
            (Q(poste__nom__icontains=q) |
             Q(poste__type_poste__nom__icontains=q)) &
            (Q(date_fin=None) |
             Q(date_fin__gte=(
                    datetime.date.today() -
                    datetime.timedelta(365)
                    ))
              ))

        # if groups.is_user_dans_services_centraux(request.user):
        #     q_place = Q(**{prefixe_implantation: employe.implantation})
        # else:
        #     q_place = Q(**{
        #         prefixe_implantation + '__zone_administrative':
        #         employe.implantation.zone_administrative
        #     })


        # user_groupes = [g.name for g in request.user.groups.all()]
        # if groups.DRH_NIVEAU_1 in user_groupes or \
        #    groups.DRH_NIVEAU_2 in user_groupes:
        #     q_filtre = q_recherche
        # else:
        #     q_filtre = q_place & q_recherche

        q_filtre = q_recherche

        return rh.Dossier.objects.ma_region_ou_service(request.user).filter(q_filtre).order_by('-date_debut')

    def format_result(self, dossier):
        annee = dossier.date_debut.year
        if dossier.date_fin is not None:
            annee = dossier.date_fin.year
        return u"[%s] %s %s" % (
            dossier.poste.implantation, annee, dossier.poste.nom
        )

    def format_item(self, dossier):
        """
        the display of a currently selected object in the area below the
        search box. html is OK
        """
        return self.format_result(dossier)

    def get_objects(self, ids):
        """
        given a list of ids, return the objects ordered as you would like
        them on the admin page.  this is for displaying the currently
        selected items (in the case of a ManyToMany field)
        """
        return rh.Dossier.objects.filter(pk__in=ids)