Fix django ajax selects

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

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

from django.db.models import Q
from rh import models as rh
from utils import get_employe_from_user, is_user_dans_services_centraux
from workflow import grp_drh

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 = 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 is_user_dans_services_centraux(request.user):
            q_place = Q(**{ '%s' % prefixe_implantation : employe.implantation })
        else:
            q_place = Q(**{ '%s__region' % prefixe_implantation : employe.implantation.region })


        if grp_drh in request.user.groups.all():
            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 = 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 is_user_dans_services_centraux(request.user):
            q_place = Q(**{ '%s' % prefixe_implantation : employe.implantation })
        else:
            q_place = Q(**{ '%s__region' % prefixe_implantation : employe.implantation.region })


        if grp_drh in request.user.groups.all():
            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)