import math

class ModulatedUncoupledGearbox(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2018, 11, 29)  # Set Start Date
        self.SetCash(100000)  # Set Strategy Cash
        # self.AddEquity("SPY", Resolution.Minute)

        self.__numberOfSymbols = 700
        self.__numberOfSymbolsFine = 5
        self.SetUniverseSelection(FineFundamentalUniverseSelectionModel(self.CoarseSelectionFunction, self.FineSelectionFunction, None, None))


    def OnData(self, data):
        '''OnData event is the primary entry point for your algorithm. Each new data point will be pumped in here.
            Arguments:
                data: Slice object keyed by symbol containing the stock data
        '''
        # if not self.Portfolio.Invested:
        #    self.SetHoldings("SPY", 1)
        self.Log("got here")
        for symbol, security in data:
            self.Log(str(symbol) + ": " + str(security))
            self.Quit()
            

    # sort the data by daily dollar volume and take the top 'NumberOfSymbols'
    def CoarseSelectionFunction(self, coarse):
        # sort descending by daily dollar volume
        sortedByDollarVolume = sorted(coarse, key=lambda x: x.DollarVolume, reverse=True)
    
        # return the symbol objects of the top entries from our sorted collection
        return [ x.Symbol for x in sortedByDollarVolume[:self.__numberOfSymbols] ]
    
    # get stocks with earnings reports due tomorrow
    def FineSelectionFunction(self, fine):
        earningsTomorrow = [x for x in \
            filter(lambda x: (self.Time \
            - x.EarningReports.PeriodEndingDate).days == 1, fine)]
    
        return [ x.Symbol for x in earningsTomorrow[:self.__numberOfSymbolsFine] ]