import pandas
import os
from __future__ import division

root="/projects/d2e32407-baf4-4631-8512-ce12e859eb9c/Priprema DataSeta/Dionice"
file_list=[]
for file in os.listdir(root):
    if file.endswith(".xlsx"):
        file_list.append(os.path.join(root, file))
#print(file_list)

data = {}
for path in file_list:
    base_name_without_ext = os.path.basename(path)
    base_name_without_ext = os.path.splitext(base_name_without_ext)[0]
    try:
        data[base_name_without_ext] = pandas.read_excel(open(path,'rb'), sheetname='Povijesni podaci').to_dict(orient='index')
    except Exception as e:
        print(e, base_name_without_ext)
        pass

root="/projects/d2e32407-baf4-4631-8512-ce12e859eb9c/Priprema DataSeta"
file_list=[]
for file in os.listdir(root):
    if file.endswith(".xlsx"):
        file_list.append(os.path.join(root, file))
print(file_list)

br_dionica = {}
for path in file_list:
    base_name_without_ext = os.path.basename(path)
    base_name_without_ext = os.path.splitext(base_name_without_ext)[0]
    try:
        br_dionica[base_name_without_ext] = pandas.read_excel(open(path,'rb'), sheetname='Sheet1').to_dict(orient='index')
    except Exception as e:
        print(e, base_name_without_ext)
        pass
br_dionica['dionice'][0]

Sastav_Crobex10 = ['zse_export_RIVP-R-A','zse_export_ADPL-R-A','zse_export_ADRS-P-A','zse_export_PODR-R-A','zse_export_ATGR-R-A',
                    'zse_export_ATPL-R-A','zse_export_ERNT-R-A','zse_export_HT-R-A','zse_export_KOEI-R-A','zse_export_LEDO-R-A']

Sastav_CrobexTurist =['zse_export_RIVP-R-A','zse_export_MAIS-R-A','zse_export_LRH-R-A','zse_export_ARNT-R-A']

Sastav_Crobex_Industrija =['zse_export_ADPL-R-A','zse_export_DDJH-R-A','zse_export_ERNT-R-A','zse_export_INA-R-A','zse_export_KOEI-R-A',
                           'zse_export_PTKM-R-A','zse_export_VART-R-1','zse_export_VLEN-R-B']

Sastav_Crobex_Konstrukt =['zse_export_DLKV-R-A','zse_export_IGH-R-A','zse_export_INGR-R-A','zse_export_THNK-R-A']

import numpy as np
import plotly.offline as py
import plotly.graph_objs as go

py.init_notebook_mode()

def draw1(data, name1, _color = 'rgb(205, 12, 24)', ):
    trace = go.Scatter(
    x = [x[0] for x in data],
    y = [x[1] for x in data], #value
    name = name1,
    line = dict(
        color = (_color),
        width = 4))
    data = [trace]
    # Edit the layout
    layout = dict(title = name1,
              xaxis = dict(title = 'Datum'),
              yaxis = dict(title = 'Vrijednost'),
              )
    fig = dict(data=data, layout=layout)
    py.iplot(fig, filename=name1)
    
def _draw2(data1, data2, name1, name2):
    trace1 = go.Scatter(
    x = [x[0] for x in data1],
    y = [x[1] for x in data1], #value
    name = name1 ,
    line = dict(
        color = ('rgb(205, 12, 24)'),
        width = 4))
    trace2 = go.Scatter(
    x = [x[0] for x in data2],
    y = [x[1] for x in data2], #value
    name = name2,
    line = dict(
        color = ('rgb(33, 136, 220)'),
        width = 4))
    data = [trace1, trace2]
    # Edit the layout
    layout = dict(title = _title,
              xaxis = dict(title = 'Datum'),
              yaxis = dict(title = 'Vrijednost'),
              )
    fig = dict(data=data, layout=layout)
    py.iplot(fig, filename=name1)
    
    
