如何使用可变长度的arg-lis制作Django自定义模板标记

Question

我正在编写一个自定义模板标记“firstnotnone”，类似于Django的“firstof”模板标记。如何使用可变长度参数？下面的代码导致TemplateSyntaxError，firstnotnone接受1个参数。在

模板：

{% load library %}
{% firstnotnone 'a' 'b' 'c' %}

自定义模板标记库：

^{pr2}$

Answer 1

firstof标记不是通过simple_tag修饰符实现的，它使用template.Node子类的长形式和一个单独的标记函数。您可以在django.template.defaulttags中看到代码-为了您的目的更改它应该相当简单。在

Answer 2

自定义模板标记：

from django.template import Library, Node, TemplateSyntaxError
from django.utils.encoding import smart_unicode

register = Library()

class FirstNotNoneNode(Node):
    def __init__(self, vars):
        self.vars = vars

    def render(self, context):
        for var in self.vars:
            value = var.resolve(context, True)
            if value is not None:
                return smart_unicode(value)
        return u''

def firstnotnone(parser,token):
    """
    Outputs the first variable passed that is not None
    """
    bits = token.split_contents()[1:]
    if len(bits) < 1:
        raise TemplateSyntaxError("'firstnotnone' statement requires at least one argument")
    return FirstNotNoneNode([parser.compile_filter(bit) for bit in bits])

firstnotnone = register.tag(firstnotnone)