[#3165] Remplacé les filtres et protection par région pour des filtres et protections...

[auf_rh_dae.git] / project / dae / templatetags / dae.py

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

import os

from django import template

from project import groups 

from project.dae.workflow import ETATS_EDITABLE


register = template.Library()

@register.filter
def test_membre_drh(user):
    grps = [g.name for g in user.groups.all()]
    if groups.DRH_NIVEAU_1 in grps or groups.DRH_NIVEAU_2 in grps:
        return True
    else:
        return False

@register.filter
def peut_ajouter(user):
    for g in [g.name for g in user.groups.all()]:
        if g in (groups.ADMINISTRATEURS,
                groups.CORRESPONDANT_RH,
                groups.DRH_NIVEAU_1,
                groups.DRH_NIVEAU_2):
            return True
    return False

@register.filter
def peut_importer(user):
    return groups.DRH_NIVEAU_1 in [g.name for g in user.groups.all()]

@register.filter
def est_editable(obj, user):
    klass = obj.__class__
    groupes_users = [g.name for g in user.groups.all()]
    if obj.etat in ETATS_EDITABLE and \
       (obj in klass.objects.mes_choses_a_faire(user).all() or \
            groups.DRH_NIVEAU_1 in groupes_users or \
            groups.DRH_NIVEAU_2 in groupes_users):
        return True
    else:
        return False

@register.filter
def basename(path):
    return os.path.basename(path)

@register.inclusion_tag('dae/sort_header.html', takes_context=True)
def sort_header(context, field, title):
    """Génère une entête qu'on peut cliquer pour trier la colonne correspondante dans une table."""
    qs = context['request'].GET.copy()
    current = qs.get('tri', None)
    if current == field:
        cls = 'header headerSortUp'
        qs['tri'] = '-' + field
    elif current == '-' + field:
        cls = 'header headerSortDown'
        qs['tri'] = field
    else:
        cls = 'header'
        qs['tri'] = field
    return {'title': title, 'qs': qs.urlencode(), 'cls': cls}

@register.inclusion_tag('dae/pagination.html', takes_context=True)
def pagination(context, page):
    """Génère la navigation permettant de se promener de page en page."""
    qs = context['request'].GET
    previous_qs = qs.copy()
    previous_qs['page'] = page.previous_page_number()
    next_qs = qs.copy()
    next_qs['page'] = page.next_page_number()
    return {'page': page, 'previous_qs': previous_qs.urlencode(), 'next_qs': next_qs.urlencode()}