如何编写 Django 模板标签来进行访问控制？

Question

我徒劳地尝试创建一个简单的 Django 模板标签来显示或隐藏我网站上提交的评论旁边的“删除”链接。

简而言之，我想将评论对象传递给模板标签，确定当前登录的用户是否有权删除评论，然后显示或不显示链接。

我的模板中的用法如下：

{% load access_tags %}
{% if_authorized comment %}
   <a href="{% url delete_comment comment.id %}">Delete</a>
{% endif_authorized %}

请放心，如果用户有权删除评论，我还会检查相应的视图。

这种类型的标签有具体的名称吗？如果是的话，它肯定会对我的谷歌搜索有所帮助。感谢您的帮助！

更新 1：

根据我网站的工作方式，两个人可能有权删除评论：1) 评论创建者和 2) 留下评论的帖子的所有者。因此，我需要根据评论确定是否存在这些条件之一。

我不认为我可以使用 Django 的内置权限系统之类的东西，因为它要求权限“按对象类型全局设置，而不是按特定对象实例设置”。

就我而言，用户“Bob”可能有权删除评论（如果他写了评论或评论位于他创建的帖子上），但也可能不允许他删除评论（如果他正在查看对某人的评论）别人的帖子）。

更新2：

看来您无法将对象传递给模板标签，只能传递字符串：“虽然您可以使用 token.split_contents() 将任意数量的参数传递给模板标签，但参数都被解包为字符串字面意思。”我想我会传递相关评论对象的 ID 并将其拉入标签中。

我对此是错误的，只需访问传入的对象，例如：

self.comment.resolve(context).user 

vs.

self.comment.user

Answer 1

好的，我就是这样做的...

标签在模板中的使用方式如下：

   {% load access_tags %}
   {% if_authorized comment.user object.user user %}
      <a href="{% url delete_comment comment.id %}">Delete</a>
   {% endif_authorized %}

模板标签文件名为“access_tag.py”，位于我的应用程序的“templatetags”目录中。这是“access_tag.py”的内容：

from django.template import Node, NodeList, TemplateSyntaxError
from django.template import Library, Variable, VariableDoesNotExist

register = Library()

def do_if_authorized(parser, token):
    """
    Outputs the contents of the block if the 'comment owner' or the 
    'page owner' is also the 'authenticated user'. As well, you can use
    an {% else %} tag to show text if the match fails.

    Takes three parameters:
      1) the comment owner
      2) the page owner
      3) the current authenticated user
    """
    bits = token.contents.split()
    if len(bits) != 4:
        raise TemplateSyntaxError("%s tag takes three arguments: \
                                   1) the comment owner \
                                   2) the page owner \
                                   3) the current authenticated user" % bits[0])
    nodelist_true = parser.parse(('else', 'endif_authorized'))
    token = parser.next_token()

    if token.contents == 'else':
        nodelist_false = parser.parse(('endif_authorized',))
        parser.delete_first_token()
    else:
        nodelist_false = NodeList()
    return IfAuthorizedNode(bits[1], bits[2], bits[3], nodelist_true, nodelist_false)

class IfAuthorizedNode(Node):
    def __init__(self, comment_owner, page_owner, authenticated_user, nodelist_true, nodelist_false):
        self.nodelist_true = nodelist_true
        self.nodelist_false = nodelist_false
        self.comment_owner = Variable(comment_owner)
        self.page_owner = Variable(page_owner)
        self.authenticated_user = Variable(authenticated_user)

    def render(self, context):
        try:
            comment_owner = self.comment_owner.resolve(context)
            page_owner = self.page_owner.resolve(context)
            authenticated_user = self.authenticated_user.resolve(context)
        except VariableDoesNotExist:
            return ''

        if comment_owner == authenticated_user or page_owner == authenticated_user:
            return self.nodelist_true.render(context)
        else:
            return self.nodelist_false.render(context)

register.tag('if_authorized', do_if_authorized)

完成。最后，使用内置的 {% if %} 标签来进行此比较是非常容易的，但由于我还有其他每个对象的授权要做，所以我将继续构建这些自定义“访问标签”。另外，模板代码看起来更加整洁:)

Answer 2

已经存在一个旨在做您想做的事情的项目。

django-authority 允许对模板中的权限进行细粒度控制。

Django 1.2 还在模板中包含用户权限。

Answer 3

怎么样...创建一个自定义标签 在上下文中写入一个变量，然后使用 {% if %} 测试该变量，

就像这样：

{% check_access comment %}
{% if has_access %}
    <a href="{% url delete_comment comment.id %}">Delete</a>
{% endif %}

当然是“check_access”标签将在上下文中写入“has_access”。

祝你好运