From 36ecf7fd5489e225df58bc79a5d6bb50a3ff4782 Mon Sep 17 00:00:00 2001
From: Christopher Neugebauer <chrisjrn@gmail.com>
Date: Sat, 26 Mar 2016 15:14:28 +1100
Subject: [PATCH] Adds more tags for the dashboard.

---
 .../templatetags/registrasion_tags.py         | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/registrasion/templatetags/registrasion_tags.py b/registrasion/templatetags/registrasion_tags.py
index 4357d16a..a6741c0e 100644
--- a/registrasion/templatetags/registrasion_tags.py
+++ b/registrasion/templatetags/registrasion_tags.py
@@ -1,11 +1,47 @@
 from registrasion import models as rego
 
+from collections import namedtuple
 from django import template
+from django.db.models import Sum
 
 register = template.Library()
 
+ProductAndQuantity = namedtuple("ProductAndQuantity", ["product", "quantity"])
 
 @register.assignment_tag(takes_context=True)
 def available_categories(context):
     ''' Returns all of the available product categories '''
     return rego.Category.objects.all()
+
+@register.assignment_tag(takes_context=True)
+def invoices(context):
+    ''' Returns all of the invoices that this user has. '''
+    return rego.Invoice.objects.filter(cart__user=context.request.user)
+
+@register.assignment_tag(takes_context=True)
+def items_pending(context):
+    ''' Returns all of the items that this user has in their current cart,
+    and is awaiting payment. '''
+
+    all_items = rego.ProductItem.objects.filter(
+        cart__user=context.request.user,
+        cart__active=True,
+    )
+    return all_items
+
+@register.assignment_tag(takes_context=True)
+def items_purchased(context):
+    ''' Returns all of the items that this user has purchased '''
+
+    all_items = rego.ProductItem.objects.filter(
+        cart__user=context.request.user,
+        cart__active=False,
+    )
+
+    products = set(item.product for item in all_items)
+    out = []
+    for product in products:
+        pp = all_items.filter(product=product)
+        quantity = pp.aggregate(Sum("quantity"))["quantity__sum"]
+        out.append(ProductAndQuantity(product, quantity))
+    return out