baroness/app/plot.py

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 *
import thread as th

def plot_all_thread(user = None):
    #if user != None:
    #    th.start_new_thread(plot_all, (user,))
    #else:
    #    th.start_new_thread(plot_all, ())
#def plot_all(user = None):
#
    if user != None:
        plot_total(user)
    plot_total()
    plot_list(4)
    print 'plot_all'

def plot_total(user = None):

    print 'plot_total'
    today = datetime.date.today()
    delta = datetime.timedelta(days=1)
    begin = today - datetime.timedelta(weeks=2)
    dates = drange(begin, today + delta, delta)

    #print begin
    #print today
    #print dates

    products = get_products()
    allconsumptions = [[0 for x in range(len(dates))] for product in products]

    consumed = get_consumed()
    for consumption in consumed:
        if user == None or consumption.consumer == user.id:
            i = 1
            for consumptions in allconsumptions:
                if consumption.prodnr == i:
                    if consumption.time.date() > begin:
                        consumptions[(consumption.time.date() - begin).days] += 1
                i += 1
    plt.xkcd()

    #print allconsumptions
    fig, ax = plt.subplots()

    i = 0
    for consumptions in allconsumptions:
        ax.plot(dates, consumptions, linestyle='-', marker='', label=products[i].name)
        i += 1

    ax.legend(loc=2,prop={'size':15})

   # 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('Konsumate')

    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)

def plot_list(duration):

    today = datetime.date.today()
    begin = today - datetime.timedelta(weeks=duration)

    users = get_users()
    products = get_products()
    #consumptions = [0 for user in users]
    allconsumptions = [[0 for user in users] for product in products]

    consumed = get_consumed()
    for consumption in consumed:
        allconsumptions[consumption.prodnr-1][consumption.consumer-1] += 1

    #cumulate consumptions for cumulative bar graph
    i = 0
    for consumptions in allconsumptions:
        if i > 0:
            j = 0
            for consumption in consumptions:
                allconsumptions[i][j] += allconsumptions[i-1][j]
                j += 1
        i += 1

    plt.xkcd()

    fig, ax = plt.subplots()

    colors = ['blue', 'green', 'red', 'yellow', 'orange' , 'black']

    #plot reversed to print longest bar lowest
    i = len(allconsumptions) - 1
    for consumptions in reversed(allconsumptions):
        ax.barh(np.arange(len(consumptions)), consumptions, label=products[i].name, align='center', height=(0.5), color=colors[i])
        i -= 1

    names = list()

    for user in users:
        names.append(user.longname)

    plt.yticks(np.arange(len(names)), names)

    ax.legend(loc=2)

    ax.spines['right'].set_visible(False)
    ax.spines['top'].set_visible(False)

    ax.yaxis.set_ticks_position('none')
    ax.xaxis.set_ticks_position('none')
    plt.subplots_adjust(left=0.15)
    #plt.tick_params(which='minor', length=4)
    #plt.tick_params(which='major', length=5)

    #plt.annotate(
    #    'THE DAY I REALIZED\nI COULD COOK BACON\nWHENEVER I WANTED',
    #    xy=(30, 1), arrowprops=dict(arrowstyle='->'), xytext=(15, -10))

    plt.xlabel('Konsumate')
    #plt.ylabel('Konsumate')

    plt.title("Bierliste")

    #1024x768
    #fig.set_size_inches(10.24, 7.68)
    #plt.savefig('app/static/bierliste.png', dpi=100)
    #800x600
    fig.set_size_inches(15, 10)
    plt.savefig('app/static/bierliste_small.png', dpi=72)