Django在属性字段上注释

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的前十名联系人。

在我看来，我正在过滤数据

@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的amount_due contact财产集团的总和。

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实例分割到期值，那么就像这样注释：

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}
]> 

参考

1. Coalesce