在 Django 模板中：检查用户是否有权访问由名称指定的视图

Question

假设我有一个使用 user_passes_test 装饰器装饰的视图：

# myapp/views.py
from django.views.generic import TemplateView
from django.contrib.auth.decorators import user_passes_test

def has_perm1_or_perm2(user):
    return user.has_perm('myapp.perm1') or user.has_perm('myapp.perm2')

@user_passes_test(has_perm1_or_perm2)
class MyView(TemplateView):
    # my view code goes here

然后我将它连接到一个 URL，如下所示：

# myapp/urls.py
from django.conf.urls import url
from . import views

urlpatterns = [
    ...,
    url(r'^myview$', views.MyView.as_view(), name='myview'),
    ...
]

如果我想在模板中放置一个指向该视图的链接，它看起来像：

<a href="{% url 'myapp:myview' %}">Check out my view!</a>

但我希望能够在模板中添加一些控制，以便仅当当前登录的用户被为视图指定的user_passes_test 函数批准时才会显示上述内容。

即我想让我的模板看起来像：

{% if check_user_has_view_permission request.user 'myapp:myview' %}
<a href="{% url 'myapp:myview' %}">Check out my view!</a>
{% else %}
Nothing to see here...
{% endif %}

我怎样才能做到这一点？

谢谢！

Answer 1

你可以写一个custom template filter来做这个：

# myapp/templatetags/my_app_tags.py (probably need a better name...)
from django import template
from myapp.views import has_perm1_or_perm2

register = template.Library()

@register.filter
user_has_special_perms(user):
    return has_perm1_or_perm2(user)

然后在你的模板中：

{% load my_app_tags %}

{% if request.user|user_has_special_perms %}
<a href="{% url 'myapp:myview' %}">Check out my view!</a>
{% else %}
Nothing to see here...
{% endif %}

您也可以像这样直接检查各个权限：

{% if perms.myapp.perm1 and perms.myapp.perm2 %}

（显然这不是 DRY）。

编辑 - 如何使它更通用？

为了使这个更通用，你可以做这样的事情。首先创建某种视图到权限的映射，例如：

# myapp/views.py
VIEW_PERMISSIONS = {
    'MyView': has_perm1_or_perm2,
    # etc for other views
}

然后像这样修改模板过滤器：

from myapp.views import VIEW_PERMISSIONS

@register.filter
user_has_special_perms(user, view_class):
    perm_func = VIEW_PERMISSIONS.get(view_class, None)
    if perm_func is not None:
        return perm_func(user)
    return false

然后在模板中：

{% if request.user|user_has_special_perms:"MyView" %}

我确信不需要通过VIEW_PERMISSIONS 进行显式映射，通过对视图类本身进行某种内省，就可以实现这一点。