schinckel / django-computed-field Goto Github PK

class Work(models.Model):
    task = models.ForeignKey("Task", on_delete=models.PROTECT)
    date = models.DateField()
    hours_done = models.FloatField()

class TaskManager(models.Manager):
    def get_queryset(self):
        qs = super().get_queryset()
        qs = qs.annotate(total_hours_done=models.Subquery(
            Work.objects.filter(task=OuterRef("pk"))
            .values("task")
            .order_by()
            .annotate(total=Sum("hours_done"))
            .values("total"),
            output_field=models.FloatField(),
        ))
        return qs

class Task(models.Model):
    objects = TaskManager()
    name = models.CharField(max_length=255)

class Task(models.Model):
    name = models.CharField(max_length=255)
    total_hours_done = ComputedField(
        models.Subquery(
            Work.objects.filter(task=OuterRef("pk"))
            .values("task")
            .order_by()
            .annotate(total=Sum("hours_done"))
            .values("total"),
            output_field=models.FloatField(),
        )
    )