from QuantConnect.Algorithm import *
from QuantConnect.Securities.Option import OptionPriceModels

class BasicTemplateOptionsFilterUniverseAlgorithm(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2019, 1, 1)
        self.SetEndDate(2019, 1, 16)
        self.SetCash(100000)
        
        self.option = self.AddOption("AAPL")
        self.option.SetFilter(-1, +1, 14, 60) 
        
        self.option.PriceModel = OptionPriceModels.BinomialTian()
        self.SetWarmUp(30, Resolution.Daily)
        self.done = False
        
        
    def OnData(self,slice):

        if self.IsWarmingUp:
            return

        for chain in slice.OptionChains:
            for symbol, chain in slice.OptionChains.items():
                if not self.done:
                    contracts = [c for c in chain if c.Right == OptionRight.Call]
                    sorted_contracts = sorted(contracts, key = lambda x: x.Expiry, reverse=True)
                    self.contract = sorted_contracts[0]
                    self.MarketOrder(self.contract.Symbol, 1)
                    self.done = True
                else:
                    contract = [c for c in chain if str(c) == str(self.contract)]
                    if len(contract) == 0:
                        return
                    contract = contract[0]
                    self.Plot("Price", "Theoretical", contract.TheoreticalPrice)
                    self.Plot("Volatility", "Value", self.Securities[symbol.Underlying].VolatilityModel.Volatility)
                    self.Plot("Delta", "Value", contract.Greeks.Delta)