Подтвердить что ты не робот

Выражение запроса Django для вычисленных полей, которые требуют условий и литья

Я пытаюсь запустить запрос агрегации, который примерно равен:

select 
    sum(impressions) as impressions, 
    sum(clicks) as clicks, 
    sum(clicks)/sum(impressions) as ctr 
from stats 
group by product
order by ctr;

Используемая база данных - PostgreSQL.

Я сделал это выражение запроса (Django 1.9):

Stats.objects.values('product').annotate(
    impressions = models.Sum('impressions'), 
    clicks = models.Sum('clicks'), 
    ctr = models.ExpressionWrapper(
        models.F('clicks')/models.F('impressions')), output_field = models.FloatField()
    )
).order_by('ctr')

Есть две проблемы:

  • ctr равен 0.0, поскольку он делит целые числа на уровне базы данных
  • он бросает division by zero, если показы 0

Какое правильное решение?

4b9b3361

ОТВЕТЫ

Ответ 1

Используйте условные выражения:

from django.db.models import Case, F, Sum, When

Stats.objects.values('product').annotate(
    tot_impressions=Sum('impressions'), 
    tot_clicks=Sum('clicks')
).annotate(
    ctr=Case(When(tot_impressions=0, then=None),  # or other value, e.g. then=0
             # 1.0*... is to get float in SQL
             default=1.0*F('tot_clicks')/F('tot_impressions'),
             output_field=models.FloatField())
).order_by('ctr')

Ответ 2

Точная математика о том, как вы масштабируете это, зависит от вас, но достаточно добавить константу python внутри вложенного ExpressionWrapper:

ctr = models.ExpressionWrapper(
    models.F('clicks')/models.ExpressionWrapper(
        models.F('impressions') + 1, output_field = models.FloatField()
    ),
    output_field = models.FloatField()
)