from QuantConnect.Data.Market import *
from QuantConnect.Data.Consolidators import *
from datetime import *
from System import *
from QuantConnect import *
from QuantConnect.Algorithm import *
from QuantConnect.Indicators import *
from QuantConnect.Securities.Option import OptionPriceModels
import decimal as d 
import operator

class BasicTemplateOptionsAlgorithm(QCAlgorithm):
                        
    def Initialize(self):   
        self.SetStartDate(2019, 1, 4)
        self.SetEndDate(2019, 1, 4)
        self.tickers_one = ["AAPL", "FB", "TSLA", "AMD"]

        self.symbols = []
        for ticker in self.tickers_one:
            symbol = self.AddEquity(ticker, Resolution.Minute).Symbol
            option = self.AddOption(ticker, Resolution.Minute)
            self.symbols.append(option.Symbol)
            option.SetFilter(-5, 5, timedelta(0), timedelta(14))

            
        self.AddEquity("SPY")
        self.Schedule.On(self.DateRules.EveryDay("SPY"),
                        self.TimeRules.At(12,40),
                        self.OnceADay1)
        
        self.Schedule.On(self.DateRules.EveryDay("SPY"),
                        self.TimeRules.At(12,45),
                        self.OnceADay2)
                        
        self.Schedule.On(self.DateRules.EveryDay("SPY"),
                        self.TimeRules.At(12,50),
                        self.OnceADay3)
                        
        
        self.Log("initialize")
    
        
    def OnceADay1(self):
            self.Log("before")
            for security in self.ActiveSecurities.Keys:
                self.Log(f'Security in Universe: {security.Value}')
                
    def OnceADay2(self):
            for symbol in self.symbols:
                self.RemoveSecurity(symbol)
                
    def OnceADay3(self):
            self.Log("after")
            for security in self.ActiveSecurities.Keys:
                self.Log(f'Security in Universe: {security.Value}')