PEP8

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

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

from project.rh.groups import grp_drh, grp_drh2
from project.decorators import redirect_interdiction


def in_drh_or_admin(user):
    """
    Teste si un user Django fait parti du groupe DRH, DRH2 ou s'il est admin
    """
    groups = user.groups.all()
    if user.is_superuser or \
            grp_drh in groups or \
            grp_drh2 in groups:
        return True
    else:
        return False


def drh_or_admin_required(fn):
    """
    Teste si un user Django fait parti du groupe DRH, DRH2 ou s'il est admin
    """
    def inner(request, *args, **kwargs):
        user = request.user
        if in_drh_or_admin(user):
            return fn(request, *args, **kwargs)

        msg = u"Votre compte ne permet pas d'accéder à " \
                u"cette partie de l'application."
        return redirect_interdiction(request, msg)

    return inner