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

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

import urllib
from datetime import date
from itertools import izip
import StringIO

import pygraphviz as pgv

from django import forms
from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.core.servers.basehttp import FileWrapper
from django.core.urlresolvers import reverse
from django.db.models import Q
from django.http import HttpResponse
from django.shortcuts import render_to_response, get_object_or_404
from django.template import RequestContext
from django.utils.encoding import smart_str

from datamaster_modeles import models as ref

from project.rh import models as rh
from project.rh.lib import calc_remun
from project.rh.decorators import drh_or_admin_required
from project.rh.templatetags.rapports import SortHeaders
from project.rh.change_list import RechercheTemporelle
from project.rh import graph as rh_graph
from project.rh.masse_salariale import MasseSalariale

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


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


@login_required
def employes_liste(request):
    """Liste des employés."""
    today = date.today()
    employes = rh.Employe.objects   \
                    .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))


@login_required
@drh_or_admin_required
def rapports_poste(request):

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

    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':
            del lookup_params[key]

    sort_order = 'ASC' if request.GET.get('ot') == 'asc' else 'DESC'

    postes = rh.Poste.objects.select_related('implantation') \
            .extra(select={
                'employe_id':
                'SELECT GROUP_CONCAT(employe SEPARATOR "|") '
                'FROM rh_dossier '
                'WHERE poste = rh_poste.id '
                'AND rh_dossier.date_fin IS NULL ' + (
                    'ORDER BY employe ' + sort_order
                    if request.GET.get('o') == 'employe_id' else ''
                )
            }) \
            .extra(select={
                'employe_nom':
                'SELECT GROUP_CONCAT(rh_employe.nom SEPARATOR "|") '
                'FROM rh_dossier INNER JOIN rh_employe '
                'ON rh_dossier.employe = rh_employe.id '
                'WHERE poste = rh_poste.id '
                'AND rh_dossier.date_fin IS NULL ' + (
                    'ORDER BY rh_employe.nom ' + sort_order
                    if request.GET.get('o') == 'employe_nom' else ''
                )
            }) \
            .extra(select={
                'employe_prenom':
                'SELECT GROUP_CONCAT(rh_employe.prenom SEPARATOR "|") '
                'FROM rh_dossier INNER JOIN rh_employe '
                'ON rh_dossier.employe = rh_employe.id '
                'WHERE poste = rh_poste.id '
                'AND rh_dossier.date_fin IS NULL ' + (
                    'ORDER BY rh_employe.prenom ' + sort_order
                    if request.GET.get('o') == 'employe_prenom' else ''
                )
            })

    postes = postes.filter(**lookup_params)
    if 'o' in request.GET:
        postes = postes.order_by(
            ('-' if sort_order == "DESC" else '') + request.GET['o']
        )

    out = []
    for p in postes:
        out.append({
            'id': p.id,
            'nom': p.nom,
            'implantation': p.implantation,
            'employes': [] if not p.employe_id else [
                {'id': id, 'nom': nom, 'prenom': prenom}
                for id, nom, prenom in izip(
                    p.employe_id.split('|'),
                    p.employe_nom.split('|'),
                    p.employe_prenom.split('|')
                )
            ]
        })

    headers = [
        ("id", u"# du poste"),
        ("nom", u"Nom du poste"),
        ("implantation__id", u"Implantation"),
        ("employe_nom", u"Nom"),
    ]
    h = SortHeaders(request, headers, order_field_type="ot", order_field="o")
    c = {
        'title': 'Rapport des postes',
        'postes': out,
        'count': len(out),
        '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):
    if 'HTTP_REFERER' in request.META.keys():
                referer = request.META['HTTP_REFERER']
                referer = "/".join(referer.split('/')[3:])
                referer = "/%s" % referer.split('?')[0]
                if referer != reverse('rhr_contrats'):
                    params = request.GET.copy()
                    params.update({'statut': 'Actif'})
                    request.GET = params

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

    cl = RechercheTemporelle(dict(request.GET.items()), rh.Contrat)
    lookup_params = cl.purge_params(lookup_params)
    q_temporel = cl.get_q_temporel(contrats)
    q = Q(**lookup_params) & q_temporel
    contrats = contrats.filter(q).exclude(dossier__employe__supprime=1)

    if 'o' in request.GET:
        contrats = contrats.order_by(
            ('-' if request.GET.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"Employé"),
        ("type_contrat__nom", u"Poste"),
        ("dossier__poste__implantation__region", u"Région"),
        ("dossier__poste__implantation", u"Implantation"),
        ("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 = {
        'cl': cl,
        '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_masse_salariale(request):

    class RechercheTemporelle(forms.Form):
        CHOICE_ANNEES = range(
            rh.Remuneration.objects.exclude(date_debut=None)
            .order_by('date_debut')[0].date_debut.year,
            date.today().year + 1
        )

        annee = forms.CharField(
            initial=date.today().year,
            widget=forms.Select(
                choices=((a, a) for a in reversed(CHOICE_ANNEES))
            )
        )

        region = forms.CharField(
                widget=forms.Select(choices=[('', '')] +
                    [(i.id, i) for i in ref.Region.objects.all()]
                )
            )

        implantation = forms.CharField(
                widget=forms.Select(choices=[('', '')] +
                    [(i.id, i) for i in ref.Implantation.objects.all()]
                )
            )

        #date_debut = forms.DateField(widget=adminwidgets.AdminDateWidget)
        #date_fin = forms.DateField(widget=adminwidgets.AdminDateWidget)

    form = RechercheTemporelle(request.GET)
    get_filtre = [
        (k, v) for k, v in request.GET.items()
        if k not in ('date_debut', 'date_fin', 'implantation')
    ]
    query_string = urllib.urlencode(get_filtre)

    date_debut = None
    date_fin = None
    if request.GET.get('annee', None):
        date_debut = "01-01-%s" % request.GET.get('annee', None)
        date_fin = "31-12-%s" % request.GET.get('annee', None)

    implantation = request.GET.get('implantation')
    region = request.GET.get('region')

    custom_filter = {}
    if implantation:
        custom_filter['dossier__poste__implantation'] = implantation
    if region:
        custom_filter['dossier__poste__implantation__region'] = region

    c = {
            'title': 'Rapport de masse salariale',
            'form': form,
            'headers': [],
            'query_string': query_string,
    }
    if date_debut or date_fin:
        masse = MasseSalariale(date_debut, date_fin, custom_filter,
                request.GET.get('ne_pas_grouper', False))
        if masse.rapport:
            if request.GET.get('ods'):
                for h in (
                    h for h in masse.headers if 'background-color' in h[2]
                ):
                    del h[2]['background-color']
                masse.ods()
                output = StringIO.StringIO()
                masse.doc.save(output)
                output.seek(0)

                response = HttpResponse(
                    FileWrapper(output),
                    content_type=(
                        'application/vnd.oasis.opendocument.spreadsheet'
                    )
                )
                response['Content-Disposition'] = \
                        'attachment; filename=Masse Salariale %s.ods' % \
                            masse.annee
                return response
            else:
                c['rapport'] = masse.rapport
                c['header_keys'] = [h[0] for h in masse.headers]
                #on enleve le background pour le header
                for h in (
                    h for h in masse.headers if 'background-color' in h[2]
                ):
                    h[2]['background'] = 'none'
                h = SortHeaders(request, masse.headers, order_field_type="ot",
                        not_sortable=c['header_keys'], order_field="o")
                c['headers'] = list(h.headers())
                for key, nom, opts in masse.headers:
                    c['headers']
                c['total'] = masse.grand_totaux[0]
                c['total_euro'] = masse.grand_totaux[1]
                c['colspan'] = len(c['header_keys']) - 1
                get_filtre.append(('ods', True))
                query_string = urllib.urlencode(get_filtre)
                c['url_ods'] = "%s?%s" % (
                        reverse('rhr_masse_salariale'), query_string)

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


@login_required
@drh_or_admin_required
def rapports_remuneration(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

    employes = rh.Employe.objects.all()
    if 'o' in request.GET:
        employes = employes.order_by(
            ('-' if request.GET.get('ot') == "desc" else '') + request.GET['o']
        )

    employes = employes.filter(**lookup_params)

    output = []
    headers = [
            ("id", u"# de l'employé"),
            ("nom", u"Nom"),
            ("prenom", u"Prénom"),
            ("", u"Salaire"),
            ("", u"RAS"),
            ("", u"Indemnités"),
            ("", u"Accessoire"),
            ("", u"Charges patronales"),
            ("", u"Total"),
    ]
    h = SortHeaders(request, headers, order_field_type="ot", order_field="o")

    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,
        'headers': list(h.headers()),
    }

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


@login_required
@drh_or_admin_required
def rapports_employe_sans_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

    employes_query = rh.Employe.objects
    if 'o' in request.GET:
        employes_query = employes_query.order_by(
            ('-' if request.GET.get('ot') == "desc" else '') + request.GET['o']
        )

    employes = {}

    dossiers_en_cours = rh.Dossier.objects.filter(
        Q(date_fin=None) | Q(date_fin__gt=date.today())
    )
    tous_contrats_echus = rh.Contrat.objects.filter(
        date_fin__lt=date.today(), dossier__in=dossiers_en_cours
    )
    contrats = tous_contrats_echus.filter(**lookup_params).all()
    for c in contrats:
        if c.dossier.employe.id not in employes.keys():
            employes[c.dossier.employe.id] = {
                'employe': c.dossier.employe,
                'dossiers': []
            }
        employes[c.dossier.employe.id]['dossiers'] += [c.dossier]

    headers = [
        ("id", u"# de l'employé"),
        ("nom", u"Nom"),
        ("prenom", u"Prénom"),
        ("dossier", u"Dossiers"),
    ]
    h = SortHeaders(
        request, headers, order_field_type="ot", order_field="o",
        not_sortable=('dossier',)
    )

    c = {
        'title': u'Rapport des employés sans contrat',
        'employes': employes,
        'count': len(employes),
        'headers': list(h.headers()),
    }

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


@login_required
@drh_or_admin_required
def rapports_postes_modelisation(request):
    c = {}
    data = []

    for categorie in rh.CategorieEmploi.objects.all():
        types = rh.TypePoste.objects.filter(categorie_emploi=categorie)
        data_types = []
        for t in types.all():
            postes = rh.Poste.objects.filter(type_poste=t)
            data_types.append({
                'num_postes': postes.count(),
                'postes': postes.all(),
                'type': categorie,
                })

        data.append({
            'categorie': categorie,
            'nb_types': types.count(),
            'types': data_types
        })

    c['data'] = data

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


@login_required
@drh_or_admin_required
def rapports_postes_implantation(request):
    c = {}
    data = []
    for r in ref.Region.objects.all():
        implantations = []
        for i in ref.Implantation.objects.filter(region=r):
            implantations.append({
                'implantation': i,
                'postes': rh.Poste.objects.filter(implantation=i),
                'num_postes': rh.Poste.objects.filter(implantation=i).count(),
                })
        data.append({
            'region': r,
            'implantations': implantations
            })

    c['data'] = data

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


@login_required
@drh_or_admin_required
def rapports_postes_service(request):
    c = {}
    data = []
    for s in rh.Service.objects.all():
        postes = rh.Poste.objects.filter(service=s).all()
        num_postes = rh.Poste.objects.filter(service=s).count()
        data.append({'service': s, 'num_postes': num_postes, 'postes': postes})

    c['data'] = data
    return render_to_response(
        'rh/rapports/postes_service.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):
    d = get_object_or_404(rh.Dossier, pk=dossier_id)
    c = {
        'title': u"Dossier %s" % (d, ),
        'is_popup': request.GET.get('_popup', False),
        'dossier': d,
        'pieces':  rh.DossierPiece.objects.filter(dossier__exact=d),
        'contrats': rh.Contrat.objects.filter(dossier__exact=d),
        'commentaires': rh.DossierCommentaire.objects.filter(dossier=d).all(),
        'media_url': settings.PRIVE_MEDIA_URL,
    }
    return render_to_response(
        'admin/rh/dossier/apercu.html', c, RequestContext(request)
    )


@region_protected(rh.Poste)
def poste_apercu(request, poste_id):
    p = get_object_or_404(rh.Poste, pk=poste_id)
    c = {
        'title': u"Poste %s" % (p, ),
        'is_popup': request.GET.get('_popup', False),
        'poste': p,
        'financements': (
            rh.PosteFinancement.objects.filter(poste=poste_id).all()
        ),
        'pieces': rh.PostePiece.objects.filter(poste=poste_id).all(),
        'dossiers': (
            rh.Dossier.objects.filter(poste=poste_id)
            .order_by("-date_debut").all()
        ),
        'comparaisons': (
            rh.PosteComparaison.objects.filter(poste=poste_id).all()
        ),
        'commentaires': (
            rh.PosteCommentaire.objects.filter(poste=poste_id).all()
        ),
        'media_url': settings.PRIVE_MEDIA_URL,
    }
    return render_to_response(
        'admin/rh/poste/apercu.html', c, RequestContext(request)
    )


def employe_apercu(request, employe_id):
    employe = get_object_or_404(rh.Employe, pk=employe_id)
    user_groups = request.user.groups.all()
    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(q).order_by('-date_debut')

    c = {
        'title': u"Employe %s" % (employe, ),
        'is_popup': request.GET.get('_popup', False),
        'employe': employe,
        'dossiers': dossiers,
        'media_url': settings.PRIVE_MEDIA_URL,
    }
    return render_to_response(
        'admin/rh/employe/apercu.html', c, RequestContext(request)
    )


@login_required
@drh_or_admin_required
def organigrammes_employe(request, id, level="all"):

    poste = get_object_or_404(rh.Poste, pk=id)
    dossiers_by_poste = dict(
        (d.poste_id, d)
        for d in rh.Dossier.objects.select_related('employe', 'poste').all()
    )
    postes_by_id = dict((p.id, p) for p in rh.Poste.objects.all())

    e = dossiers_by_poste[poste.id].employe
    name = u"Organigramme de [%s] %s %s" % (e.id, e.nom.upper(), e.prenom)
    graph = pgv.AGraph()

    if rh.Poste.objects.filter(responsable=poste).count() > 0:
        postes_handle = [poste]
        while postes_handle:
            postes_handle = rh.Poste.objects.select_related('implantation') \
                    .filter(
                        Q(date_fin__gt=date.today()) | Q(date_fin=None),
                        Q(date_debut__lt=date.today()) | Q(date_debut=None),
                        responsable__in=postes_handle
                    ).exclude(supprime=True).exclude(responsable=None).all()

            for p in postes_handle:
                if p.responsable_id != p.id:
                    graph.add_edge(
                        dossiers_by_poste[p.responsable_id].poste_id, p.id
                    )

    else:
        graph.add_node(poste.id)

    if level != "all":
        postes_niveau = [poste.id]
        for niveau in range(int(level)):
            postes_niveau = [
                p.id for p in
                rh.Poste.objects.filter(responsable__in=postes_niveau).all()
            ]

        while postes_niveau:
            postes_niveau = [
                p.id for p in
                rh.Poste.objects.filter(responsable__in=postes_niveau).all()
            ]
            if postes_niveau:
                for p in postes_niveau:
                    if graph.has_node(p):
                        graph.delete_node(p)

    a = graph
    a.name = name.encode('ascii', 'xmlcharrefreplace')

    poste_remontant = poste
    while poste_remontant.responsable_id:
        a.add_edge(poste_remontant.responsable_id, poste_remontant.id)
        poste_remontant = poste_remontant.responsable

    rh_graph.bind_poste_to_graph(a, postes_by_id)
    #a.graph_attr['normalize'] = True
    #a.graph_attr['level'] = 2
    a.layout(prog='dot')

    svg = a.draw(format='svg')

    c = {
        'svg': svg
    }

    if 'forcer' in request.GET:
        response = HttpResponse(svg, mimetype='image/svg+xml')
        response['Content-Disposition'] = \
                'attachment; filename=organigramme.svg'
        return response

    return render_to_response(
        'rh/organigrammes/employe.html', c, RequestContext(request),
        mimetype="image/svg+xml"
    )


@login_required
@drh_or_admin_required
def organigrammes_service(request, id):

    service = get_object_or_404(rh.Service, pk=id)
    svg = rh_graph.organigramme_postes_cluster( \
            cluster_filter={"service": service}, \
            titre=u"Organigramme du service %s" % service.nom,
            cluster_titre=service.nom)

    c = {
        'svg': svg
    }

    return render_to_response(
        'rh/organigrammes/vide.html', c, RequestContext(request),
        mimetype="image/svg+xml"
    )


@login_required
@drh_or_admin_required
def organigrammes_implantation(request, id):

    implantation = get_object_or_404(ref.Implantation, pk=id)
    svg = rh_graph.organigramme_postes_cluster( \
            cluster_filter={"implantation": implantation}, \
            titre=u"Organigramme de l'implantation %s" % implantation.nom,
            cluster_titre=implantation.nom)

    c = {
        'svg': svg
    }

    return render_to_response(
        'rh/organigrammes/vide.html', c, RequestContext(request),
        mimetype="image/svg+xml"
    )


@login_required
@drh_or_admin_required
def organigrammes_region(request, id):

    region = get_object_or_404(ref.Region, pk=id)
    svg = rh_graph.organigramme_postes_cluster( \
            cluster_filter={"implantation__region": region}, \
            titre=u"Organigramme du bureau de %s" % region.nom,
            cluster_titre=region.nom)

    c = {
        'svg': svg
    }

    return render_to_response(
        'rh/organigrammes/vide.html', c, RequestContext(request),
        mimetype="image/svg+xml"
    )