Django查询:如何获取非全部产品在客户黑名单的Advisory列表
我懂你现在的困扰——你想拿到指定客户的Advisory列表,要求这些Advisory里至少有一个产品不在客户的黑名单里,但当前的查询把本该保留的Advisory1也排除了,这完全不符合预期。咱们一步步来解决这个问题。
问题回顾
先明确下你的模型和场景:
模型定义
class Product(BaseModel): ''' whatever ''' class Customer(BaseModel): blacklist = models.ManyToManyField(Product, blank=True) class Advisory(BaseModel): product_names = models.ManyToManyField(Product)
数据库示例
- Advisory 1(应保留):关联Product1、Product2、Product3
- Advisory 2(应排除):关联Product2、Product3
- Customer1的黑名单:Product2、Product3、Product4、Product5
现有错误查询
queryset = Advisory.objects.all() blacklist = Customer.blacklist.all() # 这里还犯了个小错:应该是指定客户的黑名单,不是所有客户的! queryset = queryset.exclude(product_names__in=blacklist).distinct()
这个查询会把Advisory1和Advisory2一起排除,原因很简单:exclude(product_names__in=blacklist)会排除所有关联了黑名单中任意产品的Advisory——哪怕这个Advisory还有其他不在黑名单的产品,就像Advisory1里的Product1明明符合要求,但因为它关联了黑名单里的Product2,就被一并排除了。
正确解决方案
我们需要的是:筛选出至少有一个产品不在客户黑名单里的Advisory,或者反过来,排除所有产品都在黑名单里的Advisory。这里给你两种可行的方法:
方法一:用annotate统计非黑名单产品数量
这种方法直观易懂,适合新手:
from django.db.models import Count, Q # 首先获取你要查询的指定客户 target_customer = Customer.objects.get(pk=你的客户ID) # 获取该客户的黑名单产品ID列表(用ID比用对象更高效) blacklist_ids = target_customer.blacklist.values_list('id', flat=True) # 给每个Advisory标注它关联的非黑名单产品数量,然后筛选数量>0的 valid_advisories = Advisory.objects.annotate( non_blacklisted_count=Count( 'product_names', filter=~Q(product_names__id__in=blacklist_ids) ) ).filter(non_blacklisted_count__gt=0)
原理:annotate会给每个Advisory对象新增一个non_blacklisted_count字段,统计它关联的产品中不在黑名单里的数量。我们只保留这个数量大于0的Advisory,也就是至少有一个符合要求的产品的记录。
方法二:用exists子查询筛选关联非黑名单产品的Advisory
这种方法性能更优,尤其是数据量大的时候:
# 同样先获取指定客户和黑名单ID target_customer = Customer.objects.get(pk=你的客户ID) blacklist_ids = target_customer.blacklist.values_list('id', flat=True) # 筛选关联了至少一个非黑名单产品的Advisory valid_advisories = Advisory.objects.filter( product_names__in=Product.objects.exclude(id__in=blacklist_ids) ).distinct()
原理:Product.objects.exclude(id__in=blacklist_ids)先拿到所有不在黑名单里的产品,然后筛选关联了这些产品的Advisory——Advisory1因为关联了Product1(非黑名单)会被选中,Advisory2关联的全是黑名单产品,自然会被排除。
额外提醒
一定要注意:你之前的Customer.blacklist.all()是获取所有客户的黑名单产品,这肯定不是你想要的!必须先拿到具体的客户实例(比如target_customer = Customer.objects.get(id=xxx)),再获取该客户的黑名单。
内容的提问来源于stack exchange,提问作者TheStigmj

