from clr import AddReference
AddReference("System")
AddReference("QuantConnect.Algorithm")
AddReference("QuantConnect.Indicators")
AddReference("QuantConnect.Common")

from System import *
from QuantConnect import *
from QuantConnect.Algorithm import *
from QuantConnect.Indicators import *
from datetime import datetime

class MACDTrendAlgorithm(QCAlgorithm):

    def Initialize(self):
        '''Initialise the data and resolution required, as well as the cash and start-end dates for your algorithm. All algorithms must initialized.'''

        self.SetStartDate(2007, 6, 1)    #Set Start Date
        self.SetEndDate(DateTime.Now)      #Set End Date
        self.SetCash(50000)             #Set Strategy Cash
        self.AddEquity("SPY", Resolution.Hour)

        # define our daily macd(12,26) with a 9 day signal
        self.macd = self.MACD("SPY", 12, 26, 9, MovingAverageType.Exponential, Resolution.Hour)

        #define our daily RSI with a 14 day period
        self.rsi = self.RSI("SPY", 14, MovingAverageType.Simple, Resolution.Hour)
        


    def OnData(self, data):
        '''OnData event is the primary entry point for your algorithm. Each new data point will be pumped in here.'''
        # wait for our macd and RSI to fully initialize
        if not self.macd.IsReady and self.rsi.IsReady: return

        # define a small tolerance on our checks to avoid bouncing for MACD
        tolerance = 0.0025
        holdings = self.Portfolio["SPY"].Quantity
        signalDeltaPercent = (self.macd.Current.Value - self.macd.Signal.Current.Value)/self.macd.Fast.Current.Value

        # if our macd is greater than our signal, then let's go long
        if (holdings <= 0 and 
            signalDeltaPercent > tolerance and 
            self.rsi.Current.Value < 30):
                self.SetHoldings("SPY", 1.0)

        # of our macd is less than our signal, then let's go short
        elif (holdings >= 0 and 
            signalDeltaPercent < -tolerance and 
            self.rsi.Current.Value > 70):
                self.Liquidate("SPY")