project/rh/graph.py

   1 # -*- encoding: utf-8 -*-
   2
   3 import pygraphviz as pgv
   4
   5 from datetime import date
   6 from django.db.models import Q
   7 from django.core.urlresolvers import reverse
   8
   9 from rh import models as rh
  10
  11 def bind_poste_to_graph(graph, postes_by_id):
  12     for n in graph.nodes():
  13         dossiers = rh.Dossier.objects.select_related('employe').filter(
  14                 (Q(date_fin__gt=date.today()) | Q(date_fin=None)) &
  15                 (Q(date_debut__lt=date.today()) | Q(date_debut=None)) &
  16                 Q(poste__id=n)
  17         ).exclude(supprime=True).all()
  18
  19
  20         if dossiers:
  21             employes = "\\n".join(
  22                         ["[%s] %s %s" %
  23                             (d.employe_id, d.employe.nom.upper(),
  24                                 d.employe.prenom
  25                             ) for d in dossiers]
  26                     )
  27             label = u"%s\\n%s\\n%s" % (d.poste.nom, employes, d.poste.implantation)
  28         else:
  29             label = u"%s\\n---\\n%s" % (d.poste.nom, d.poste.implantation)
  30             n.attr['fillcolor'] = 'azure4'
  31             n.attr['style'] = 'filled'
  32
  33         n.attr['label'] = label.encode('ascii', 'xmlcharrefreplace')
  34         n.attr['href'] = reverse("admin:rh_poste_change", args=(n,))
  35
  36     return graph
  37
  38 def organigramme_postes_cluster(cluster_filter, titre=u"Organigramme", cluster_titre=u"Cluster 1"):
  39     """ Crée un organigramme des postes avec un cluster défini par le keyword qui sera
  40     rajouté au queryset en tant que filter.
  41     cluster_filter doit être un map de format (field: value) qui sera appliqué au queryset.
  42     Par exemple: cluster_filter={"service__exact": 19}
  43     """
  44     if type(titre) != type(unicode()):
  45         raise "Le titre du graphique doit être un unicode"
  46     if type(cluster_titre) != type(unicode()):
  47         raise "Le titre du cluster doit être un unicode"
  48
  49     postes_by_id = dict((p.id, p) for p in rh.Poste.objects.all())
  50
  51     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)) ).filter(**cluster_filter).exclude(supprime=True, responsable=None).all()
  52
  53     nom = titre.encode('ascii', 'xmlcharrefreplace')
  54     graph = pgv.AGraph(directed=True, name=nom)
  55     #pour mettre l'organigramme de gauche à droite (au lieu du haut à bas)
  56     #graph.rankdir='LR'
  57     graph_service = graph.subgraph(nbunch=[1,2], \
  58             name="cluster1", \
  59     #        style='filled', \
  60     #        color='lightgrey', \
  61     #        label=cluster_titre.encode('ascii', 'xmlcharrefreplace'),
  62     #        labeljust="l")
  63     )
  64     graph_service.graph_attr['style'] = 'filled'
  65     graph_service.graph_attr['color'] = 'lightgrey'
  66     graph_service.graph_attr['label'] = cluster_titre.encode('ascii', 'xmlcharrefreplace')
  67     graph_service.graph_attr['labeljust'] = 'l'
  68
  69     for p in postes:
  70         graph_service.add_node(p.id)
  71         if p.responsable_id:
  72             graph.add_edge(p.responsable_id, p.id)
  73
  74     for p_id in graph_service.nodes():
  75         if postes_by_id[int(p_id)].responsable_id:
  76             poste_remontant = postes_by_id[int(p_id)]
  77             while poste_remontant.responsable_id and poste_remontant.responsable_id and poste_remontant.responsable_id != poste_remontant.id:
  78                 poste_remontant = postes_by_id[poste_remontant.responsable_id]
  79                 if poste_remontant.responsable_id:
  80                     graph.add_edge(poste_remontant.responsable_id, poste_remontant.id)
  81
  82     bind_poste_to_graph(graph, postes_by_id)
  83
  84     graph.layout(prog='dot')
  85
  86     svg = graph.draw(format='svg')
  87
  88     return svg
  89
  90