5 - Get Data by Custom QuerySet

Author: codingforentrepreneurs

src/analytics/templates/analytics/sales.html

@@ -12,7 +12,7 @@ <h1>Sales Data</h1>
 
 <div class='row'>
     <div class='col'>
-        <p>Recent Total: ${{ recent_orders_total.total__sum }}</p>
+        <p>Recent Total: ${{ recent_orders_data.total__sum }}</p>
         <ol>
         {% for order in recent_orders %}
             <li>{{ order.order_id }}
@@ -22,7 +22,7 @@ <h1>Sales Data</h1>
     </ol>
     </div>
     <div class='col'>
-        <p>Shipped</p>
+        <p>Shipped Total: ${{ shipped_orders_data.total__sum }}</p>
         <ol>
         {% for order in shipped_orders %}
              <li>{{ order.order_id }}
@@ -32,7 +32,7 @@ <h1>Sales Data</h1>
         </ol>
     </div>
     <div class='col'>
-        <p>Paid</p>
+        <p>Paid Totals: ${{ paid_orders_data.total__sum }}</p>
         <ol>
         {% for order in paid_orders %}
              <li>{{ order.order_id }}

src/analytics/views.py

@@ -23,18 +23,11 @@ def get_context_data(self, *args, **kwargs):
         qs = Order.objects.all()
         context['orders'] = qs
         context['recent_orders'] = qs.recent().not_refunded()[:5]
-        context['recent_orders_total'] = context['recent_orders'].aggregate(
-                                        Sum("total"), 
-                                        Avg("total"), 
-                                        # Avg("cart__products__price"), 
-                                        # Count("cart__products")
-                                    )
-        # context['recent_cart_data'] = context['recent_orders'].aggregate(
-        #                                 Avg("cart__products__price"), 
-        #                                 Count("cart__products")
-        #                             )
-        # qs = Order.objects.all().aggregate(Sum("total"), Avg("total"), Avg("cart__products__price"), Count("cart__products"))
-        # ann = qs.annotate(product_avg=Avg('cart__products__price'), product_total = Sum('cart__products__price'), product__count = Count('cart__products'))
-        context['shipped_orders'] = qs.recent().not_refunded().by_status(status='shipped')[:5]
-        context['paid_orders'] = qs.recent().not_refunded().by_status(status='paid')[:5]
+        context['recent_orders_data'] = context['recent_orders'].totals_data()
+        context['recent_orders_cart_data'] = context['recent_orders'].cart_data()
+        context['shipped_orders'] = qs.recent().not_refunded().by_status(status='shipped')[:1]
+        context['shipped_orders_data'] =  context['shipped_orders'].totals_data()
+        context['paid_orders'] = qs.recent().not_refunded().by_status(status='paid')[:4]
+        context['paid_orders_data'] = context['paid_orders'].totals_data()
+
         return context

src/orders/models.py

@@ -1,6 +1,7 @@
 import math
 from django.conf import settings
 from django.db import models
+from django.db.models import Count, Sum, Avg
 from django.db.models.signals import pre_save, post_save
 from django.core.urlresolvers import reverse
 
@@ -22,6 +23,16 @@ class OrderManagerQuerySet(models.query.QuerySet):
     def recent(self):
         return self.order_by("-updated", "-timestamp")
 
+    def totals_data(self):
+        return self.aggregate(Sum("total"), Avg("total"))
+
+    def cart_data(self):
+        return self.aggregate(
+                        Sum("cart__products__price"), 
+                        Avg("cart__products__price"), 
+                        Count("cart__products")
+                                    )
+
     def by_status(self, status="shipped"):
         return self.filter(status=status)