2016-01-12 31 views
10

Próbuję uruchomić kwerendę agregacji, która jest w przybliżeniu równa:Django wyrażenie kwerendy dla obliczonych pól, które wymagają warunków i odlewania

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

Baza danych PostgreSQL jest stosowany.

że wykonane wyrażenie zapytania (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') 

są 2 problemy z nim:

  • ctr wynosi 0,0 ponieważ dzieli całkowite na poziomie bazy
  • generuje division by zero jeśli wyświetlenia to 0

Jakie jest właściwe rozwiązanie?

Odpowiedz

10

Korzystając conditional expressions:

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') 
+0

Działa, dzięki. Nazwy adnotacji mogą być takie same. – serg

0

Dokładna matematyki, w jaki sposób skalować to zależy od ciebie, ale dodając stałą Pythona wewnątrz zagnieżdżonej ExpressionWrapper powinno wystarczyć:

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

można użyć wyrażenie zapytania, które obsługuje dzielenie przez zero i zwraca wartości zmiennoprzecinkowe:

from pg_utils import divide 

Stats.objects.annotate(ctr=divide(F('clicks'), F('impressions')))