布尔注释导致重复?

时间:2017-11-22 19:56:17

标签: python django python-3.x django-queryset django-1.11

我正在尝试实施基本的“收藏夹”。系统基于外键表。

我们说我有以下简单的模型:

class Item(models.Model)
  id = models.IntegerField()

class User(models.Model)
  id = models.IntegerField()

class UserFavourites(models.Model)
  user = models.ForeignKey(User, related_name='user_for_fav_rel')
  item = models.ForeignKey(Item, related_name='item_for_fav_rel')

现在,我为项目生成以下查询集,以查看用户是否已将其标记为收藏:

queryset = Item.objects.all()

USER_ID = request.user.id

queryset = queryset.annotate(
    favourite=Case(
        When(
            item_for_fav_rel__user__id=USER_ID,
            then=Value('True')
        ),
        default=Value('False'),
        output_field=BooleanField()
    )
)

所有这些都很有效,但在回复中,如果该项目确实受到青睐,我会在查询集中收到该特定项目的副本。知道怎么避免这个吗?

结果SQL查询(编辑到我认为的最小例子......)

SELECT 
    "core_item"."id", 
    CASE 
        WHEN "core_userfavourites"."user_id" = 1 THEN True 
        ELSE False 
    END AS "favourite" 
FROM "core_item" 
LEFT OUTER JOIN "core_userfavourites" 
    ON ("core_item"."id" = "core_userfavourites"."item_id")

1 个答案:

答案 0 :(得分:2)

问题是你为core_item和core_userfavorites的每个组合获得了一行。似乎没有办法在没有原始SQL的情况下完成您的尝试,但幸运的是Django最近(1.11)添加了编写SubQuery子句的功能,您可以在这里使用

from django.db.models.expressions import OuterRef, Subquery

queryset = Item.objects.all()

USER_ID = request.user.id
user_favorites = UserFavourites.objects.filter(
    user_id=USER_ID, 
    item_id=OuterRef('id')
)[:1].values('user_id')

queryset = queryset.annotate(user_favorite=Subquery(user_favorites))

如果用户对其进行了优惠,则会在user_id字段中显示user_favorite,如果没有,则为None

基本上你正在编写一个子查询来从相关表中选择一个任意值。