# -*- coding: utf-8 -*-
"""
Created on Sun Feb 24 21:08:01 2019

@author: cpace
"""

from Modules.dataseries import dataseries
from Modules.indicators import stochastics
from Modules.indicators import macd
from Modules.indicators import adx
from Modules.indicators import rsi
import json
import sys
import numpy as np

crop = 'Corn'
month = 'May'
year = 2018
indicator = 'adx'

#crop = str(sys.argv[1]) 
#month = str(sys.argv[2]) 
#year = int(sys.argv[3])
#indicator = str(sys.argv[4]) 

df = dataseries(crop, month, year).dataseries

if indicator == 'stochastics':
    ind = stochastics(df).stochastics
elif indicator == 'adx':
    ind = adx(df,14).adx_all
elif indicator == 'macd':
    ind = macd(df).difference
elif indicator == 'rsi':
    ind = rsi(df, period = 4).rsi

#out = ind.astype(str).to_json(orient='records')[1:-1].replace('},{', '} {').replace('"', '"')   
#data1=json.dumps(out)

npdf = np.asarray(ind)
out = []
    
for i in range(0,npdf.shape[1]):
    col = []
    for j in range(0,len(npdf)):
        if i == 0:
            col.append(str(npdf[j,i]))
        else:
            col.append(npdf[j,i])
    out.append(col)
    

print(json.dumps(out))