django - annotate() - 对另一列进行过滤的列的 Sum()

Question

我有以下两个型号。

class Product(models.Model):
    product_group=models.ForeignKey('productgroup.ProductGroup', null=False,blank=False)
    manufacturer=models.ForeignKey(Manufacturer, null=False,blank=False)
    opening_stock=models.PositiveIntegerField(default=0)

    class Meta:
        unique_together = ('product_group', 'manufacturer')

和

TRANSACTION_TYPE=(('I','Stock In'),('O','Stock Out'))
class Stock(models.Model):
    product=models.ForeignKey('product.Product', blank=False,null=False)
    date=models.DateField(blank=False, null=False,)
    quantity=models.PositiveIntegerField(blank=False, null=False)
    ttype=models.CharField(max_length=1,verbose_name="Transaction type",choices=TRANSACTION_TYPE, blank=False)

我需要列出具有 stock_in_sum=Sum(of all stock ins) 、 stock_out_sum=Sum(of all stock outs) 和 blance_stock=opening_stock 的所有产品+stock_in_sum - stock_out_sum

这就是我迄今为止所取得的成就。

class ProductList(ListView):
    model=Product

    def get_queryset(self):
        queryset = super(ProductList, self).get_queryset()
        queryset = queryset.prefetch_related('product_group','product_group__category','manufacturer')
        queryset = queryset.annotate(stock_in_sum = Sum('stock__quantity'))
        queryset = queryset.annotate(stock_out_sum = Sum('stock__quantity'))

我需要得到

1. stock_in_sum 作为 sum(quantity)，其中 ttype='I'
2. stock_out_sum 作为 sum(quantity)，其中 ttype='O'
3. blance_stock 为 product.opening_stock + stock_in_sum - stock_out_sum

以及每个 Product 对象。

如何实现这一目标？

谢谢。

Answer 1

您可以使用conditional aggregation

queryset = queryset.annotate(
    stock_in_sum = Sum(Case(When(stock__ttype='I', then=F('stock__quantity')), output_field=DecimalField(), default=0)),
    stock_out_sum = Sum(Case(When(stock__ttype='O', then=F('stock__quantity')), output_field=DecimalField(), default=0)))
)

求和，然后用 F() expression 计算余额

queryset = queryset.annotate(balance_stock=F('opening_stock') + F('stock_in_sum') - F('stock_out_sum'))

您还可以链接不同的操作而不是多次分配:

queryset = queryset.prefetch_related(...).annotate(...).annotate(...)