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

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)

我有一个列表视图子类来列出所有产品。

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以及每个产品。

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

如何将这些数据添加到列表视图的查询集中？

感谢您的任何帮助。

Answer 1

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

我在 django 的另一个线程的帮助下得到了我的解决方案 - annotate(( - 一列的 Sum(( 在另一列上带有过滤器

这是我现在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