First statistics :)

2 files changed, 41 insertions, 3 deletions

diff --git a/app/views/__init__.py b/app/views/__init__.py
index b8ae6b3..4f079c3 100644
--- a/app/views/__init__.py
+++ b/app/views/__init__.py
@@ -50,14 +50,16 @@ def page_not_found (error):
     return render_template('404.jinja', page = request.path), 404
 
 # Now import the views
-from . import categories, consts, expenses, user, api
+from . import categories, consts, expenses, user, stats
+#from . import api
 
 app.register_blueprint(expenses.mod)
 app.register_blueprint(user.mod, url_prefix='/user')
 app.register_blueprint(consts.mod, url_prefix='/const')
 app.register_blueprint(categories.mod, url_prefix='/cat')
+app.register_blueprint(stats.mod, url_prefix='/stats')
 
-for m in api.mods:
-    app.register_blueprint(m, url_prefix='/api')
+#for m in api.mods:
+#    app.register_blueprint(m, url_prefix='/api')
 
 app.add_url_rule('/', endpoint = 'index', build_only = True)
diff --git a/app/views/stats.py b/app/views/stats.py
new file mode 100644
index 0000000..b1e9b1a
--- /dev/null
+++ b/app/views/stats.py
@@ -0,0 +1,36 @@
+from . import Blueprint, flash, db, \
+        current_user, login_required, \
+        assert_authorisation, templated, redirect, request, \
+        today
+
+from .. import forms as F
+from ..model import ConstExpense
+import sqlalchemy as sql
+import time
+from collections import defaultdict
+
+mod = Blueprint('stats', __name__)
+
+def next_date(d):
+    if d.month == 12:
+        return d.replace(year = d.year + 1, month = 1)
+    else:
+        return d.replace(month = d.month + 1)
+
+@mod.route('/')
+@login_required
+@templated
+def show():
+    # easy way: fetch them all and then do some computation
+    expenses = defaultdict(int)
+    t = next_date (today().replace(day = 1))
+    for e in ConstExpense.of(current_user):
+        cur = e.start
+        end = min(e.end, t)
+        while cur <= end:
+            expenses[time.mktime(cur.timetuple()) * 1000] += e.monthly
+            cur = next_date(cur)
+
+    expenses = list(sorted((int(k),v) for k,v in expenses.iteritems()))
+
+    return { 'consts': expenses }