gpt4 book ai didi

django - 如何按计算属性的总和对 Django 模型进行排序?

转载 作者:搜寻专家 更新时间:2023-10-30 19:50:11 24 4
gpt4 key购买 nike

我有两个模型,Project 和 Session。一个项目有多个 session ,一个用户有多个项目:

class Project(models.Model):
class Meta:
ordering = [models.functions.Lower("name")]

name = models.CharField(max_length=255)
user = models.ForeignKey(User, on_delete=models.CASCADE)

class Session(models.Model):
start = models.DateTimeField()
end = models.DateTimeField()
timezone = TimeZoneField()
breaks = models.IntegerField(default=0, validators=[MinValueValidator(0)])
project = models.ForeignKey(Project, on_delete=models.CASCADE)

def duration(self):
# returns minutes in (end - start)

我想要一种方法来获取给定用户的所有项目,并按其所有 session 的持续时间总和排序。由于 session.duration() 不是数据库字段,而是根据数据库字段计算得出的,因此我无法在单个数据库查询中获取此信息。

我目前的解决方案是:

sessions = Session.objects.filter(project__user=self)
groups = [[a, sum([s.duration() for s in b])] for a, b in groupby(
sessions, key=lambda s: s.project
)]
groups = sorted(groups, key=lambda g: g[1], reverse=True)
return [g[0] for g in groups]

这会在单个查询中获取所有相关 session ,但随后我按项目对它们进行分组,这花费的时间太长了 - 当大约有 100 个项目时大约需要一秒钟。有没有一种方法可以花费更少的时间来完成此任务?理想情况下不需要为每个项目调用数据库?

我正在使用 Django 2.0。

最佳答案

您可以使用注释和聚合来实现这一点。首先,通过更改这一行来稍微修改 session 模型:

project = models.ForeignKey(Project, on_delete=models.CASCADE)

为此:

project = models.ForeignKey(Project, related_name='sessions', on_delete=models.CASCADE)

-现在每个 Project 实例都会有一个 sessions 字段,它将包含与该 相关的所有 Session 的查询集项目

您可以获取所有用户的项目并循环遍历每个项目的 session ,而不是像现在这样获取所有用户 session :

projects = Project.objects.filter(user=self)
for p in projects:
sessions = p.sessions.all()

然后您可以操作sessions 查询集,使用如下表达式字段注释它们:

from django.db.models import ExpressionWrapper, F, fields

duration_ = ExpressionWrapper(F('end') - F('start'), output_field=fields.DurationField())
sessions = p.sessions.annotate(d=duration_)

此时 sessions 查询集的每个成员都有一个名为 d 的字段,其中包含相应的 Session 的持续时间。为了总结持续时间,我们可以使用 Django 查询集的聚合功能,如下所示:

from django.db.models import Sum
total = sessions.aggregate(total_duration=Sum('d'))["total_duration"]

我们在第二行所做的是通过将所有值添加到 d 字段中,从查询集中创建单个元素(“聚合”) ,并将结果分配给名为 total_duration 的字段。该表达式的结果:

sessions.aggregate(total_duration=Sum('d'))

是一个只有一个键(total_duration)的dict,我们从中获取值。

接下来,您可以构建项目和持续时间列表,然后按持续时间对其进行排序,例如像这样:

import operator
plist = []
for p in projects:
sessions = p.sessions.annotate(d=duration_)
total = sessions.aggregate(total_duration=Sum('d'))["total_duration"]
# total holds the sum of this project's sessions
plist.append({'p':p,'total':total})
plist.sort(key=operator.itemgetter('total'))

projects = [item['p'] for item in plist]

总结一下:

import operator
from django.db.models import F, Sum, ExpressionWrapper, fields

duration_ = ExpressionWrapper(F('end') - F('start'), output_field=fields.DurationField())
projects = Project.objects.filter(user=self)
plist = []

for p in projects:
sessions = p.sessions.annotate(d=duration_)
total = sessions.aggregate(total_duration=Sum('d'))["total_duration"]
# total holds the sum of this project's sessions
plist.append({'p':p,'total':total})

plist.sort(key=operator.itemgetter('total'))

projects = [item['p'] for item in plist]

引用:this answer , Django Query Expressions , Django Aggregation

关于django - 如何按计算属性的总和对 Django 模型进行排序?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50190147/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com