project/dae/templatetags/dae.py

   1 # -*- encoding: utf-8 -*-
   2
   3 from django import template
   4 from project.dae.workflow import ETATS_EDITABLE
   5 from project.dae.workflow import grp_correspondants_rh, grp_administrateurs, grp_drh
   6
   7 register = template.Library()
   8
   9 @register.filter
  10 def test_membre_drh(user):
  11     return grp_drh in user.groups.all()
  12
  13 @register.filter
  14 def peut_ajouter(user):
  15     for g in user.groups.all():
  16         if g in (grp_administrateurs, grp_correspondants_rh, grp_drh):
  17             return True
  18     return False
  19
  20 @register.filter
  21 def est_editable(obj, user):
  22     klass = obj.__class__
  23     groupes_users = user.groups.all()
  24     if obj.etat in ETATS_EDITABLE and \
  25        (obj in klass.objects.mes_choses_a_faire(user).all() or grp_drh in groupes_users):
  26         return True
  27     else:
  28         return False
  29
  30 @register.filter
  31 def region_ou_service(implantation):
  32     if implantation.id == 15:
  33         return u"Services centraux de Montréal (SCM)"
  34     if implantation.id == 19:
  35         return u"Services centraux de Paris (SCP)"
  36     return implantation.region