from datetime import timedelta
class OptionBot1(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2020, 6, 1)
        self.SetEndDate(2020,9,1)
        self.SetCash(10000)
        self.spy = self.AddEquity("SPY", Resolution.Minute)
        self.option = self.AddOption("SPY",Resolution.Minute)
        self.symbol = self.option.Symbol
        self.option.SetFilter(-2,-1,timedelta(3),timedelta(7))
        self.SetBenchmark(self.spy.Symbol)
        self.options_sold = 0
        
    def OnData(self, data):
        for kvp in data.OptionChains:
            if kvp.Key != self.symbol:
                continue
            optionchain = kvp.Value
            put = [x for x in optionchain if x.Right == OptionRight.Put]
            price = optionchain.Underlying.Price
            contracts = [x for x in put if price - x.Strike > 0]
            contracts = sorted(contracts, key = lambda x: x.Expiry, reverse = True)
            if len(contracts) == 0: 
                continue
            symbol = contracts[0].Symbol
            self.Sell(symbol, 1)
            self.Plot('Custom', 'Options Profit', sum([x.UnrealizedProfit for x in self.Portfolio.Values if x.Type == SecurityType.Option]))
            self.Plot('Custom1', 'Assets Held', len([x for x in self.Portfolio.Values if x.Invested]))
            self.Plot('Custom2', 'Non-options Assets Held', len([x for x in self.Portfolio.Values if x.Invested and x.Type != SecurityType.Option]))  
            self.Plot('Custom3', 'Options Sold', self.options_sold)  
            self.Plot('Custom4', 'Option Assets Held', len([x for x in self.Portfolio.Values if x.Invested and x.Type == SecurityType.Option]))
                
    def OnOrderEvent(self, orderEvent):
        if orderEvent.Symbol.SecurityType == SecurityType.Option:
            self.options_sold += -orderEvent.FillQuantity
            self.Log(f'Options Sold: {self.options_sold}')