From a87ea6cf5865a74d56aa59754cb26c4833039e61 Mon Sep 17 00:00:00 2001
From: Bernhard Stampfer <stampfer@mail.com>
Date: Mon, 15 Feb 2016 20:14:37 +0100
Subject: [PATCH] First plot: beer over time - beer still hardcoded at index 1

---
 app/plot.py  | 77 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 app/views.py |  5 +++-
 2 files changed, 81 insertions(+), 1 deletion(-)
 create mode 100644 app/plot.py

diff --git a/app/plot.py b/app/plot.py
new file mode 100644
index 0000000..7c3fcd3
--- /dev/null
+++ b/app/plot.py
@@ -0,0 +1,77 @@
+from matplotlib import pyplot as plt
+from matplotlib.dates import WeekdayLocator, DayLocator, HourLocator, DateFormatter, drange, MONDAY
+import numpy as np
+from user import User
+from database import *
+
+def plot_total(user = None):
+
+    today = datetime.date.today()
+    delta = datetime.timedelta(days=1)
+    begin = datetime.date.today() - datetime.timedelta(weeks=4)
+    dates = drange(begin, today, delta)
+
+#    all_consumptions = list(10)
+ #   for consumptions in all_consumptions: #todo fix
+    consumptions = np.zeros(len(dates))
+
+    #print consumptions
+
+    consumed = get_consumed()
+    for consumption in consumed:
+        if consumption.prodnr == 1:
+            if user == None or consumption.consumer == user.id:
+                if consumption.time.date() > begin:
+                    consumptions[(consumption.time.date() - begin).days - 1] += 1
+    plt.xkcd()
+
+
+    fig, ax = plt.subplots()
+
+    ax.plot(dates, consumptions, linestyle='-', marker='')
+
+   # ax.axes.margins = 1  # x margin.  See `axes.Axes.margins`
+   # ax.axes.ymargin = 1  # y margin See `axes.Axes.margins`
+
+    plt.xticks(rotation='vertical')
+
+    ax.spines['right'].set_visible(False)
+    ax.spines['top'].set_visible(False)
+
+    ax.yaxis.set_ticks_position('left')
+    ax.xaxis.set_ticks_position('bottom')
+
+    plt.tick_params(which='minor', length=4)
+    plt.tick_params(which='major', length=5)
+
+    #ax.xaxis.set_major_locator(WeekdayLocator(MONDAY))
+    #ax.xaxis.set_major_locator(DayLocator())
+    ax.xaxis.set_major_formatter(DateFormatter('%d.%m'))
+    #ax.xaxis.set_minor_formatter(DateFormatter('%d.%m'))
+
+    #ax.fmt_xdata = DateFormatter('%d.%m')
+    fig.autofmt_xdate()
+
+    plt.annotate(
+        'THE DAY I REALIZED\nI COULD COOK BACON\nWHENEVER I WANTED',
+        xy=(30, 1), arrowprops=dict(arrowstyle='->'), xytext=(15, -10))
+
+    plt.xlabel('Datum')
+    plt.ylabel('Halbe')
+
+    if user == None:
+        tit = "Bierkonsum FET"
+        fils = "app/static/total.png"
+        fill = "app/static/total_big.png"
+    else:
+        tit = "Bierkonsum %s" % user.name
+        fils = "app/static/total%03d.png" % user.id
+        fill = "app/static/total%03d_big.png" % user.id
+    plt.title(tit)
+
+    #480x320
+    fig.set_size_inches(4.8, 3.2)
+    plt.savefig(fils, dpi=100)
+
+    fig.set_size_inches(4.8, 3.2)
+    plt.savefig(fill, dpi=400)
diff --git a/app/views.py b/app/views.py
index 396890d..cd298bf 100644
--- a/app/views.py
+++ b/app/views.py
@@ -2,6 +2,7 @@ from check_rights import *
 from flask import render_template, request, redirect, session, send_from_directory
 from app import app
 from database import *
+from plot import *
 from user import User
 from product import Product
 from consumption import Consumption
@@ -17,6 +18,7 @@ def static_proxy(path):
 @app.route('/index')
 def index():
     consumed = get_consumed()
+    plot_total()
     return render_template("index.html", consumed=consumed, user=get_user_by_name(session.get('name')))
 
 
@@ -147,9 +149,10 @@ def consume():
         username = session.get('name')
         add_consume(username, prod.id)
         message = "Du hast gerade ein %s konsumiert." % prod.name
+        plot_total(get_user_by_name(session.get('name')))
+        plot_total()
     return render_template('consume.html', products=products, message=message, user=get_user_by_name(session.get('name')))
 
-
 @app.route('/billing')
 @requires_baron
 def billing():