如何使用自定义django templatetag与django模板if语句?

时间:2013-02-08 07:17:00

标签: python django django-templates

我制作了一个django模板标记,用于计算我的自定义用户多对多字段长度:

from django import template

register = template.Library()

@register.simple_tag(takes_context=True)
def unread_messages_count(context):
    user = context['request'].user
    return len(user.messages_unread.all())

并且在模板本身内,我想只在大于零的情况下才向用户显示它,所以我尝试了:

{% ifnotequal unread_messages_count 0 %}
   some code...
{% endifnotequal %}

但显然它不起作用。甚至没有'带'声明:

{% with unread_messages_count as unread_count %}
    {% ifnotequal unread_count 0 %}
        some code...
    {% endifnotequal %}
{% endwith %}

如何检查变量是否大于0,并且只有当变量大于0时,才向用户显示一些代码(包括变量本身中的数字)。 感谢。

2 个答案:

答案 0 :(得分:9)

最简单的方法是使用赋值标记..

https://docs.djangoproject.com/en/dev/howto/custom-template-tags/#assignment-tags

@register.assignment_tag(takes_context=True)
def unread_messages_count(context):
    user = context['request'].user
    return len(user.messages_unread.all())

{% unread_messages_count as cnt %}
{% if cnt %}
   foo
{% endif %}

答案 1 :(得分:2)

您可以使用django自定义过滤器https://docs.djangoproject.com/en/dev/howto/custom-template-tags/#writing-custom-template-filters

def unread_messages_count(user_id):
  # x = unread_count   ## you have the user_id
  return x

并在模板中

{% if request.user.id|unread_messages_count > 0 %}
  some code...
{% endif %}