from QuantConnect.Data.Market import TradeBar
from datetime import timedelta
from System import *
from QuantConnect import *
from QuantConnect.Algorithm import *
from QuantConnect.Indicators import *
import decimal as d 

class MyAlgorithm(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(2016,05,1) #Set Start Date
        self.SetEndDate(2016,06,5) #Set End Date
        self.SetCash(100000) #Set Strategy Cash
        # Find more symbols here: http://quantconnect.com/data
        self.AddEquity("SPY", Resolution.Second)

        consolidator = TradeBarConsolidator(timedelta(1))
        consolidator.DataConsolidated += self.OnDailyData
        self.SubscriptionManager.AddConsolidator("SPY", consolidator)
        
        consolidatorm = TradeBarConsolidator(60)
        consolidatorm.DataConsolidated += self.OnMinuteData
        self.SubscriptionManager.AddConsolidator("SPY", consolidatorm)

        self.daily = RollingWindow[TradeBar](2)
        self.minute = RollingWindow[TradeBar](2)
        self.window = RollingWindow[TradeBar](2)
        
        self.Schedule.On(self.DateRules.EveryDay(),  
                 self.TimeRules.At(9, 31),  
                 Action(self.One))
        
     
        

    # Add daily bar to daily rolling window
    def OnDailyData(self, sender, bar):
        self.daily.Add(bar)
        
    def OnMinuteData(self, sender, bar):
    	self.minute.Add(bar)
      
    def One(self):
    	if not (self.window.IsReady and self.daily.IsReady and self.minute.IsReady): return
        currBar = self.window[0].Close
        yesterdayc = self.daily[1].Close
        minuteBarC = self.minute[1].Close
        minuteBar0 = self.minute[1].Open
        self.Debug(str(self.Securities["SPY"].Price) + "," + str(currBar)+str(minuteBar0)+str(minuteBarC))
        if not self.Portfolio.Invested and currBar<yesterdayc and minuteBar0<minuteBarC:
            self.SetHoldings("SPY", 1)
        


    # Add second bar to window rolling window
    def OnData(self, data):
    	self.Debug("haha")
        self.window.Add(data["SPY"])
        factor = d.Decimal(1.01)
        if not (self.window.IsReady and self.daily.IsReady): return
        currBar = self.window[0].Close
        if self.Portfolio["SPY"].AveragePrice *factor< currBar:
		    self.SetHoldings("SPY", 0)