Version

[auf_references_client.git] / auf_references_client / decorators.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-

from django.http import HttpResponseForbidden
from django.contrib.auth import logout
from django.template.loader import get_template
from django.template import Context

def has_group(group = ""):
    def _dec(view_func):
        def _view(request, *args, **kwargs):
            allowed = False
            if request.user.is_authenticated():
                try:
                    p = personne(request.user)
                    g = Groupe.objects.get(nom = group)
                    if g in p.groupes:
                        allowed = True
                except: pass
            else:
                allowed = True
    
            if allowed:
                r = view_func(request, *args, **kwargs)
            else:
                logout(request)
                t = get_template('references_403.html')
                html = t.render(Context({}))
                r = HttpResponseForbidden(html)

            return r

        _view.__name__ = view_func.__name__
        _view.__dict__ = view_func.__dict__
        _view.__doc__ = view_func.__doc__

        return _view
    return _dec