from clr import AddReference
AddReference("System")
AddReference("QuantConnect.Algorithm")
AddReference("QuantConnect.Common")

from System import *
from QuantConnect import *
from QuantConnect.Algorithm import *
from datetime import timedelta
from decimal import Decimal

import QuantConnect.Securities.Option
import QuantConnect.Securities.Equity


class OptionIssues(QCAlgorithm):
    def Initialize(self):
        self.SetStartDate(2015, 12, 24)
        self.SetEndDate(2016, 1, 31)
        self.SetCash(10000)

        self.option = self.AddOption("GOOG")
        self.equity = self.AddEquity("GOOG", Resolution.Minute)
        
        # set our strike/expiry filter for this option chain
        self.min_days_to_expire = 7
        self.option.SetFilter(-10000000, +0, timedelta(self.min_days_to_expire), timedelta(40))

        # use the underlying equity as the benchmark
        self.SetBenchmark("GOOG")

        self.Schedule.On(
            self.DateRules.EveryDay(self.option.Symbol.Value),
            self.TimeRules.BeforeMarketClose(
                self.option.Symbol.Value,
                30),
            Action(self.action)
        )

    def action(self):
        contracts = list(filter(
            lambda x: isinstance(x, QuantConnect.Securities.Option.Option),
            self.Securities.Values)
        )
        puts = list(filter(
            lambda opt: opt.Right == OptionRight.Put,
            contracts
        ))

        expiry = sorted(
            puts,
            key=lambda opt: opt.Expiry,
        )
        
        self.Debug(
            'Current date: %s. First close contract: %s.' 
            'Days to expiry: %d. Min days constrain: %d' %
            (
                self.Time, 
                expiry[0].Expiry,
                (expiry[0].Expiry - self.Time).days,
                self.min_days_to_expire,
            )
        )
        return