# -*- coding: utf-8 -*-
"""
Created on Mon Feb 11 08:11:25 2019

@author: cpace
"""
import numpy as np
from Modules.dataseries import dataseries
from Modules.indicators import bolinger
from Modules.indicators import rsi
from Modules.indicators import macd
from Modules.contractmonths import contractmonths as months
from Modules.indicators import adx
from Modules.indicators import stochastics
from Modules.pnp import pnp
import datetime

#crop = 'corn'
#month = 'jul'
#year = 2013

finalout = [] # [$win,$loss, ctwin, ctloss, totcount]
finalout1 = []
days = 4

#a = dataseries(crop, month, year)

final = []
total = []

def endofrise(row, col, thresh=70):
    if npdf[row][col] <= npdf[row-1][col] and npdf[row][col] >= thresh:
        return True
    elif npdf[row][col] > 85 and npdf[row-1][col] < thresh:
        return True
    else:
        return False
    
def endoffall(row, col, thresh=30):
    if npdf[row][col] >= npdf[row-1][col] and npdf[row][col] <= thresh:
        return True
    else:
        return False
    
def istrending(trending, row, col, thresh=30, endthresh=18):  # looks like 18 for stochastics
    if npdf[row][col] >= thresh:
        return True
    elif trending and npdf[row][col] > endthresh:
        return True
    else:
        return False
    
def beginrise(row, col, thresh=30):
    if npdf[row][col] > thresh and npdf[row-1][col] < thresh and npdf[row-1][col] != 0 and npdf[row][5] != 0:
        return True
    else:
        return False
    
def beginfall(row, col, thresh=70):
    if npdf[row][col] < thresh and npdf[row-1][col] > thresh and npdf[row-1][col] != 0 and npdf[row][5] != 0:
        return True
    else:
        return False
    
def belowpnplow(row):
    month = npdf[row][0].strftime('%m')
    year = int(npdf[row][0].strftime('%Y'))
    frontmonth = {'01':'Mar', '02':'Mar', '03':'May', '04':'May', '05':'Jul', '06':'Jul', '07':'Sep', '08':'Sep', '09':'Dec', '10':'Dec', '11':'Dec', '12':'Mar'}
    if month == '12': year += 1
    b = pnp('Wheat', frontmonth[month], year)
    bestlow = b.minBestFit
    a1 = dataseries('Wheat', frontmonth[month], year)
    df1 = a1.dataseries
    data = df1.loc[df1['Date']==npdf[row][0]].Close
    if float(data.item()) < bestlow: return True
    
upper = 70
lower = 30
trend = 30   
i = 0   

for crop in ['Soybeans']:
    for year in range(2000,2020):
        contractmonths = months(crop)
        monthlist = contractmonths.months
        for month in monthlist:
            
            yeartotal = 0
            trades = 0
            a = dataseries(crop, month, year)
            df = a.dataseries
            df = adx(df,14).adx
            df = stochastics(df).stochastics
            df = macd(df).difference
               
            npdf = np.asarray(df)
            
            i = 0 
            look = True
            long = False
            short = False
            trending = False
            position = ''
            indcol = 7
            
#   Wheat -- If not trending and yesterday's MACD is rising thu 0 and Stoch today rising thru 30 - 91% $2.26
#   Soybeans -- If not trending and today's MACD is rising thru 0 and yesterday's Stoch rising thru 30 - 71% $2.98
            
            while i < len(npdf)-2:
                while look and i < npdf.shape[0]:
                    trending = istrending(trending, i, 5)
                    if beginrise(i-1,indcol,30) and beginrise(i,9,0) and not trending and npdf[i,indcol] < 70:
                        positionentry = npdf[i,4]
                        entrydate = npdf[i,0]
                        look = False
                        short = False
                        long = True
                        break
#                    elif trending and not short and npdf[i,indcol]<30: 
#                        if not belowpnplow(i):
#                            positionentry = npdf[i,4]
#                            entrydate = npdf[i,0]
#                            look = False
#                            short = True
#                            long = False
#                            break
                    else:
                        i += 1
                    i+=1
                                   
                while long and i < len(npdf):
                    trending = istrending(trending, i, 5)
                    if endofrise(i,indcol,70):
                        positionexit = npdf[i,4]
                        exitdate = npdf[i,0]
                        look = True
                        short = False
#                        long = False
                        break
                    i += 1
                i+=1
                
                if i > npdf.shape[0]-3:
                    positionexit = npdf[i-3,4]
                    exitdate = npdf[i-3,0]
 
                if long:
                    profit = positionexit - positionentry
                    long = False
                    position = 'Long'
                elif short:
                    profit = positionentry - positionexit
                    short = False
                    position = 'Short'
                else:
                    position = ''
                
                if position != '':
                    finalout.append([position, entrydate, exitdate, profit, positionentry, positionexit, month, year])
                    look = True
                    long = False
                    short = False
                    position = ''
                    exitdate = ''
                    positionentry = 0
                    positionexit = 0
                    entrydate = ''
            
            final = [0,0,0]        
            for i in range(len(finalout)):
                if  finalout[i][0] != 'complete':
                    if finalout[i][3] > 0:
                        win = 1
                        loss = 0
                    else:
                        win = 0
                        loss = 1
                    final = [final[0]+finalout[i][3], final[1] + win, final[2] + loss]
                    final.append(final[1]/(final[1]+final[2])*100)
                    
                    
    

a.closeconn()