带有条件函数的Django查询何时(?)

时间:2017-02-20 09:00:00

标签: python django postgresql django-queryset

我有一个用餐者的数据库:

class Product(models.Model):
    name = models.CharField(max_length=250)
    sides = models.ManyToManyField(Sides, blank=True)
    price = models.DecimalField(max_digits=5, decimal_places=2)

    def __unicode__(self):
        return '%s %s %s' % (self.name, self.sides, self.price)

    def __repr__(self):
        return unicode(self).encode('utf-8')


class Menu(models.Model):
    date = models.DateField()
    product = models.ForeignKey(Product)

    def __unicode__(self):
        return '%s %s' % (self.date, self.product)

    def __repr__(self):
        return unicode(self).encode('utf-8')


class Order(models.Model):
    date = models.DateField()
    user = models.ForeignKey(User, null=True)

    def __unicode__(self):
        return '%s %s' % (self.date, self.user)

    def __repr__(self):
        return unicode(self).encode('utf-8')


class OrderItems(models.Model):
    order = models.ForeignKey(Order)
    product = models.ForeignKey(Menu, null=True, blank=True)
    quantity = models.IntegerField(default=0)
    take_away = models.BooleanField(default=False)

    def __unicode__(self):
        return '%s %s %s %s' % (self.order, self.product, self.quantity, self.take_away)

    def __repr__(self):
        return unicode(self).encode('utf-8')

我需要在数据库中搜索我的用户在给定时间内订购的菜肴的费用,因此第一个查询非常简单:

report = OrderItems.objects.filter(Q(order__date__range=(strfrom, strto)), ~Q(order__user_id=None))

问题始于计算此期间每位用户的总费用,因为我必须为每个标记为带走的项目添加1(布尔字段)。到目前为止,我的查询忽略了布尔字段,就像那​​样:

per_person = report.values('order__user').annotate(per_capita=Sum(F('quantity')*(F('product__product__price')), output_field=DecimalField()))

我的问题是:如何在布尔字段标记为True的每个商品价格中添加1?我假设我必须使用条件函数'When'。

我尝试过类似的事情:

report.values('order__user').annotate(per_capita=Sum(F('quantity')*(F(When('take_away'==True, then='product__product__price')+1)|(When('take_away'==False, then='product__product__price'))), output_field=DecimalField()))

但是,很明显,它会给出错误__init__() takes either a Q object or lookups as keyword arguments

1 个答案:

答案 0 :(得分:1)

你可以使用Case-When。查看https://docs.djangoproject.com/en/1.10/ref/models/conditional-expressions/#case

中的示例
.annotate(per_capita=Sum(
    F('quantity') * (F('product__product__price') +
    Case(When(take_away=True, then=Value(1)),
         default_case=Value(0),
         output_field=DecimalField())
    )
))