是否可以将参数传递给Django模板中的模型类的方法?

时间:2012-05-09 03:26:09

标签: python django django-templates

我有一个名为UserProfile的模型,定义为

class UserProfile(models.Model):
    user       = models.OneToOneField(User, related_name='userprofile_from_user')
    user_types = models.ManyToManyField(UserType, related_name='userprofiles_from_user_types', null=True, blank=True)

    def has_user_types(self, user_types):
        return self.user_types.filter(name__in=user_types).count()

UserType定义为

class UserType(models.Model):
    TYPE_CHOICES = (
        ('ad', 'administrator'   ), # 1
        ('mo', 'moderator'       ), # 2
        ('vi', 'viewer'          ), # 3
        ('pm', 'property manager'), # 4
        ('po', 'property owner'  ), # 5
        ('vm', 'vendor manager'  ), # 6
        ('ve', 'vendor'          ), # 7
        ('te', 'tenant'          ), # 8
    )

    name = models.CharField(max_length=2, choices=TYPE_CHOICES)

我希望能够使用UserProfile的方法has_user_types()。在一个视图中,我会做类似

的事情
if user.profile.has_user_types(['ad', 'mo', 'pm']):
    # The user is any combination of an administrator, moderator, or property manager.

但是我可以在模板中做同样的事情吗?我专门检查一些用户类型,所以我想做一些像

这样的事情
{% if user.profile.has_user_types(['te']) %}

我知道我可以简单地在模型中定义另一个名为is_tenant()的方法(不带参数),但我也想检查其他用户类型,我想知道是否可以合并has_user_types()

附带问题:如果Django的默认模板无法执行此操作,那么可以Jinja2进行吗?


解决方案

感谢Ignacio Vazquez-Abrams的帮助!

custom_tags.py

@register.assignment_tag
def has_user_types(user_pk, *args):
    user = User.objects.get(pk=user_pk)

    return user.profile.has_user_types(args)

模板:

{% load has_user_types from custom_tags %}

{# I pass the pk because I want to be able to pass any user, not just request.user #}
{% has_user_types user.pk "te" as is_tenant %}
{% if is_tenant %}
    {# Show something #}
{% endif %}

1 个答案:

答案 0 :(得分:2)

没有。写一个custom filter or template tag来检查它们。