def draw4(data1, data2, data3, data4, _title = 'Razlike'):
    trace1 = go.Scatter(
    x = [x[0] for x in data1],
    y = [x[1] for x in data1], #value
    name = 'Avg',
    line = dict(
        color = ('rgb(205, 12, 24)'),
        width = 4))
    trace2 = go.Scatter(
    x = [x[0] for x in data2],
    y = [x[1] for x in data2], #value
    name = 'Max',
    line = dict(
        color = ('rgb(255, 136, 0)'),
        width = 4))
    trace3 = go.Scatter(
    x = [x[0] for x in data3],
    y = [x[1] for x in data3], #value
    name = 'Min',
    line = dict(
        color = ('rgb(2, 136, 250)'),
        width = 4))
    data = [trace1, trace2, trace3]
    # Edit the layout
    layout = dict(title = _title,
              xaxis = dict(title = 'Datum'),
              yaxis = dict(title = 'Vrijednost'),
              )
    fig = dict(data=data, layout=layout)
    py.iplot(fig, filename=_title)

#def _draw4(data1, data2, data3, data4):

#Maksimalna cijena ikada postignuta
def MAX(price):
    value = 0
    mprice = [x[1] for x in price]
    for i in range(len(mprice)):
        if (mprice[i] > value):
            value = mprice[i]
    return value

#normalizirane cijene za usporedbu
def fnormalize(price, value):
    mprice = [x[1] for x in price]
    date = [x[0] for x in price]
    normprice = [(i/value)*100 for i in mprice]
    return list(zip(date, normprice))


def analiza(dionica):
    
    prosjecna = [] #cijena transakcije
    kolicina = [] #kolicina dionica s kojima se trgovalo u jednom danu
    br_tr = [] #broj transakcija
    minn = [] #najniža dnevna cijena
    maxi = [] #najviša dnevna cijena
    prva = [] #prva cijena
    zadnja = [] #zadnja cijena
    promet = [] #dnevni promet u kn
    promjena = [] #promjena zadnje cijene u odnosu na zadnju cijenu od dana prije

    #samo za normalne transakcije(???), one kojim ćemo mi trgovati
    for i in range(len(data[dionica])):
        if data[dionica][i][u'Vrsta prometa'] == 'N':
            prosjecna.append((data[dionica][i][u'Datum'],data[dionica][i][u'Prosječna']))
            kolicina.append((data[dionica][i][u'Datum'],data[dionica][i][u'Količina']))     
            br_tr.append((data[dionica][i][u'Datum'],data[dionica][i][u'Broj transakcija'])) 
            minn.append((data[dionica][i][u'Datum'],data[dionica][i][u'Najniža']))     
            maxi.append((data[dionica][i][u'Datum'],data[dionica][i][u'Najviša']))     
            prva.append((data[dionica][i][u'Datum'],data[dionica][i][u'Prva'])) 
            zadnja.append((data[dionica][i][u'Datum'],data[dionica][i][u'Zadnja'])) 
            promet.append((data[dionica][i][u'Datum'],data[dionica][i][u'Promet']))
            promjena.append((data[dionica][i][u'Datum'],data[dionica][i][u'Promjena']))
    
    #za vizualizaciju sve odrežemo na manje od 100
    maksimum = MAX(prosjecna)
    normprosjecna = fnormalize(prosjecna,maksimum)
    
    #crtanje
    #0. prosjecna
    #draw1(normprosjecna, 'Prosjecna')
    #1. min prosječna max (količina kao Bar)
    #2. prva prosječna zadnja
    #3. razlika najmanje i najviše i promjena
    #4. prosjecna min max
    #draw3(prosjecna, maxi, minn, 'Dnevne razlike')
    
    return normprosjecna

j = len(Sastav_CrobexTurist)
navg = [None]*j
l=0

for i in (Sastav_CrobexTurist):
    navg[l] = analiza(i)
    draw1(navg[l], i)
    ++l
        

#j = len(Sastav_Crobex10)
#navg = [None]*j
#l=0

#for i in (Sastav_Crobex10):
#    navg[l] = analiza(i)
#    draw1(navg[l], i)
#    ++l

j = len(Sastav_Crobex_Industrija)
navg = [None]*j
l=0

for i in (Sastav_Crobex_Industrija):
    navg[l] = analiza(i)
    draw1(navg[l], i)
    ++l

j = len(Sastav_Crobex_Konstrukt)
navg = [None]*j
l=0

for i in (Sastav_Crobex_Konstrukt):
    navg[l] = analiza(i)
    draw1(navg[l], i)
    ++l