from datetime import timedelta
from AlgorithmImports import *



class FocusedSkyBlueGalago(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2020, 1, 2)
        #self.SetEndDate(2016, 2, 1)
        #self.SetEndDate(2009,5,1)

        self.InitCash = 10000
        self.SetCash(self.InitCash)
        # self.AddEquity("SPY", Resolution.Minute)
        self.SetWarmUp(5)

        
        # SHILE: I think I forgot the .Symbol?
        self.bnd = self.AddEquity('BND', Resolution.Minute).Symbol
        
        # Options Parameters ===================================
        spy = self.AddEquity("SPY", Resolution.Minute)
        spy.SetDataNormalizationMode(DataNormalizationMode.Raw)
        qqq = self.AddEquity("QQQ", Resolution.Minute)
        qqq.SetDataNormalizationMode(DataNormalizationMode.Raw)
        tqqq = self.AddEquity("TQQQ", Resolution.Minute)
        tqqq.SetDataNormalizationMode(DataNormalizationMode.Raw)
        self.spy = spy.Symbol
        self.qqq = qqq.Symbol
        self.tqqq = tqqq.Symbol
        self.spycontract = None
        self.tqqqcontract = None
        
        # Rebalance beginning of every month =======================
        self.Schedule.On(self.DateRules.MonthStart("SPY"), self.TimeRules.AfterMarketOpen("SPY", 1), self.monthlyRebalance)
        self.Schedule.On(self.DateRules.EveryDay("SPY"), self.TimeRules.AfterMarketOpen("SPY", 1), self.captureSpy)
        
        #Variables used in stoploss=================================
        self.stoplosshold = 0
        self.dailythresh = 0
        
        # Setting Brokerage Model to allow for live deployment
        self.SetBrokerageModel(BrokerageName.AlphaStreams)

        self.weights = {
            self.spy: .3,
            self.qqq: .3,
            self.bnd: .3
        }
        
        #self.SetPortfolioConstruction(InsightWeightingPortfolioConstructionModel(lambda time: None))

    # not necessary, but it saves a lot of typing
    def InsightHelper(self, symbol, percentage):
        if abs(percentage) < 0.001:
            return Insight.Price(symbol, timedelta(1), InsightDirection.Flat)
        elif percentage > 0:
            return Insight.Price(symbol, timedelta(1), InsightDirection.Up, None, None, None, percentage)
        else:
            return Insight.Price(symbol, timedelta(1), InsightDirection.Down, None, None, None, abs(percentage))
        
    def OnData(self, data):
        ''' OnData event is the primary entry point for your algorithm. Each new data point will be pumped in here.
            Arguments:
                data: Slice object keyed by symbol containing the stock data
        '''
        
        if self.IsWarmingUp:
            return
        
        # Begin Stoploss Logic
        self.stoploss(data)
        
        insights = []
        
        if self.stoplosshold == 1:
            for equity in self.weights:
                insights.append(self.InsightHelper(equity, 0))
        else:
            if not self.Portfolio.Invested:
                self.monthlyRebalance()
        #if insights:
        #   self.EmitInsights(insights)
        # End Stoploss Logic
        
        '''
        Begin Hedging On Data
        1. Purchase Hedges
        2. Liquidate and Purchase New Hedges for Time value
        3. Invest 
        '''
        # 1. Purchase Hedges 
        if self.spycontract is None:
            self.spycontract = self.GetSpy()
            return
        if self.tqqqcontract is None:
            self.tqqqcontract = self.GetTqqq()
            return
        
        #Liquidate and Purchase New Hedges for Time value
        if (self.spycontract.ID.Date - self.Time).days < 180:
            self.Liquidate(self.spycontract)
            self.RemoveSecurity(self.spycontract)
            self.spycontract = None
            return
        
        if (self.tqqqcontract.ID.Date - self.Time).days < 180:
            self.Liquidate(self.tqqqcontract)
            self.RemoveSecurity(self.tqqqcontract)
            self.tqqqcontract = None
            return
        
        
        #Purchase LongPut Contracts
        if not self.Portfolio[self.spycontract].Invested:
            self.SetHoldings(self.spycontract, 0.06)
            
        if not self.Portfolio[self.tqqqcontract].Invested:    
                self.SetHoldings(self.tqqqcontract,  0.04)
        

        #Exercise Contracts when they increase a certain % in intrinsic value
        if  self.Securities[self.spy].Price < self.spycontract.ID.StrikePrice * 1.2:
            self.Liquidate(self.spycontract)
            self.RemoveSecurity(self.spycontract)
        
        if  self.Securities[self.tqqq].Price < self.tqqqcontract.ID.StrikePrice * 1.2:    
            self.Liquidate(self.tqqqcontract)
            self.RemoveSecurity(self.tqqqcontract)
        #End hedging Logic




    def GetSpy(self):
        # Target strike as 40% OTM long put 
        targetStrike = (self.Securities[self.spy].Price * 0.60) - (self.Securities[self.spy].Price * 0.60)%5
        contracts = self.OptionChainProvider.GetOptionContractList(self.spy, self.Time)
        puts = [x for x in contracts if x.ID.OptionRight == OptionRight.Put]
        puts = sorted( sorted(puts, key = lambda x: x.ID.Date, reverse = True),
                       key = lambda x: x.ID.StrikePrice)
        puts = [x for x in puts if x.ID.StrikePrice == targetStrike]
        puts = [x for x in puts if 270 < (x.ID.Date - self.Time).days <= 420]
        if len(puts) == 0:
            self.Log("No SPY Puts")
            return None
        self.AddOptionContract(puts[0], Resolution.Minute)
        return puts[0]


    def GetTqqq(self):
        # Target strike as 40% OTM long put 
        targetStrike = (self.Securities[self.tqqq].Price * 0.60) - (self.Securities[self.tqqq].Price * 0.60)%5
        contracts = self.OptionChainProvider.GetOptionContractList(self.tqqq, self.Time)
        puts = [x for x in contracts if x.ID.OptionRight == OptionRight.Put]
        puts = sorted( sorted(puts, key = lambda x: x.ID.Date, reverse = True),
                       key = lambda x: x.ID.StrikePrice)
        # SHILE ADDITION
        # below will get exact strike price if possible, else get closest
        puts = sorted(puts, key=lambda x: abs(x.ID.StrikePrice - targetStrike))
        puts = [x for x in puts if 270 < (x.ID.Date - self.Time).days <= 420]
        if len(puts) == 0:
            puts = sorted(puts, key = lambda x: x.ID.Date, reverse=True)
        if len(puts) == 0:
            return None
        self.AddOptionContract(puts[0], Resolution.Minute)
        return puts[0]





    def captureSpy(self):
        #Grabs the daily opening price of spy for our stoploss method
        if self.CurrentSlice.Bars.ContainsKey(self.spy):
            self.dailythresh = self.CurrentSlice[self.spy].Open
            self.stoplosshold = 0
            return
          
            
    def monthlyRebalance(self):
        # Rebalance portfolio monthly basis
        if self.IsWarmingUp:
            return

        insights = []
        for symbol,weight in self.weights.items():
            insights.append(self.InsightHelper(symbol, weight))
            self.SetHoldings(symbol, weight)
        if insights:
            self.EmitInsights(insights)
         

        
    
    def stoploss(self, data):
        '''
        Stoploss logic:
            1. If spy drops more than 5% liquidate entire equity portfolio
            2. Change stoplosshold value to 1, this indicates that the portfolios SL has been hit 
            and were going to hold until the next trading day
        '''
        if self.IsWarmingUp:
            return
        if self.CurrentSlice.Bars.ContainsKey(self.spy):
            #self.Debug((self.dailythresh - self.CurrentSlice[self.spy].Close)/self.CurrentSlice[self.spy].Close)
            if ((self.dailythresh - self.CurrentSlice[self.spy].Open)/self.dailythresh) < -.05:
                self.SetHoldings(self.spy, 0)
                self.SetHoldings(self.qqq, 0)
                self.stoplosshold = 1
                #self.Log('HIT')