Merge branch 'hotfix/another_fix_for_comparison' into dev

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

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

from datetime import date

from django.db.models import Q
from django.core.urlresolvers import reverse
import pygraphviz as pgv

from project.permissions import user_can_change_obj
from project.rh import models as rh


def bind_poste_to_graph(user, graph, postes_by_id):
    for n in graph.nodes():
        # poste = node
        poste = postes_by_id[int(n)]
        
        # dossiers actifs pour ce poste
        dossiers = rh.Dossier.objects.select_related('employe').filter(
                (Q(date_fin__gt=date.today()) | Q(date_fin=None)) &
                (Q(date_debut__lt=date.today()) | Q(date_debut=None)) &
                Q(poste__id=n)
        ).all()

        # affichage
        if dossiers:
            employes = "\\n".join(
                        ["[%s] %s %s" %
                            (d.employe_id, d.employe.nom.upper(),
                                d.employe.prenom
                            ) for d in dossiers]
                    )
            label = u"%s\\n%s\\n%s" % (
                d.poste.nom, employes, d.poste.implantation
            )
        else:
            label = u"%s\\n---\\n%s" % (poste.nom, poste.implantation)
            n.attr['fillcolor'] = 'azure4'
            n.attr['style'] = 'filled'

        n.attr['label'] = label.encode('ascii', 'xmlcharrefreplace')
        # lien seulement si user peut editer poste
        if user_can_change_obj(user, poste):
            n.attr['href'] = reverse("admin:rh_poste_change", args=(n,))

    return graph


def organigramme_postes_cluster(user, cluster_filter, titre=u"Organigramme",
                                cluster_titre=u"Cluster 1"):
    """
    Crée un organigramme des postes avec un cluster défini par le keyword
    qui sera rajouté au queryset en tant que filter.

    cluster_filter doit être un map de format (field: value) qui sera
    appliqué au queryset.

    Par exemple: cluster_filter={"service__exact": 19}
    """
    if type(titre) != type(unicode()):
        raise "Le titre du graphique doit être un unicode"
    if type(cluster_titre) != type(unicode()):
        raise "Le titre du cluster doit être un unicode"

    postes_by_id = dict((p.id, p) for p in rh.Poste.objects.all())

    postes = 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),
        **cluster_filter
    ).exclude(responsable=None)

    nom = titre.encode('ascii', 'xmlcharrefreplace')
    graph = pgv.AGraph(directed=True, name=nom)
    #pour mettre l'organigramme de gauche à droite (au lieu du haut à bas)
    #graph.rankdir='LR'
    graph_service = graph.subgraph(nbunch=[1, 2], name="cluster1")
    graph_service.graph_attr['style'] = 'filled'
    graph_service.graph_attr['color'] = 'lightgrey'
    graph_service.graph_attr['label'] = \
            cluster_titre.encode('ascii', 'xmlcharrefreplace')
    graph_service.graph_attr['labeljust'] = 'l'

    for p in postes:
        graph_service.add_node(p.id)
        if p.responsable_id:
            graph.add_edge(p.responsable_id, p.id)

    for p_id in graph_service.nodes():
        if postes_by_id[int(p_id)].responsable_id:
            poste_remontant = postes_by_id[int(p_id)]
            while poste_remontant.responsable_id \
                  and poste_remontant.responsable_id \
                  and poste_remontant.responsable_id != poste_remontant.id:
                poste_remontant = postes_by_id[poste_remontant.responsable_id]
                if poste_remontant.responsable_id:
                    graph.add_edge(
                        poste_remontant.responsable_id, poste_remontant.id
                    )

    bind_poste_to_graph(user, graph, postes_by_id)

    graph.layout(prog='dot')

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

    return svg