78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
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)
|