fix import

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

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

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
        postes = rh.Poste.objects.filter(
            Q(nom__icontains=q) |
            Q(type_poste__nom__icontains=q) |
            Q(rh_dossiers__employe__nom__icontains=q) |
            Q(rh_dossiers__employe__prenom__icontains=q)
            ).distinct()
        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.filter(filtre)

        nom_poste = poste.nom

        if len(dossiers) == 1:
            dossier = dossiers[0]
            employe = dossier.employe
        else:
            dossiers = poste.rh_dossiers.all()
            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)

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

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

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

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