class UncoupledTransdimensionalPrism(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2012, 1, 1)  # Set Start Date
        self.SetEndDate(2022, 5, 1)
        self.SetCash(1000000)  # Set Strategy Cash
        spy = self.AddEquity("SPY", Resolution.Minute)
        spy.SetDataNormalizationMode(DataNormalizationMode.Raw)
        self.tlt = self.AddEquity('TLT', Resolution.Minute).Symbol
        self.spy = spy.Symbol
        self.contract = None
        self.alpha =  0.01/12 # the weight of options position
        self.strike = 0.725 # the ratio of strike to market

    def OnData(self, data): 
        
        if (self.Time.hour,self.Time.minute) != (15, 30): return 
        
        if not self.Portfolio[self.spy].Invested:
            self.SetHoldings(self.spy, 1.00*(1-self.alpha))
            self.SetHoldings(self.tlt, 0.00*(1-self.alpha))

        # DO HEDGE
        if self.contract is None:
            self.contract = self.GetContract()
            return
        
        if (self.contract.ID.Date - self.Time).days < 1:
            # self.Liquidate(self.contract)
            self.RemoveSecurity(self.contract)
            self.contract = None
            return
        
        # ES Added: Ensure bar data contains the contract
        if not data.ContainsKey(self.contract): return
        
               
        if not self.Portfolio[self.contract].Invested:
            self.SetHoldings(self.contract, self.alpha)
            self.Debug(self.contract)
        
        if self.Securities[self.spy].Price < self.contract.ID.StrikePrice * 1.0:
            self.Liquidate(self.contract)
            self.RemoveSecurity(self.contract)
    
    def GetContract(self):
        targetStrike = self.Securities[self.spy].Price * self.strike
        targetStrike = (targetStrike) - (targetStrike)%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 30 < (x.ID.Date - self.Time).days <= 45]
        if len(puts) == 0:
            return None
        self.AddOptionContract(puts[0], Resolution.Minute)
        return puts[0]