Django: get unique object list from QuerySet

2019-03-23 13:22发布

I have the following (simplified) models in my Django app:

class Color(models.Model):
    name = models.CharField(max_length=10)

class Item(models.Model):
    name = models.CharField(max_length=200)
    color = models.ForeignKey(Color, blank=True, null=True)

class Favorite(models.Model):
    user = models.ForeignKey(User)
    item = models.ForeignKey(Item)

I'm currently getting all the items I need using the following query:

favorites = Favorite.objects.filter(user=request.user)

How can I get all the distinct colors for the items in that QuerySet?

I need the a list of the actual color objects, not just the color ids, which I can get using

favorites.values_list('item__color').distinct

4条回答
Rolldiameter
2楼-- · 2019-03-23 13:46

The is values_list(*fields, flat=False, named=False) method, so run it on your objects, for example:

user.groups.values_list('name', flat=True)
查看更多
劫难
3楼-- · 2019-03-23 13:52

If I understand you correctly, the following should do the trick:

favorites = Favorite.objects.filter(user=request.user)
color_ids = favorites.values_list('item__color', flat=True).distinct()
colors = Color.objects.filter(id__in=color_ids)

There has to be a cleaner way than that though.

Edit: A much cleaner solution:

colors = Color.objects.filter(item__favorite__user=request.user).distinct()
查看更多
甜甜的少女心
4楼-- · 2019-03-23 14:04

Can you do:

favorites = Favorite.objects.filter(user=request.user).distinct('item__color')
查看更多
仙女界的扛把子
5楼-- · 2019-03-23 14:08

Can you do:

Color.objects.filter(item__favorite__user = request.user).distinct()

You might have to set some related_names on your foreign keys if these aren't the defaults (I can never remember the defaults).

查看更多
登录 后发表回答