1

我有一个名为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's 方法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 %}
4

1 回答 1

2

不。编写一个自定义过滤器或模板标签来检查它们。

于 2012-05-09T03:28:12.120 回答