Question

我有一个名为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 %}

Answer 1

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

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

1 个答案: