带注释的Queryset带OuterRef和Sum的Subquery

Question

我正试图返回Subquery中另一个模型的字段总和。

我的主查询集返回company类型的所有用户。我必须通过从CreditOrder中获取数据并对credit_used字段求和来返回所使用的信用总额。

我使用的是django modelcluster 中的ClusterableModel和ParentalKey

我的CreditOrder型号

class CreditOrder(ClusterableModel):
credit = ParentalKey(
Credit, on_delete=models.CASCADE, related_name="credit_order"
)
order = ParentalKey(Order, on_delete=models.CASCADE, related_name="credit_order")
credit_used = models.DecimalField(
max_digits=12, decimal_places=2, null=True, blank=True
)

我的User型号

class User(AbstractUser, ClusterableModel):
username = models.CharField(max_length=40, null=True, blank=True)
user_type = models.CharField(max_length=20, choices=TIPO_UTENTE, default="dipendente")

使用类模型User 的查询集

def get_queryset(self, request):
qs = super().get_queryset(request)
qs = qs.filter(user_type='company')
credits_used_subquery = Subquery(CreditOrder.objects.filter(credit__font__company__id=OuterRef('id')).order_by()
.values('credit_used').annotate(credit_used_sum=Sum('credit_used'))
.values('credit_used_sum'), output_field=DecimalField())
qs = qs.annotate(
_credits_used_sum=credits_used_subquery
)
return qs

但这个错误正在返回我：

django.db.utils.ProgrammingError: more than one row returned by a subquery used as an expression

Answer 1

如果你只需要把公司使用的所有信用加起来，你可以做：

qs.annotate(_credits_used_sum=Sum('font__credit__credit_used'))

Answer 2

我用这个类解决了我的问题：

class SubquerySum(Subquery):
template = "(SELECT COALESCE(SUM(%(field)s), %(zero_value)s) FROM (%(subquery)s) _sum)"
def as_sql(self, compiler, connection, template=None, **extra_context):
if 'field' not in extra_context and 'field' not in self.extra:
if len(self.queryset._fields) > 1:
raise FieldError('You must provide the field name, or have a single column')
extra_context['field'] = self.queryset._fields[0]
if 'zero_value' not in extra_context and 'zero_value' not in self.extra:
extra_context['zero_value'] = 0
return super().as_sql(compiler, connection, template=template, **extra_context)

和

def get_queryset(self, request):
credit_query=CreditOrder.objects.filter(credit__font__company__id=OuterRef('id')).order_by()
.values('credit_used')
qs = super().get_queryset(request)
qs = qs.filter(user_type='company')
qs = qs.annotate(
_credits_used_sum=SubquerySum(credit_query, zero_value=0, field='credit_used')
)
return qs

Answer 3

如果看不到CreditOrder和User之间的所有模型，就很难准确地判断出哪里出了问题。看起来Credit模型链接到Font，Font可能有一个名为Company的属性，这是User模型的外键？

在任何情况下，您的第一个values调用都有错误的参数，您需要按外部引用中链接的基本相同的东西进行分组。所以我建议

.values('credit__font__company__id')

在对值的第一次调用中。并保持对值的注释和第二次调用相同。

另一个答案建议使用联接而不是子查询进行求和，如果您喜欢该api的简单性，但仍想使用子查询，则可以使用django-sql-utils包。pip install django-sql-utils

from sql_util.utils import SubquerySum
qs.annotate(_credits_used_sum=SubquerySum('font_credit_credit_used')