Rapports trie, WIP et optimisation de requete SQL

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

# -*- encoding: utf-8 -*-
from datetime import date

from django.db.models import Q
from django.contrib.auth.decorators import login_required
from django.utils.encoding import smart_str
from django.shortcuts import redirect, render_to_response, get_object_or_404
from django.template import RequestContext
from django.http import Http404
from sendfile import sendfile

from datamaster_modeles import models as ref
from project.lib import get_employe_from_id

from rh import models as rh
from rh.lib import calc_remun
from rh.decorators import drh_or_admin_required
from rh.templatetags.rapports import SortHeaders

# pas de reference a DAE devrait etre refactorisé
from dae.utils import get_employe_from_user
from dae.decorators import redirect_interdiction
from dae.workflow import grp_drh, grp_correspondants_rh

@login_required
def piece(request, filename):
    """Téléchargement d'une pièce jointe à un poste."""
    model, id, f = filename.split('/')
    if model == 'employe':
        # TODO definir peut-être un controle d'accès
        piece = get_object_or_404(rh.EmployePiece, fichier=filename)
        return sendfile(request, piece.fichier.path)
    if model == 'poste':
        piece = get_object_or_404(rh.PostePiece, fichier=filename)
        if rh.Poste.objects.ma_region_ou_service(request.user).filter(id=piece.poste_id).exists():
            return sendfile(request, piece.fichier.path)
        else:
            return redirect_interdiction(request)

# homes
@login_required
def profil(request):
    """Profil personnel de l'employé - éditable"""
    rc = RequestContext(request)
    c = {}
    
    employe = rc['this_employe']
        
    c['user'] = request.user
    c['employe'] = employe
    return render_to_response('rh/profil.html', c, rc)
    
# employes
@login_required
def employes_liste(request):
    """Liste des employés."""
    today = date.today()
    employes = rh.Employe.objects   \
                    .filter(actif=True, supprime=False)   \
                    .filter(dossiers__actif=True, dossiers__supprime=False) \
                    .exclude(dossiers__date_debut__gt=today) \
                    .exclude(dossiers__date_fin__lt=today) \
                    .order_by('nom')
    c = {
        'user':request.user,
        'employes':employes,
        }
    return render_to_response('rh/employes_liste.html', c, RequestContext(request))
    
@login_required
def employe(request, id):
    """Information publique sur un employé."""
    try:
        employe = rh.Employe.objects.get(pk=id)
    except:
        employe = rh.Employe.objects.none()
    c = {
        'user':request.user,
        'employe':employe,
        }
    return render_to_response('rh/employe.html', c, RequestContext(request))


# Rapports

@login_required
@drh_or_admin_required
def rapports_poste(request):

    lookup_params = dict(request.GET.items())

    comble = 'all'

    for key, value in lookup_params.items():
        if key == 'o' or key == 'ot':
            del lookup_params[key]
            continue
        if not isinstance(key, str):
            # 'key' will be used as a keyword argument later, so Python
            # requires it to be a string.
            del lookup_params[key]
            lookup_params[smart_str(key)] = value

        if key == 'comble':
            comble = value
            del lookup_params[key]

    #postes = rh.Poste.actifs.select_related('implantation', 'dossier').all()
    postes = rh.Poste.actifs.select_related('implantation') \
            .extra(select={'employe_id':'select group_concat(employe) from rh_dossier where poste=rh_poste.id and rh_dossier.actif = 1'}) \
            .extra(select={'employe_nom':'select group_concat(rh_employe.nom) from rh_dossier inner join rh_employe on rh_dossier.employe = rh_employe.id where poste=rh_poste.id and rh_dossier.actif = 1'}) \
            .extra(select={'employe_prenom':'select group_concat(rh_employe.prenom) from rh_dossier inner join rh_employe on rh_dossier.employe = rh_employe.id where poste=rh_poste.id and rh_dossier.actif = 1'})

    postes = postes.filter(**lookup_params)
    if 'o' in request.GET:
        postes = postes.order_by("%s%s" % ('-' if 'ot' in request.GET and request.GET['ot'] == "desc" else '', request.GET['o']))

    headers = [
        ("id", u"# du poste"),
        ("nom", u"Nom du poste"),
        ("implantation__id", u"Implantation"),
        ("employe_id", u"# de l'employé"),
        ("employe_nom", u"Nom"),
        ("employe_prenom", u"Prénom"),
    ]
    h = SortHeaders(request, headers, order_field_type="ot", order_field="o")
    c = {
        'title': 'Rapport des postes',
        'postes': postes,
        'count': len(postes),
        'headers': list(h.headers()),
    }

    return render_to_response('rh/rapports/postes.html', c, RequestContext(request))

