Django Sum 浮点值

Question

我在 Django 中有一个模型，它有三个字段。浮点数和两个 CharFields。我需要将其他两个浮点字段相加。

models.py

class Hours(model.Model):
    hours_worked = models.FloatField()
    order_number = models.CharField()
    oper = models.CharField()

我需要做的是遍历数据库中的所有数据，并对相同的 order_number+oper 求和 hours_worked。 所以假设我有 db 4 记录：

|hours_worked|order_number|oper|
|------------|------------|----|
|           4|         252|  10|
|           8|         320|  20|
|           8|         252|  10|
|           6|         252|  20|

我需要像这样返回三个对象的查询集：

|hours_worked|order_number|oper|
|------------|------------|----|
|          12|         252|  10|
|           8|         320|  20|
|           6|         252|  20|

有什么简单的方法吗？

Answer 1

这里有两种可能的解决方案（当然还有其他的）

使用django-group-by

from django_group_by import GroupByMixin

class GroupedHoursQuerySet(QuerySet, GroupByMixin):
    def select_related(self, *fields):
        try:
            return super().select_related(*fields)  # once grouping was applied, select_related will fail
        except TypeError:
            return self

class GroupedHours(Hours):
    objects = GroupedHoursQuerySet.as_manager()
    class Meta:
        proxy = True

并查询数据：

from django.contrib.postgres.aggregates import ArrayAgg, StringAgg
from django.db.models import Sum

qs = GroupedHours.objects.group_by('order_number', 'oper').annotate(
    hours_worked=Sum('hours_worked'),
    pks=ArrayAgg('id'),
    pk=StringAgg(Cast('id', output_field=TextField()), '-'),
).distinct()

返回的查询集中的对象具有与原始模型相同的属性，但hours_worked 包含总和：

for gh in qs:
    print(f'Operator {gh.oper} worked {gh.hours_worked} on order {gh.order_number}')

Django 的values()

如果您不需要 Django 模型对象只是实际数字，您当然可以在没有 django-group-by 模块的情况下使用聚合和注释。

以下方法也可能有效（未经测试），请参阅the documentation 了解更多详情：

from django.db.models import Sum

qs = Hours.objects.values('order_number', 'oper').annotate(
    worktime=Sum('hours_worked')

可能需要添加.distinct()。

这会返回一个字典：

for gh in qs:
    print(f'Operator {gh['oper']} worked {gh['hours_worked']} on order {gh['order_number']}')