From ba33a05f6650a6235f08fcbcd19975619e599fd3 Mon Sep 17 00:00:00 2001
From: Wout Schellaert <wout.schellaert@gmail.com>
Date: Sat, 10 Sep 2016 23:01:13 +0200
Subject: [PATCH] Add top4 products

---
 app/fatmodels.py         | 46 ++++++++++++++++++++++++++++++++++------
 app/templates/stats.html | 10 ++++-----
 app/views/stats.py       | 16 +++++++++-----
 3 files changed, 56 insertions(+), 16 deletions(-)

diff --git a/app/fatmodels.py b/app/fatmodels.py
index 3b684ad..9f7673f 100644
--- a/app/fatmodels.py
+++ b/app/fatmodels.py
@@ -1,20 +1,54 @@
-from models import User, Location, Order, OrderItem
+from models import User, Location, Order, OrderItem, Product
+from sqlalchemy.sql import func, desc
 
 
-class CountableModel:
+class FatModel:
+
+    @classmethod
+    def all(cls):
+        return cls.query.all()
 
     @classmethod
     def amount(cls):
         return cls.query.count()
 
 
-class FatLocation(Location, CountableModel): pass
+class FatLocation(Location, FatModel): pass
 
 
-class FatOrder(Order, CountableModel): pass
+class FatOrder(Order, FatModel):
+
+    # It's hard to add the unique user constraint,
+    #  as DISTINCT seems to apply after a GROUP BY and aggregate
+    # So DISTINCT ... count(user_id) ... will count all users,
+    #  even if they get reduced by the disctinct afterwards.
+    @classmethod
+    def items_per_order(cls):
+        return Order.query.\
+            join(OrderItem).\
+            group_by(Order.id).\
+            with_entities(Order.id, func.count(OrderItem.user_id).label('total'))
 
 
-class FatUser(User, CountableModel): pass
+class FatUser(User, FatModel): pass
 
 
-class FatOrderItem(OrderItem, CountableModel): pass
+class FatOrderItem(OrderItem, FatModel): pass
+
+
+class FatProduct(Product, FatModel):
+
+    @classmethod
+    def top4(cls):
+        top4 = OrderItem.query\
+            .join(Product)\
+            .join(Location)\
+            .group_by(Product.id)\
+            .with_entities(Product.name,
+                           Location.name,
+                           func.count(Product.id).label('count')
+                           )\
+            .order_by(desc('count'))\
+            .limit(4)
+        for top in top4:
+            print(top)
\ No newline at end of file
diff --git a/app/templates/stats.html b/app/templates/stats.html
index 035c3c2..454fefa 100644
--- a/app/templates/stats.html
+++ b/app/templates/stats.html
@@ -4,11 +4,11 @@
   <h2>Stats bruh</h2>
   <div class="jumbotron">
     <h5>
-      In
-      <strong>{{ data.order_amount }}</strong> orders,
-      <strong>{{ data.user_amount }}</strong> users have ordered
-      <strong>{{ data.orderitem_amount }}</strong> items in
-      <strong>{{ data.location_amount }}</strong> locations.
+      Over
+      <strong>{{ data.amount.orders }}</strong> orders,
+      <strong>{{ data.amount.users }}</strong> users have ordered
+      <strong>{{ data.amount.orderitem }}</strong> items in
+      <strong>{{ data.amount.location }}</strong> locations.
     </h5>
   </div>
 {% endblock %}
\ No newline at end of file
diff --git a/app/views/stats.py b/app/views/stats.py
index d498665..0a72969 100644
--- a/app/views/stats.py
+++ b/app/views/stats.py
@@ -1,15 +1,21 @@
 from flask import render_template
 
-from fatmodels import FatLocation, FatOrder, FatOrderItem, FatUser
+from fatmodels import FatLocation, FatOrder, FatOrderItem, FatUser, FatProduct
 from app import app
 
 
 @app.route('/stats/')
 def stats():
     data = {
-        'order_amount': FatOrder.amount(),
-        'location_amount': FatLocation.amount(),
-        'user_amount': FatUser.amount(),
-        'orderitem_amount': FatOrderItem.amount()
+        'amount': {
+            'orders': FatOrder.amount(),
+            'locations': FatLocation.amount(),
+            'users': FatUser.amount(),
+            'orderitems': FatOrderItem.amount(),
+            'products': FatProduct.amount(),
+        },
+        'top4': {
+            'products': FatProduct.top4()
+        }
     }
     return render_template('stats.html', data=data)