python – 从queryset构建最高价格列表的最有效方法？

在我的应用程序的一个页面中,我试图为每个公司展示最昂贵的汽车.我的模型大致如下：

class Company(models.Model):
    id = models.IntegerField(primary_key=True)
    company = models.CharField(max_length=100)
    headcount = models.IntegerField(null=False)
    info = models.CharField(max_length=100)

class Car(models.Model):
    id = models.IntegerField(primary_key=True)
    company_unique = models.ForeignKey(Company)
    company = models.CharField(max_length=50)
    name = models.CharField(max_length=100)
    price = models.DecimalField(max_digits=9, decimal_places=2, default=0.00)

所以,我想建立一个包含每个公司最昂贵的Car对象的列表.

我这样解决了这个问题：

company_list = Company.objects.all()
most_expensive = []
for company in company_list:
    most_expensive.append(Car.objects.filter(company_unique=company.id).order_by("-price")[0])

然而,这似乎是一种非常低效的方法.我可以看到Django Debug Toolbar这个代码正在制作太多的mysql查询.

有人可以建议一个更好的方法来构建这个列表,可能只是一次或两次命中MySQL吗？

解决方法:

虽然你正在处理的是一个很常见的情况,但显而易见的解决方案似乎缺乏.

解决方案1,发现于this article.您可以尝试以下这些方面：

companies = Company.objects.annotate(max_price=Max('car__price'))
values = tuple((company.id, company.max_price) for company in companies)

expensive_cars = Car.objects.extra(where=['(company_unique_id, price) IN %s' % (values,)])

不能说我喜欢解决方案 – 应该避免使用.extra – 但我想不出更好的方法.我也不完全确定它会起作用.

解决方案2,次优.你可以使用custom Prefetch object.

prefetch = Prefetch('cars', queryset=Car.objects.order_by('-price'), to_attr='cars_by_price')
companies = Company.objects.prefetch_related(prefetch)

most_expensive_cars = []
for company in companies:
    most_expensive_cars.append(list(company.cars_by_price.all())[0])

这绝对可以工作并在两个查询中获取所有内容,但是非常浪费,因为它会将与给定公司集相关的所有汽车加载到内存中.请注意,list()部分不是可选的：无论您采用切片还是索引,都会复制查询集并生成单独的数据库查询,从而否定预取,而实例化列表将使用所述预取的结果.

如果您之后需要访问公司,例如Car.company,请不要回避使用select_related,正如Erik在评论中所建议的那样.

标签：python,mysql,django,django-queryset,django-orm
来源： https://codeday.me/bug/20190706/1393097.html