Question

鉴于我有一个django模型，它有一个与自身链接的ForeignKey。

class DjangoModel():

    [...]

    successor = models.ForeignKey('self', null=True)

我能够编写这样的自定义django数据库函数：

from django.db.models import BooleanField
from django.db.models import Func


class IsNull(Func):
    """
    See docs: https://docs.djangoproject.com/en/1.8/ref/models/database-functions/
    """
    template = '%(expressions)s IS NULL'

    def __init__(self, *args, **kwargs):
        kwargs['output_field'] = BooleanField()
        super(IsNull, self).__init__(*args, **kwargs)

所以我可以这样做：

queryset = DjangoModel.objects.all()
queryset = queryset.annotate(**{'is_latest': IsNull('successor')})

如果使用queryset.values() ..我得

[{'pk': 1, is_latest': True}, {'pk': 2, 'is_latest': False}, ]

当对象的is_latest == True字段为NULL时successor。{/ p>

现在我想做类似的事情，但不知道从哪里开始！

捆绑的django.contrib.auth.models.User与django.contrib.auth.models.Group模型有一个ManyToMany关系

对于我的项目，有多种用户组类型，例如客户/营销/财务等

我想要做的是...使用User字段注释is_FOO查询集，其中FOO是一个组名。例如is_customer或is_marketing

因此，如果我在查询集上使用.values()，我应该得到这样的结果：

[{'pk': 1, 'is_customer': True, 'is_marketing': False }, {'pk': 1, 'is_customer': True, 'is_marketing': True }]

组名可以是硬编码的，例如

queryset.annotate(**{'is_customer': IsGroupMember('customer')})

我只需要帮助IsGroupMember数据库功能！

这甚至可能吗？让我开始的任何提示或技巧？

任何帮助都将得到真诚的赞赏。非常感谢

Answer 1

我有2个问题的解决方案： - ）

鉴于我正在使用django auth用户模型

from django.contrib.auth.models import User

和我感兴趣的群组名称：

CUSTOMER_GROUP_NAME = 'customer'

解决方案1  - 正如Todor Velichkov在django-users Google群组中所建议的那样。

下面的答案略有修改版本：

from django.db.models.expressions import RawSQL


def is_group_member(group_name):
    return RawSQL("""EXISTS(
        SELECT 1 FROM `auth_group`
        WHERE `auth_group`.`name` = %s
            AND `auth_group`.`id` IN (
                SELECT `auth_user_groups`.`group_id`
                FROM `auth_user_groups`
                WHERE `auth_user_groups`.`user_id` = `auth_user`.`id`
            )
    )""", (group_name,), output_field=BooleanField())


qs1 = User.objects.all().annotate(is_customer=is_group_member(CUSTOMER_GROUP_NAME))

链接到原始答案：https://groups.google.com/d/msg/django-users/BLCGZtUzEcY/w5w87frbBgAJ

我做了一些研究，并了解了django条件表达式。

文档：https://docs.djangoproject.com/en/1.9/ref/models/conditional-expressions/

解决方案2 - 使用django表达式： - ）

from django.contrib.auth.models import User
from django.db.models import BooleanField
from django.db.models import When, Case, Value

query = When(groups__name__in=[CUSTOMER_GROUP_NAME, ], then=Value(1))
qs2 = User.objects.annotate(
    is_customer=Case(query, default=Value(0), output_field=BooleanField()))

我认为“解决方案2”更清晰，更易于阅读。所以我将使用它： - ）

使用组名注释django用户查询集

1 个答案: