django - 如何按计算属性的总和对 Django 模型进行排序?
问题描述
我有两个模型,项目和会话。一个项目有很多会话,一个用户有很多项目:
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.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]
这会在一个查询中获取所有相关会话,但是然后我按项目对它们进行分组,这需要很长时间 - 当有大约 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
与 that 相关的所有 s 的查询集Project
。
您可以获取所有用户的项目并循环遍历每个项目的会话,而不是像现在这样获取所有用户会话,例如:
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
保存 sorresponding 的持续时间的字段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'))
是dict
只有一个键 ( total_duration
) 的 a,我们从中获取值。
接下来,您可以构建项目和持续时间列表,然后按持续时间对其进行排序,例如:
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
推荐阅读
- ios - 如何知道 CollectionView 中 Header 的特定部分是否滚动到屏幕边界之外?
- wcf - 如何使用 WCF 服务?
- dart - Flutter:PageController 不保留页面,尽管 keepPage:true
- sql - SQL 查询:使用第 5 行作为标题“插入”
- oracle - Oracle 将 NCLOB 数据插入 NVARCHAR2
- python - “pip3 install virtualenv”什么也没做
- matlab - matlab不规则中的子图问题
- mongodb - 如何仅获取 mongodb 中具有 3 个不同值的文档?
- django - Django:使用表单工具在页面中拆分表单并保存到数据库
- android - 清除排队的视图无效(由 postInvalidateDelayed 排队)