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)