arona/PyRIGS
1
0
Fork 0
mirror of https://github.com/nottinghamtec/PyRIGS.git synced 2026-01-17 05:22:16 +00:00
Code Issues Projects Releases Wiki Activity

Optimise down to just the one SQL query for waiting invoices.

Browse Source 
Cuts down on all the unnecessary queries and selects everything in one go.
Down from ~6 queries per invoice
This commit is contained in:
Tom Price
2017-03-01 02:42:39 +00:00
parent 4773b43081
commit 7ec1c726a6
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
RIGS/finance.py
View File

@@ -1,6 +1,7 @@
import cStringIO as StringIO
import datetime
import re
from decimal import Decimal
from django.contrib import messages
from django.core.urlresolvers import reverse_lazy
@@ -31,12 +32,13 @@ class InvoiceIndex(generic.ListView):
def get_queryset(self):
query = self.model.objects.annotate(
cost=Sum(F('event__items__cost') * F('event__items__quantity'), output_field=FloatField()),
payment_count=Count('payment'),
payments=Sum('payment__amount'),
_sum_total=Sum(F('event__items__cost') * F('event__items__quantity'), output_field=FloatField()),
_payment_count=Count('payment'),
_payment_total=Sum('payment__amount'),
).filter(
Q(cost__gt=0.0, payment_count=0) | Q(cost__lt=Value('payments'), cost__gt=Value('payments'))
).select_related('event', 'event__organisation', 'event__person', 'event__venue')
Q(_sum_total__gt=0.0, _payment_count=0) |
Q(_sum_total__lt=Value('_payment_total'), _sum_total__gt=Value('_payment_total'))
).select_related('event', 'event__organisation', 'event__person', 'event__venue', 'event__mic')
return query

8
RIGS/models.py
View File

@@ -1,11 +1,11 @@
import datetime
import hashlib
import pytz
import random
import string
from collections import Counter
from decimal import Decimal
import pytz
import reversion
from django.conf import settings
from django.contrib.auth.models import AbstractUser
@@ -509,6 +509,8 @@ class Invoice(models.Model):
@property
def sum_total(self):
if getattr(self, '_sum_total', None):
return Decimal(getattr(self, '_sum_total'))
return self.event.sum_total
@property
@@ -517,6 +519,10 @@ class Invoice(models.Model):
@property
def payment_total(self):
if hasattr(self, '_payment_total') and hasattr(self, '_payment_count'):
if getattr(self, '_payment_count') == 0:
return Decimal(0.00)
return Decimal(getattr(self, '_payment_total', 0.00))
total = self.payment_set.aggregate(total=models.Sum('amount'))['total']
if total:
return total