From a6f71195fd240fa8b3833841f22595a7e9ec1bdd Mon Sep 17 00:00:00 2001
From: Wout Schellaert <wout.schellaert@gmail.com>
Date: Sat, 10 Sep 2016 16:52:46 +0200
Subject: [PATCH] First data

---
 app/fatmodels.py         | 20 ++++++++++++++++++++
 app/templates/stats.html | 12 ++++++++++--
 app/views/stats.py       | 12 ++++++++++--
 3 files changed, 40 insertions(+), 4 deletions(-)
 create mode 100644 app/fatmodels.py

diff --git a/app/fatmodels.py b/app/fatmodels.py
new file mode 100644
index 0000000..3b684ad
--- /dev/null
+++ b/app/fatmodels.py
@@ -0,0 +1,20 @@
+from models import User, Location, Order, OrderItem
+
+
+class CountableModel:
+
+    @classmethod
+    def amount(cls):
+        return cls.query.count()
+
+
+class FatLocation(Location, CountableModel): pass
+
+
+class FatOrder(Order, CountableModel): pass
+
+
+class FatUser(User, CountableModel): pass
+
+
+class FatOrderItem(OrderItem, CountableModel): pass
diff --git a/app/templates/stats.html b/app/templates/stats.html
index 5518a47..035c3c2 100644
--- a/app/templates/stats.html
+++ b/app/templates/stats.html
@@ -1,6 +1,14 @@
 {% extends "layout.html" -%}
 {% set active_page = "stats" -%}
 {% block container %}
-  <h1>Stats bruh</h1>
-  <p>TOP 4</p>
+  <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.
+    </h5>
+  </div>
 {% endblock %}
\ No newline at end of file
diff --git a/app/views/stats.py b/app/views/stats.py
index f235d07..d498665 100644
--- a/app/views/stats.py
+++ b/app/views/stats.py
@@ -1,7 +1,15 @@
-from app import app
 from flask import render_template
 
+from fatmodels import FatLocation, FatOrder, FatOrderItem, FatUser
+from app import app
+
 
 @app.route('/stats/')
 def stats():
-    return render_template('stats.html')
+    data = {
+        'order_amount': FatOrder.amount(),
+        'location_amount': FatLocation.amount(),
+        'user_amount': FatUser.amount(),
+        'orderitem_amount': FatOrderItem.amount()
+    }
+    return render_template('stats.html', data=data)