将字符串与数据库对象django比较

Question

信封：Python 2.7.3 || 的Django 1.4.1

我试图根据登录用户的角色来限制网页菜单选项。 我已经在模型中创建了UserProfile类，并将其链接到了文档中所描述的User（在添加用户时，在Django Admin中可以完美运行）。 例如，普通用户将仅具有“主页”和“ mytasks”，而管理员用户还将具有“域任务”。

这是base.html模板中的代码：

{% if user.get_profile.role == "Admin" %}
    <ul id="nav">
    <li><a href="{% url home %}">Home</a></li>
    <li><a href="{% url mytasks %}">My Tasks</a></li>
    <li><a href="{% url domaintasks %}">Domain Tasks</a></li>
    </ul>

呈现整个网站的导航栏时，选中以上选项。

我认为问题在于将字符串与数据库对象进行比较（如下所示），但还没有找到如何使比较工作的方法。

>>> b = Role.objects.all()
>>> print b
[<Role: Admin>, <Role: User>]

Answer 1

解决问题的正确方法是使用django admin在用户注册时分配这些组，为每种类型的用户创建组，甚至可以为现有用户编写脚本，然后使用in_group模板过滤器检查是否用户具有所需的特权。

但是按照您的方法，角色是在用户配置文件类中返回字符串的属性还是函数？ 如果是的话，我认为你的代码应该工作。 但是，如果角色是另一个模型的重要关键对象，则应使用类似以下内容的方法进行比较：

user.get_profile.role.attribute_in_the_other_model

Answer 2

有两种方法可以做到这一点。

• 检查Role的适当属性，而不是执行操作时检查的对象。 因此，如果您在Role具有name属性，则可以检查为

{% if user.get_profile.role.name == "Admin" %}

• 可以修改Role模型代码的另一种方法是在Role模型中实现一种方法，以检查角色是否为admin并在模板中使用它。

{% if user.get_profile.role.is_admin %}

Answer 3

另一种方法是调用角色模型的字符串表示方法

{% if user.get_profile.role.__str__ == "Admin" %}

要么

{% if user.get_profile.role.__unicode__ == "Admin" %}