PEP8

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

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

import urllib
from datetime import date
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, get_object_or_404

from auf.django.references import models as ref

from project.decorators import drh_or_admin_required
from project.decorators import region_protected
from project.groups import get_employe_from_user
from project.groups import grp_drh, grp_correspondants_rh

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


@login_required
def profil(request):
    """Profil personnel de l'employé - éditable"""
    employe = get_employe_from_user(request.user)
    c = {
      'user': request.user,
      'employe': employe,
    }
    return render(request, 'rh/profil.html', c)


@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(request, 'rh/employes_liste.html', c)


@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(request, 'rh/employe.html', c)


@login_required
@drh_or_admin_required
def rapports_contrat(request):
    # statut default = actif?
    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

    contrats = rh.Contrat.objects.select_related(
        'dossier', 'dossier__poste', 'dossier__poste__implantation',
        'type_contrat', 'dossier__employe'
    )

    # filters et recherche temporelle
    cl = RechercheTemporelle(dict(request.GET.items()), rh.Contrat)
    lookup_params = get_lookup_params(request)
    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)

    # tri
    if 'o' in request.GET:
        contrats = contrats.order_by(
            ('-' if request.GET.get('ot') == "desc" else '') + request.GET['o']
        )

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

    headers = [
        ("dossier__employe__id", u"#"),
        ("dossier__employe__nom", u"Employé"),
        ("dossier", u"Dossier : Poste"),
        ("type_contrat__nom", u"Contrat"),
        ("date_debut", u"Début contrat"),
        ("date_fin", u"Fin contrat"),
        ("dossier__statut_residence", u"Statut"),
        ("dossier__poste__implantation__region", u"Région"),
        ("dossier__poste__implantation", u"Implantation"),
    ]
    h = SortHeaders(
        request, headers, order_field_type="ot", order_field="o",
        not_sortable=('dossier',)
    )

    c = {
        'cl': cl,
        'title': 'Rapport des contrats',
        'contrats': contrats,
        'count_employe': len(employes),
        'headers': list(h.headers()),
    }
    return render(request, 'rh/rapports/contrats.html', c)


@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(request, 'rh/rapports/masse_salariale.html', c)


@login_required
@drh_or_admin_required
def rapports_employes_sans_contrat(request):
    lookup_params = get_lookup_params(request)

    # contrats échus
    contrats_echus = rh.Contrat.objects.filter(
            date_fin__lt=date.today()
        ).filter(
            **lookup_params
        )

    # dossiers en cours sans contrat
    dossiers_sans_contrat = rh.Dossier.objects.filter(
            Q(date_fin=None) | Q(date_fin__gt=date.today()),
        ).exclude(
            date_debut__gt=date.today()
        ).filter(
            rh_contrats__in=contrats_echus
        )

    # employés sans contrat
    employes = rh.Employe.objects \
            .filter(rh_dossiers__in=dossiers_sans_contrat).distinct()
    if 'o' in request.GET:
        employes = employes.order_by(
            ('-' if request.GET.get('ot') == "desc" else '') + request.GET['o']
        )

    # affichage
    headers = [
        ("id", u"#"),
        ("nom", u"Employé"),
        ("dossier", u"Dossier : Poste"),
        ("dossier_date_debut", u"Début dossier"),
        ("dossier_date_fin", u"Fin dossier"),
    ]
    h = SortHeaders(
        request, headers, order_field_type="ot", order_field="o",
        not_sortable=('dossier', 'dossier_date_debut', 'dossier_date_fin')
    )

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

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


@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(request, 'rh/rapports/postes_modelisation.html', c)


@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(request, 'rh/rapports/postes_implantation.html', c)


@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(request, 'rh/rapports/postes_service.html', c)


@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(request, 'admin/rh/dossier/apercu.html', c)


@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(request, 'admin/rh/poste/apercu.html', c)


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(request, 'admin/rh/employe/apercu.html', c)


@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, content_type='image/svg+xml')
        response['Content-Disposition'] = \
                'attachment; filename=organigramme.svg'
        return response

    return render(request, 'rh/organigrammes/employe.html', c,
        content_type="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(request, 'rh/organigrammes/vide.html', c,
        content_type="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(request, 'rh/organigrammes/vide.html', c,
        content_type="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(request,
        'rh/organigrammes/vide.html', c,
        content_type="image/svg+xml"
    )