@login_required
@drh_or_admin_required
def rapports_contrat(request):

    lookup_params = dict(request.GET.items())
    if 'ot' in lookup_params:
        del lookup_params['ot']
    if 'o' in lookup_params:
        del lookup_params['o']

    for key, value in lookup_params.items():
        if not isinstance(key, str):
            # 'key' will be used as a keyword argument later, so Python
            # requires it to be a string.
            del lookup_params[key]
            lookup_params[smart_str(key)] = value

    contrats = rh.Contrat.objects.select_related('dossier', 'dossier__poste', 'dossier__poste__implantation', 'type_contrat', 'dossier__employe')
    contrats = contrats.filter(**lookup_params)
    if 'o' in request.GET:
        contrats = contrats.order_by("%s%s" % ('-' if 'ot' in request.GET and request.GET['ot'] == "desc" else '', request.GET['o']))

    employes = set([c.dossier.employe_id for c in contrats])

    headers = [
        ("dossier__employe__id", u"# de l'employé"),
        ("dossier__employe__nom", u"Nom"),
        ("dossier__employe__prenom", u"Prénom"),
        ("type_contrat__nom", u"Poste"),
        ("type_contrat__nom", u"Type de contrat"),
        ("date_debut", u"Date début"),
        ("date_fin", u"Date fin"),
    ]
    h = SortHeaders(request, headers, order_field_type="ot", order_field="o")

    c = {
        'title': 'Rapport des contrats',
        'contrats': contrats,
        'count': len(contrats),
        'count_employe' : len(employes),
        'headers': list(h.headers()),
    }

    return render_to_response('rh/rapports/contrats.html', c, RequestContext(request))


@login_required
@drh_or_admin_required
def rapports_remuneration(request):

    lookup_params = dict(request.GET.items())

    for key, value in lookup_params.items():
        if not isinstance(key, str):
            # 'key' will be used as a keyword argument later, so Python
            # requires it to be a string.
            del lookup_params[key]
            lookup_params[smart_str(key)] = value

    employes = rh.Employe.objects.all()
    employes = employes.filter(**lookup_params)

    output = []

    for employe in employes:
        line = {}
        output.append(line)

        dossiers = employe.rh_dossiers.all()

        remun = {}
        remun_sum_euro = 0

        for dossier in dossiers:
            this_remun, this_remun_sum, this_remun_sum_euro = calc_remun(dossier)

            for item in this_remun:
                if item not in remun:
                    remun[item] = this_remun[item]
                else:
                    remun[item][0] += this_remun[item][0]
                    remun[item][1] += this_remun[item][1]

            remun_sum_euro += this_remun_sum_euro

        line['remun_sum_euro'] = remun_sum_euro

        for r in remun:
            if r == u'Indemnité':
                line['Indemnite'] = remun[r][1]
            else:
                line[r] = remun[r][1]

        line['id'] = employe.id
        line['nom'] = employe.nom
        line['prenom'] = employe.prenom


    c = {
        'title': 'Rapport de remuneration',
        'employes': output,
    }

    return render_to_response('rh/rapports/remuneration.html', c, RequestContext(request))

def region_protected(model):
    def wrapper(func):
        def wrapped(request, id):
            if request.user.is_superuser:
                return func(request, id)
            user_groups = request.user.groups.all()
            if grp_drh in user_groups:
                return func(request, id)
            if grp_correspondants_rh in user_groups:
                employe = get_employe_from_user(request.user)
                q = Q(**{model.prefix_implantation: employe.implantation.region})
                qs = model.objects.filter(q)
                if int(id) in [o.id for o in qs]:
                    return func(request, id)
            return redirect_interdiction(request)
        return wrapped
    return wrapper


@region_protected(rh.Dossier)
def dossier_apercu(request, dossier_id):
    c = {
        'is_popup' : request.GET.get('_popup', False),
        'dossier' : get_object_or_404(rh.Dossier, pk=dossier_id)

    }
    return render_to_response('admin/rh/dossier/apercu.html', c, RequestContext(request))

@region_protected(rh.Poste)
def poste_apercu(request, poste_id):
    c = {
        'is_popup' : request.GET.get('_popup', False),
        'poste' : get_object_or_404(rh.Poste, pk=poste_id)

    }
    return render_to_response('admin/rh/poste/apercu.html', c, RequestContext(request))

def employe_apercu(request, employe_id):
    from dae.workflow import grp_drh, grp_correspondants_rh
    employe = get_object_or_404(rh.Employe, pk=employe_id)
    user_groups = request.user.groups.all()

    dernier_dossier = None
    dossiers = None

    if request.user.is_superuser or \
       grp_drh in user_groups:
        q = Q(employe=employe)
    if grp_correspondants_rh in user_groups:
        regions = [d.poste.implantation.region for d in employe.rh_dossiers.all()]
        q = Q(employe=employe) & Q(implantation__region__in=regions)
     
    dossiers = rh.Dossier.objects.filter(employe=employe).order_by('-date_debut')
      
    c = {
        'is_popup' : request.GET.get('_popup', False),
        'employe' : employe,
        'dossiers' : dossiers,
    }
    return render_to_response('admin/rh/employe/apercu.html', c, RequestContext(request))