[#3165] Remplacé les filtres et protection par région pour des filtres et protections...

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

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

from django.db.models import Q

from auf.django.references import models as ref

from project.groups import get_employe_from_user
from project.permissions import user_gere_obj_de_sa_region
from project.rh import models as rh


class Pays(object):
    fk_key = "code"

    def get_query(self, q, request):
        pays = ref.Pays.objects.filter(
            Q(nom__icontains=q) | Q(code__icontains=q)
        )
        return pays

    def format_result(self, pays):
        return unicode(pays)

    def format_item(self, pays):
        return self.format_result(pays)

    def get_objects(self, ids):
        return ref.Pays.objects.filter(code__in=ids)


class Implantation(object):

    def get_query(self, q, request):
        implantations = ref.Implantation.objects.filter(
            Q(nom__icontains=q) | Q(nom_court__icontains=q) |
            Q(nom_long__icontains=q) | Q(zone_administrative__nom__icontains=q)
        )
        if user_gere_obj_de_sa_region(request.user):
            employe = get_employe_from_user(request.user)
            implantations = implantations.filter(
                zone_administrative=employe.implantation.zone_administrative
            )
        return implantations

    def format_result(self, implantation):
        statut = implantation.statut
        if implantation.statut == 0:
            statut = u"Jamais ouverte OU fermée"
        if implantation.statut == 1:
            statut = u"Ouverte"
        if implantation.statut == 2:
            statut = u"Fermée"
        if implantation.statut == 3:
            statut = u"Ouverte imminente"

        return u"%s (%s) [%s]" % (implantation.nom, implantation.id, statut)

    def format_item(self, implantation):
        return self.format_result(implantation)

    def get_objects(self, ids):
        return ref.Implantation.objects.filter(id__in=ids)


class TypePoste(object):

    def get_query(self, q, request):
        typepostes = rh.TypePoste.objects.filter(
            Q(nom__icontains=q) | Q(nom_feminin__icontains=q) |
            Q(categorie_emploi__nom__icontains=q)
        )
        return typepostes

    def format_result(self, typeposte):
        return unicode(typeposte)

    def format_item(self, typeposte):
        return self.format_result(typeposte)

    def get_objects(self, ids):
        return rh.TypePoste.objects.filter(id__in=ids)


class Poste(object):

    def get_query(self, q, request):
        postes = rh.Poste.objects.ma_region_ou_service(request.user).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)
        )
        return postes.distinct()

    def format_result(self, poste):
        return unicode(poste)

    def format_item(self, poste):
        return self.format_result(poste)

    def get_objects(self, ids):
        return rh.Poste.objects.filter(id__in=ids)


class ValeurPoint(object):

    def get_query(self, q, request):
        points = rh.ValeurPoint.objects \
                .select_related('devise', 'implantation') \
                .filter(
                    Q(devise__code__icontains=q) |
                    Q(implantation__nom__icontains=q)
                )
        return points

    def format_result(self, point):
        return unicode(point)

    def format_item(self, point):
        return self.format_result(point)

    def get_objects(self, ids):
        return rh.ValeurPoint.objects.filter(id__in=ids)


class Employe(object):

    def get_query(self, q, request):
        employes = rh.Employe.objects.filter(
            Q(nom__icontains=q) | Q(prenom__icontains=q) |
            Q(nom_affichage__icontains=q)
        )
        return employes

    def format_result(self, employe):
        return unicode(employe)

    def format_item(self, employe):
        return self.format_result(employe)

    def get_objects(self, ids):
        return rh.Employe.objects.filter(id__in=ids)


class Dossier(object):

    def get_query(self, q, request):
        dossiers = rh.Dossier.objects.filter(
            Q(poste__nom=q) | Q(poste__type_poste__nom=q) |
            Q(employe__nom__icontains=q) | Q(employe__prenom__icontains=q) |
            Q(employe__nom_affichage__icontains=q)
        )
        return dossiers

    def format_result(self, dossier):
        return unicode(dossier)

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

    def get_objects(self, ids):
        return rh.Dossier.objects.filter(id__in=ids)