比较两个不同 Django 模型中的字段

Question

I am trying fill a field (hours) in Employee model based on the aggregate of a field (time_worked) in Timesheet model.

MODELS.PY

class Employee(models.Model):
    ...
    user = models.ForeignKey(User, on_delete=models.CASCADE, null=True, related_name='employees')
    hours = models.DurationField(null=True, blank=True)

class Timesheet(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, null=True, blank=True)
    employee    = models.ForeignKey(Employee, on_delete=models.CASCADE, null=True,)
    time_worked = models.DurationField(null=True, blank=True)

I tried hours=Timesheet.objects.filter(employee=employee).aggregate(Sum('time_worked'))

But when I try insert into Employee model that's where I get errors.

If I do this, Employee.objects.filter(id=employee).update(hours=hours) , I get error "id expected a number but got Jacob". But id in Timesheet is actually a number (foreignkey).

How do I add up the time_worked in Timesheet model of only employee A and insert the results into hours column of Employee model for employee A?

Or simply put, how do I match an employee in Timesheet to an employee in Employee?

Answer 1

Here's what I did to solve my problem.

class Employee(models.Model):
        ...
        hours = models.DurationField(null=True, blank=True)

class TimeLog(models.Model):
    ...
    employee    = models.IntegerField(null=True, blank=True) # No foreignkey

    time_worked = models.DurationField(null=True, blank=True)  

    def save( self, *args, **kwargs ):
        employee_id = self.employee.pk #What's common for both models, very handy 
         #in the absence of a foreignkey

        aggr_hours = TimeLog.objects.filter(employee=employee_id).values(
                'employee').annotate(sum_hours=Sum(F('time_worked'), 
                 output_field=DurationField()))

        Employee.objects.filter(id=employee_id).annotate(total_hours=Subquery(
                aggr_hours.values('sum_hours')[:1], output_field=DurationField()),
                ).update(hours=F('total_hours'))
        super().save( *args, **kwargs )

Might not be wizardry but it's solid, and reliable