Question

我有以下两种型号。

class Product(models.Model):
    product_group=models.ForeignKey('productgroup.ProductGroup', null=False,blank=False)
    manufacturer=models.ForeignKey(Manufacturer, null=False,blank=False)
    product_type=models.CharField(max_length=2, choices=PRODUCT_TYPE, null=False,blank=False)
    wheel_position=models.CharField(max_length=1, choices=FRONT_BACK, null=False,blank=False)
    opening_stock=models.PositiveIntegerField(default=0)

    class Meta:
        ordering=['product_group','manufacturer']
        unique_together = ('product_group', 'manufacturer','product_type','wheel_position')


    def get_total_stock_in(self):
        sum=Stock.objects.filter(product=self.id, ttype='I').aggregate(Sum('quantity'))

        if sum['quantity__sum'] is not None:
            return sum['quantity__sum']
        else:
            return 0

    def get_total_stock_out(self):
        sum=Stock.objects.filter(product=self.id, ttype='O').aggregate(Sum('quantity'))
        if sum['quantity__sum'] is not None:
            return sum['quantity__sum']
        else:
            return 0

    def get_balance_stock(self):
        return (self.opening_stock+self.get_total_stock_in()
                - self.get_total_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)
    added_date=models.DateTimeField(blank=False, auto_now=True)

我有一个ListView子类列出所有产品。

class ProductList(ListView):
    model=Product
    paginate_by = 20

    def get_queryset(self):
        queryset = super(ProductList, self).get_queryset()
        queryset = queryset.prefetch_related('product_group','product_group__category','manufacturer')
        return queryset.order_by('product_group__category','product_group')

我需要在列表视图中显示total stock in，total stock out和available balance以及每个产品。

现在，我正在调用get_total_stock_in()中定义的函数get_total_stock_out()，get_balance_stock()和Product model。但这会在数据库上产生大量重复查询。

如何将这些数据添加到ListView的查询集中？

感谢您的帮助。

Answer 1

让我在这里发布解决方案。

我在django - annotate() - Sum() of a column with filter on another column

的另一个帖子的帮助下得到了我的解决方案

现在是我的ListView。

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(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))
        )
        queryset = queryset.annotate(
        balance_stock = ExpressionWrapper(F('opening_stock') + F('stock_in_sum') - F('stock_out_sum'), output_field=DecimalField())
        )

        return queryset

django ListView - 通过与另一个模型连接来自定义查询集

1 个答案: