Question

我正在使用Django 2.0和Django REST Framework

我有一个像下面这样的模型。

class Contact(models.Model):
    first_name = models.CharField(max_length=100)

class AmountGiven(models.Model):
    contact = models.ForeignKey(Contact, on_delete=models.PROTECT)
    amount = models.FloatField(help_text='Amount given to the contact')

    @property
    def total_payable(self):
        return self.amount

    @property
    def amount_due(self):
        returned_amount = 0
        for returned in self.amountreturned_set.all():
            returned_amount += returned.amount

        return self.total_payable - returned_amount

class AmountReturned(models.Model):
    amount_given = models.ForeignKey(AmountGiven, on_delete=models.CASCADE)
    amount = models.FloadField()

我必须分别获得amount given和due的前10位联系人。

我认为我正在过滤类似的数据

@api_view(http_method_names=['GET'])
def top_ten(request):
    filter_type = request.query_params.get('type', None)

    if filter_type == 'due':
        # query for due type

    elif filter_type == 'given':
        qs = Contact.objects.filter(
            user=request.user
        ).values('id').annotate(
            amount_given=Sum('amountgiven__amount')
        ).order_by(
            '-amount_given'
        )[:10]

        graph_data = []
        for q in qs:
            d = {}
            contact = Contact.objects.get(pk=q['id'])

            d['contact'] = contact.full_name if contact else 'Unknown'
            d['value'] = q['amount_given']
            graph_data.append(d)

        return Response(graph_data)

    else:
        raise NotFound('No data found for given filter type')

type查询可以到期或给出。

given类型的代码可以正常工作，因为所有字段都在数据库中。但是如何根据due类型的虚拟字段进行过滤？

我需要做的是annotate的{{1}}属性组的amount_due 总和。

Answer 1

您无法基于@property进行过滤。

据我正确理解您的问题，您可以将相关的AmountGiven与AmountReturned的总和进行汇总，然后计算due字段，该字段保留字母和前者相减的结果。

查询：

from django.db.models import Sum, Value
from django.db.models.functions import Coalesce

Contact.objects.filter(  
    amountgiven__amount__gt=0
).annotate(
    due=Sum('amountgiven__amount') - Coalesce(Sum('amountgiven__amountreturned__amount'), Value(0))
).order_by('-due').values_list('due', 'id')

将返回：

<QuerySet [{'id': 3, 'due': 2500.0}, {'id': 1, 'due': 2450.0}, {'id': 2, 'due': 1500.0}]>

但是，使用此解决方案，您无法在一个AmountGiven中区分多个Contact。您会看到类似结果的大图片。

如果您想为每个AmountGiven实例拆分 due 值，则只需进行如下注释：

AmountGiven.objects.annotate(
    due=Sum('amount') - Coalesce(Sum('amountreturned__amount'), Value(0))
).order_by('-due').values_list('due', 'contact__id', 'id')

返回

<QuerySet [
   {'contact__id': 3, 'id': 3, 'due': 2500.0},
   {'contact__id': 1, 'id': 1, 'due': 1750.0},
   {'contact__id': 2, 'id': 2, 'due': 1500.0},
   {'contact__id': 1, 'id': 4, 'due': 350.0},
   {'contact__id': 1, 'id': 5, 'due': 350.0}
]>

参考

Coalesce

Django在属性字段上注释

1 个答案: