#!/usr/bin/env python
# -*-coding:utf-8-*-

import sys
try:
    import random
except BaseException:
    sys.exit("Import failed in module libStatSitools :\n\trandom module is required")
try:
    import cairo
except BaseException:
    sys.exit("Import failed in module libStatSitools :\n\tcairo module is required")
try:
    import pycha.pie as pie
except BaseException:
    sys.exit("Import failed in module libStatSitools :\n\tpycha module is required")
try:
    from datetime import datetime
except BaseException:
    sys.exit("Import failed in module libStatSitools :\n\tdatetime module is required")
try:
    import pygal
except BaseException:
    sys.exit("Import failed in module libStatSitools :\n\tpygal module is required")
try:
    from dateutil.relativedelta import relativedelta
except:
    sys.exit("Import failed in module libStatSitools :\n\tdateutil.relativedelta module is required")

#PATH FOR DEV
home_base = "/home/marc/MyDev/sitools-idoc/webstatDev/idoc_Maison"
#PATH FOR PROD
#home_base = "/var/www/django/mon_projet/"

url_png = "/static/chart/"

time_format_for_by_month = "%Y %m"

def create_dataset_from_data(dico_data):
    graph_dataset = [(data, [[0, dico_data[data]]]) for data in dico_data]
    return graph_dataset


def add_date():
    now = datetime.now()
    tmp = str(now).split(" ")
    date_to_add = tmp[0].split("-")[0]+tmp[0].split("-")[1]+tmp[0].split("-")[2]+tmp[1].split(".")[0].split(":")[0]+tmp[1].\
        split(".")[0].split(":")[1]+tmp[1].split(".")[0].split(":")[2]+tmp[1].split(".")[1]
    return date_to_add


def svg_chart(dico_data, title_chart):

    pie_chart_pygal = pygal.Pie()
    pie_chart_pygal.title = title_chart
    date = add_date()

    url_output_svg = url_png+title_chart+date+".svg"
    print "**********************************************"
    print "*************  Nb entrée pour", title_chart, " : ", len(dico_data)
    print "**********************************************"
    dico_data_tmp = dico_data.copy()
    if len(dico_data) > 24:
        dico_data2 = {}
        i = 0
        for key in dico_data_tmp:
            if i > 24:
                dico_data2[key] = dico_data_tmp[key]
                del dico_data[key]
            i += 1


    for data in dico_data:
            pie_chart_pygal.add(data, dico_data[data])

    if len(dico_data_tmp) > 24:
        for datas in dico_data2:
            pie_chart_pygal.add(datas, dico_data2[datas], secondary=True)

    output_svg = home_base+url_output_svg
    try:
        pie_chart_pygal.render_to_file(output_svg)
    except BaseException, e:
        print 'In svg Exception : ', str(e)

    url_chart_png = png_chart(dico_data, title_chart)
    return [url_output_svg, url_chart_png]


def png_chart(dico_data, title_chart):

    pie_chart_pygal_png = pygal.Pie()
    pie_chart_pygal_png.title = title_chart
    
    date = add_date()
    url_output_png = url_png+title_chart+date+".png"
    for data in dico_data:
        pie_chart_pygal_png.add(data, dico_data[data])
    output_png = home_base+url_output_png
    try:
        pie_chart_pygal_png.render_to_png(output_png)
    except BaseException, e:
        print 'In PNG Exception : ', str(e)
    return url_output_png


def svg_bar_chart(dico_data, start_date, end_date, title_chart):

    dates_tmp_not_sorted = []
    dates_tmp_str_sorted = []
    dates_sorted = []
    dico_data2 = {}
    for date in dico_data.keys():
        dates_tmp_not_sorted.append(str(date[1])+"/"+str(date[0]))

    for dd in sorted(dates_tmp_not_sorted, key=lambda d: map(int, d.split('/'))):
        dates_tmp_str_sorted.append(dd.split('/')[1]+"/"+dd.split('/')[0][2:])

#    date_min = datetime.strptime(dates_tmp_str_sorted[0], "%m/%y")
#    date_max = datetime.strptime(dates_tmp_str_sorted[len(dates_tmp_str_sorted)-1], "%m/%y")

    for i in range(0, int(diff_month(start_date, end_date)+1)):
        t = datetime.combine(start_date, datetime.min.time())+relativedelta(months=i)
        dates_sorted.append(t)

    for x in dates_tmp_str_sorted:
        date_test = datetime.strptime(str(x), "%m/%y")
        dico_data2[date_test] = dico_data[(date_test.month, date_test.year)]

    list_tmp = []
    list_tmp2 = []
    list_tmp.append("")

    bar_chart = pygal.DateY(x_label_rotation=90, human_readable=True, y_scale=1073741824)
    bar_chart.title = title_chart
    bar_chart.x_label_format = "%m/%Y"
    bar_chart.x_labels = dates_sorted
    print "************* dates_sorted : ", dates_sorted
    try:
        for data in dates_sorted:
            if data not in dico_data2.keys():
                dico_data2[data] = 0
            list_tmp2.append((data, dico_data2[data]))
        list_tmp.append(list_tmp2)

        bar_chart.add(list_tmp[0],  (list_tmp[1][i] for i in range(0, len(list_tmp[1]))))
        bar_chart.show_legend = False
        date_svg = add_date()
        url_output_bar_chart_svg = url_png+"Bar_by_Month_"+date_svg+".svg"
        bar_chart.render_to_file(home_base+url_output_bar_chart_svg)
        url_output_bar_chart_png = url_bar_chart_png(list_tmp, dates_sorted, title_chart)
    except BaseException, e:
        print str(e)
    return [url_output_bar_chart_svg, url_output_bar_chart_png]


def url_bar_chart_png(list_data, dates, title):
    bar_chart = pygal.DateY(x_label_rotation=90)
    bar_chart.title = title
    bar_chart.x_label_format = "%m/%Y"
    bar_chart.x_labels = dates
    bar_chart.add(list_data[0],  (list_data[1][i] for i in range(0, len(list_data[1]))))
    # A METTRE POUR CREER LE PNG !
    bar_chart.print_values = False
    bar_chart.show_legend = False
    url_output_png = url_png+"Bar_by_Month_"+add_date()+".png"
    bar_chart.render_to_png(home_base+url_output_png)
    return url_output_png


def diff_month(d2, d1):
    return (d1.year - d2.year)*12 + d1.month - d2.month