# # region imports
# from AlgorithmImports import *
# # endregion

def add_months(current_date, months_to_add):
    year = current_date.year
    month = current_date.month + months_to_add
    if month > 12:
        year += 1
        month = 3
    new_date = datetime(year,month,1)
    while new_date.weekday() >= 5:
        new_date = new_date + timedelta(days=1)
    return new_date

# TICKERS = ['MSFT', 'AAPL', 'NVDA', 'GOOG', 'GOOGL', 'AMZN', 'META', 'BRK.B', 'LLY', 'AVGO', 'TSLA', 'JPM', 'V', 'XOM', 'WMT', 'UNH', 'MA', 'PG', 'JNJ', 'COST', 'ORCL', 'HD', 'MRK', 'BAC', 'CVX', 'ABBV', 'NFLX', 'CRM', 'KO', 'AMD', 'PEP', 'TMO', 'QCOM', 'ADBE', 'WFC', 'LIN', 'DHR', 'TMUS', 'MCD', 'ACN', 'CSCO', 'DIS', 'INTU', 'AMAT', 'TXN', 'ABT', 'GE', 'CAT', 'AXP', 'AMGN', 'VZ', 'MS', 'PFE', 'IBM', 'NOW', 'NEE', 'PM', 'BX', 'CMCSA', 'GS', 'UNP', 'ISRG', 'MU', 'COP', 'SCHW', 'RTX', 'NKE', 'SPGI', 'INTC', 'ETN', 'UBER', 'HON', 'BKNG', 'LOW', 'ELV', 'LRCX', 'SYK', 'UPS', 'T', 'C', 'PGR', 'BLK', 'VRTX', 'BA', 'MDT', 'LMT', 'BSX', 'TJX', 'REGN', 'DE', 'CB', 'ADI', 'MMC', 'ADP', 'KLAC', 'PLD', 'PANW', 'ANET', 'CI', 'MDLZ', 'ABNB', 'AMT', 'FI', 'SBUX', 'SO', 'SNPS', 'CMG', 'BMY', 'HCA', 'GILD', 'WM', 'GD', 'APH', 'DUK', 'MO', 'CDNS', 'ZTS', 'ICE', 'SHW', 'CL', 'FCX', 'TT', 'CME', 'MCO', 'EQIX', 'TDG', 'ITW', 'EOG', 'TGT', 'CVS', 'MCK', 'CTAS', 'PH', 'CEG', 'NXPI', 'NOC', 'SLB', 'MAR', 'BDX', 'ECL', 'PYPL', 'EMR', 'CSX', 'USB', 'PNC', 'AON', 'MPC', 'FDX', 'MSI', 'PSX', 'WELL', 'CARR', 'APD', 'RSG', 'ORLY', 'ROP', 'MMM', 'AJG', 'OXY', 'MNST', 'PCAR', 'MRNA', 'EW', 'COF', 'VLO', 'TFC', 'CPRT', 'SMCI', 'AIG', 'DXCM', 'MCHP', 'MET', 'HLT', 'GM', 'WMB', 'NEM', 'NSC', 'AFL', 'PSA', 'SRE', 'TRV', 'DHI', 'JCI', 'AEP', 'AZO', 'OKE', 'SPG', 'O', 'F', 'HES', 'EL', 'ADSK', 'GWW', 'FTNT', 'DLR', 'TEL', 'URI', 'KDP', 'STZ', 'PAYX', 'D', 'A', 'KMB', 'GEV', 'LEN', 'ALL', 'ROST', 'BK', 'KMI', 'CCI', 'KHC', 'FIS', 'COR', 'AMP', 'HUM', 'IDXX', 'LHX', 'PRU', 'HSY', 'CNC', 'IQV', 'DOW', 'NUE', 'PCG', 'MSCI', 'GIS', 'PWR', 'OTIS', 'CHTR', 'CTVA', 'YUM', 'AME', 'CMI', 'KR', 'LULU', 'RCL', 'EXC', 'IR', 'KVUE', 'ACGL', 'ODFL', 'FAST', 'SYY', 'PEG', 'GEHC', 'MPWR', 'NDAQ', 'VRSK', 'MLM', 'CSGP', 'XYL', 'FANG', 'IT', 'VMC', 'HWM', 'EA', 'LVS', 'CTSH', 'FICO', 'DAL', 'ED', 'HAL', 'DD', 'BIIB', 'BKR', 'GRMN', 'VST', 'LYB', 'MTD', 'RMD', 'HPQ', 'XEL', 'DVN', 'VICI', 'DFS', 'PPG', 'ON', 'DG', 'GLW', 'CDW', 'EXR', 'ROK', 'TSCO', 'ADM', 'HIG', 'WAB', 'EIX', 'EFX', 'ANSS', 'AVB', 'CBRE', 'GPN', 'FTV', 'WEC', 'EBAY', 'TROW', 'TRGP', 'CHD', 'AWK', 'FITB', 'WTW', 'RJF', 'TTWO', 'BRO', 'MTB', 'DOV', 'KEYS', 'EQR', 'DLTR', 'IFF', 'PHM', 'VLTO', 'WST', 'ZBH', 'ETR', 'IRM', 'DTE', 'WDC', 'NVR', 'BR', 'LYV', 'CAH', 'NTAP', 'FE', 'STT', 'HPE', 'DECK', 'STE', 'WY', 'ROL', 'BF.B', 'APTV', 'PPL', 'TER', 'PTC', 'BALL', 'AXON', 'HUBB', 'ARE', 'ES', 'INVH', 'SBAC', 'TSN', 'K', 'STLD', 'GPC', 'TYL', 'WAT', 'CTRA', 'HBAN', 'BLDR', 'CCL', 'LDOS', 'WRB', 'MOH', 'HRL', 'AEE', 'ALGN', 'MKC', 'STX', 'VTR', 'PFG', 'CNP', 'TDY', 'WBD', 'CBOE', 'COO', 'CPAY', 'CMS', 'OMC', 'CINF', 'AVY', 'ULTA', 'RF', 'EQT', 'SYF', 'DPZ', 'ATO', 'DRI', 'BAX', 'UAL', 'HOLX', 'J', 'VRSN', 'NRG', 'LH', 'NTRS', 'FDS', 'ILMN', 'EG', 'CE', 'ESS', 'TXT', 'L', 'LUV', 'CLX', 'CFG', 'EXPD', 'ZBRA', 'JBHT', 'IEX', 'PKG', 'MAA', 'DGX', 'BBY', 'MAS', 'NWSA', 'NWS', 'AES', 'ALB', 'SWKS', 'MRO', 'FOXA', 'FOX', 'CAG', 'BG', 'SNA', 'AMCR', 'IP', 'KEY', 'AKAM', 'WBA', 'CF', 'POOL', 'JBL', 'EXPE', 'TRMB', 'PNR', 'RVTY', 'NDSN', 'DOC', 'CPB', 'WRK', 'SWK', 'LNT', 'NI', 'UDR', 'VTRS', 'TECH', 'HST', 'INCY', 'KIM', 'PODD', 'MGM', 'EVRG', 'LW', 'AOS', 'BEN', 'JKHY', 'DVA', 'SJM', 'IPG', 'LKQ', 'EMN', 'CRL', 'APA', 'KMX', 'CPT', 'JNPR', 'BBWI', 'REG', 'EPAM', 'TAP', 'WYNN', 'ALLE', 'UHS', 'RL', 'SOLV', 'PAYC', 'FFIV', 'TFX', 'HII', 'CTLT', 'MOS', 'BXP', 'TPR', 'CHRW', 'DAY', 'QRVO', 'HSIC', 'GNRC', 'AAL', 'PNW', 'AIZ', 'HAS', 'PARA', 'FRT', 'BWA', 'BIO', 'MKTX', 'FMC', 'MTCH', 'CZR', 'GL', 'MHK', 'NCLH', 'RHI', 'IVZ', 'CMA']

# class AdaptableYellowGreenAlpaca(QCAlgorithm):

#     # def CustomBarHandler(self, bar):
#     #     self.windows[self.tickers.index(bar.symbol)].Add(bar)

#     def change(self, before, after):
#         diff = after-before
#         return diff/before

#     def OnEndOfAlgorithm(self):
#         self.Liquidate()


#     def initialize(self):
#         self.set_start_date(2000, 3, 1)
#         self.set_end_date(2002, 3, 1)
#         self.set_cash(100000)
#         self.tickers = [self.AddEquity(i, Resolution.Daily).Symbol for i in TICKERS]
#         # self.windows = [RollingWindow[TradeBar](64) for i in range(len(TICKERS))]

#         self.period = timedelta(90)
#         self.nextEntryTime = self.Time

#         self.AMOUNT = 3

#     def on_data(self, data: Slice):
#         if self.Time > DateTime(2002,1,1):
#             self.Liquidate()
#         elif self.nextEntryTime <= self.Time:
#             self.Liquidate()
#             self.Log("SELL")
#             # for window in self.windows:
#             #     if not window.IsReady:
#             #         return
#             stocks = {}
#             for ticker in self.tickers:
#                 hist = self.History(ticker, timedelta(90), Resolution.Daily)
#                 half = len(hist.index) // 2
#                 lower = hist.iloc[:half,:]
#                 higher = hist.iloc[half:,:]
#                 try:

#                     lh = max(lower["close"])
#                     hh = max(higher["close"])

#                     diffChange = self.change(lh,hh)
#                     stocks[diffChange] = ticker
#                 except:
#                     pass

#             sortedStocks = dict(sorted(stocks.items()))
#             sortedStocks = dict(reversed(list(sortedStocks.items())))

#             toBuy = list(sortedStocks.values())[:self.AMOUNT]
#             for ticker in toBuy:
#                 self.Log("BUY " + str(ticker.value))
#                 self.SetHoldings(ticker, 1/len(toBuy))
#             # self.MarketOrder(self.spy, int(self.Portfolio.Cash / price) )
#             self.nextEntryTime = add_months(self.Time, 3)

# region imports
from AlgorithmImports import *
# endregion

#TICKERS = ['MMM', 'ABT', 'ANF', 'ADBE', 'AMD', 'AES', 'AET', 'ACS', 'AFL', 'A', 'APD', 'AKAM', 'AKS', 'AA', 'AYE', 'ATI', 'AGN', 'ALL', 'ALTR', 'MO', 'AMZN', 'AEE', 'AEP', 'AXP', 'AIG', 'AMT', 'AMP', 'ABC', 'AMGN', 'APC', 'ADI', 'AON', 'APA', 'AIV', 'APOL', 'AAPL', 'AMAT', 'ADM', 'ASH', 'AIZ', 'T', 'ADSK', 'ADP', 'AN', 'AZO', 'AVB', 'AVY', 'AVP', 'BHI', 'BLL', 'BAC', 'BK', 'BCR', 'BAX', 'BBT', 'BDX', 'BBBY', 'BMS', 'BRK.B', 'BBY', 'BIG', 'BIIB', 'BJS', 'BDK', 'HRB', 'BMC', 'BA', 'BXP', 'BSX', 'BMY', 'BRCM', 'BF.B', 'CHRW', 'CA', 'COG', 'CAM', 'CPB', 'COF', 'CAH', 'CFN', 'CCL', 'CAT', 'CBG', 'CBS', 'CELG', 'CNP', 'CTL', 'CEPH', 'CF', 'SCHW', 'CHK', 'CVX', 'CB', 'CI', 'CINF', 'CTAS', 'CSCO', 'C', 'CTXS', 'CLF', 'CLX', 'CME', 'CMS', 'COH', 'KO', 'CCE', 'CTSH', 'CL', 'CMCSA', 'CMA', 'CSC', 'CPWR', 'CAG', 'COP', 'CNX', 'ED', 'STZ', 'CEG', 'CBE', 'GLW', 'COST', 'CVH', 'CSX', 'CMI', 'CVS', 'DHI', 'DHR', 'DRI', 'DVA', 'DF', 'DE', 'DELL', 'DNR', 'XRAY', 'DVN', 'DV', 'DO', 'DTV', 'DFS', 'D', 'RRD', 'DOV', 'DOW', 'DPS', 'DTE', 'DD', 'DUK', 'DNB', 'ETFC', 'EMN', 'EK', 'ETN', 'EBAY', 'ECL', 'EIX', 'EP', 'ERTS', 'EMC', 'EMR', 'ETR', 'EOG', 'EQT', 'EFX', 'EQR', 'EL', 'EXC', 'EXPE', 'EXPD', 'ESRX', 'XOM', 'FDO', 'FAST', 'FII', 'FDX', 'FIS', 'FITB', 'FHN', 'FSLR', 'FE', 'FISV', 'FLIR', 'FLS', 'FLR', 'FMC', 'FTI', 'F', 'FRX', 'FO', 'FPL', 'BEN', 'FCX', 'FTR', 'GME', 'GCI', 'GPS', 'GD', 'GE', 'GIS', 'GPC', 'GNW', 'GENZ', 'GILD', 'GS', 'GR', 'GT', 'GOOG', 'GWW', 'HAL', 'HOG', 'HAR', 'HRS', 'HIG', 'HAS', 'HCP', 'HCN', 'HNZ', 'HES', 'HPQ', 'HD', 'HON', 'HRL', 'HSP', 'HST', 'HCBK', 'HUM', 'HBAN', 'ITW', 'RX', 'TEG', 'INTC', 'ICE', 'IBM', 'IFF', 'IGT', 'IP', 'IPG', 'INTU', 'ISRG', 'IVZ', 'IRM', 'ITT', 'JBL', 'JEC', 'JNS', 'JDSU', 'JNJ', 'JCI', 'JPM', 'JNPR', 'K', 'KEY', 'KMB', 'KIM', 'KG', 'KLAC', 'KSS', 'KFT', 'KR', 'LLL', 'LH', 'LM', 'LEG', 'LEN', 'LUK', 'LXK', 'LIFE', 'LLY', 'LTD', 'LNC', 'LLTC', 'LMT', 'L', 'LO', 'LOW', 'LSI', 'MTB', 'M', 'MRO', 'MAR', 'MMC', 'MI', 'MAS', 'MEE', 'MA', 'MAT', 'MFE', 'MKC', 'MCD', 'MHP', 'MCK', 'MJN', 'MWV', 'MHS', 'MDT', 'WFR', 'MRK', 'MDP', 'MET', 'PCS', 'MCHP', 'MU', 'MSFT', 'MIL', 'MOLX', 'TAP', 'MON', 'MWW', 'MCO', 'MS', 'MOT', 'MXB', 'MUR', 'MYL', 'NBR', 'NDAQ', 'NOV', 'NSM', 'NTAP', 'NYT', 'NWL', 'NEM', 'NWSA', 'GAS', 'NKE', 'NI', 'NBL', 'JWN', 'NSC', 'NTRS', 'NOC', 'NU', 'NOVL', 'NVLS', 'NUE', 'NVDA', 'NYX', 'ORLY', 'OXY', 'ODP', 'OMC', 'ORCL', 'OI', 'PCAR', 'PTV', 'PLL', 'PH', 'PDCO', 'PAYX', 'JCP', 'PBCT', 'POM', 'PBG', 'PEP', 'PKI', 'PFE', 'PCG', 'PM', 'PNW', 'PXD', 'PBI', 'PCL', 'PNC', 'RL', 'PPG', 'PPL', 'PX', 'PCP', 'PFG', 'PG', 'PGN', 'PGR', 'PLD', 'PRU', 'PEG', 'PSA', 'PHM', 'QLGC', 'PWR', 'QCOM', 'DGX', 'STR', 'Q', 'RSH', 'RRC', 'RTN', 'RHT', 'RF', 'RSG', 'RAI', 'RHI', 'ROK', 'COL', 'ROP', 'ROST', 'RDC', 'R', 'SWY', 'SAI', 'CRM', 'SNDK', 'SLE', 'SCG', 'SLB', 'SNI', 'SEE', 'SHLD', 'SRE', 'SHW', 'SIAL', 'SPG', 'SLM', 'SII', 'SJM', 'SNA', 'SO', 'LUV', 'SWN', 'SE', 'S', 'STJ', 'SWK', 'SPLS', 'SBUX', 'HOT', 'STT', 'SRCL', 'SYK', 'SUN', 'STI', 'SVU', 'SYMC', 'SYY', 'TROW', 'TGT', 'TE', 'TLAB', 'THC', 'TDC', 'TER', 'TSO', 'TXN', 'TXT', 'HSY', 'TRV', 'TMO', 'TIF', 'TWX', 'TWC', 'TIE', 'TJX', 'TMK', 'TSS', 'TSN', 'USB', 'UNP', 'UNH', 'UPS', 'X', 'UTX', 'UNM', 'VFC', 'VLO', 'VAR', 'VTR', 'VRSN', 'VZ', 'VIAb', 'V', 'VNO', 'VMC', 'WMT', 'WAG', 'DIS', 'WPO', 'WM', 'WAT', 'WPI', 'WLP', 'WFC', 'WDC', 'WU', 'WY', 'WHR', 'WFMI', 'WMB', 'WIN', 'WEC', 'WYN', 'WYNN', 'XEL', 'XRX', 'XLNX', 'XL', 'XTO', 'YHOO', 'YUM', 'ZMH', 'ZION']
TICKERS = ['MSFT', 'AAPL', 'NVDA', 'GOOG', 'GOOGL', 'AMZN', 'META', 'BRK.B', 'LLY', 'AVGO', 'TSLA', 'JPM', 'V', 'XOM', 'WMT', 'UNH', 'MA', 'PG', 'JNJ', 'COST', 'ORCL', 'HD', 'MRK', 'BAC', 'CVX', 'ABBV', 'NFLX', 'CRM', 'KO', 'AMD', 'PEP', 'TMO', 'QCOM', 'ADBE', 'WFC', 'LIN', 'DHR', 'TMUS', 'MCD', 'ACN', 'CSCO', 'DIS', 'INTU', 'AMAT', 'TXN', 'ABT', 'GE', 'CAT', 'AXP', 'AMGN', 'VZ', 'MS', 'PFE', 'IBM', 'NOW', 'NEE', 'PM', 'BX', 'CMCSA', 'GS', 'UNP', 'ISRG', 'MU', 'COP', 'SCHW', 'RTX', 'NKE', 'SPGI', 'INTC', 'ETN', 'UBER', 'HON', 'BKNG', 'LOW', 'ELV', 'LRCX', 'SYK', 'UPS', 'T', 'C', 'PGR', 'BLK', 'VRTX', 'BA', 'MDT', 'LMT', 'BSX', 'TJX', 'REGN', 'DE', 'CB', 'ADI', 'MMC', 'ADP', 'KLAC', 'PLD', 'PANW', 'ANET', 'CI', 'MDLZ', 'ABNB', 'AMT', 'FI', 'SBUX', 'SO', 'SNPS', 'CMG', 'BMY', 'HCA', 'GILD', 'WM', 'GD', 'APH', 'DUK', 'MO', 'CDNS', 'ZTS', 'ICE', 'SHW', 'CL', 'FCX', 'TT', 'CME', 'MCO', 'EQIX', 'TDG', 'ITW', 'EOG', 'TGT', 'CVS', 'MCK', 'CTAS', 'PH', 'CEG', 'NXPI', 'NOC', 'SLB', 'MAR', 'BDX', 'ECL', 'PYPL', 'EMR', 'CSX', 'USB', 'PNC', 'AON', 'MPC', 'FDX', 'MSI', 'PSX', 'WELL', 'CARR', 'APD', 'RSG', 'ORLY', 'ROP', 'MMM', 'AJG', 'OXY', 'MNST', 'PCAR', 'MRNA', 'EW', 'COF', 'VLO', 'TFC', 'CPRT', 'SMCI', 'AIG', 'DXCM', 'MCHP', 'MET', 'HLT', 'GM', 'WMB', 'NEM', 'NSC', 'AFL', 'PSA', 'SRE', 'TRV', 'DHI', 'JCI', 'AEP', 'AZO', 'OKE', 'SPG', 'O', 'F', 'HES', 'EL', 'ADSK', 'GWW', 'FTNT', 'DLR', 'TEL', 'URI', 'KDP', 'STZ', 'PAYX', 'D', 'A', 'KMB', 'GEV', 'LEN', 'ALL', 'ROST', 'BK', 'KMI', 'CCI', 'KHC', 'FIS', 'COR', 'AMP', 'HUM', 'IDXX', 'LHX', 'PRU', 'HSY', 'CNC', 'IQV', 'DOW', 'NUE', 'PCG', 'MSCI', 'GIS', 'PWR', 'OTIS', 'CHTR', 'CTVA', 'YUM', 'AME', 'CMI', 'KR', 'LULU', 'RCL', 'EXC', 'IR', 'KVUE', 'ACGL', 'ODFL', 'FAST', 'SYY', 'PEG', 'GEHC', 'MPWR', 'NDAQ', 'VRSK', 'MLM', 'CSGP', 'XYL', 'FANG', 'IT', 'VMC', 'HWM', 'EA', 'LVS', 'CTSH', 'FICO', 'DAL', 'ED', 'HAL', 'DD', 'BIIB', 'BKR', 'GRMN', 'VST', 'LYB', 'MTD', 'RMD', 'HPQ', 'XEL', 'DVN', 'VICI', 'DFS', 'PPG', 'ON', 'DG', 'GLW', 'CDW', 'EXR', 'ROK', 'TSCO', 'ADM', 'HIG', 'WAB', 'EIX', 'EFX', 'ANSS', 'AVB', 'CBRE', 'GPN', 'FTV', 'WEC', 'EBAY', 'TROW', 'TRGP', 'CHD', 'AWK', 'FITB', 'WTW', 'RJF', 'TTWO', 'BRO', 'MTB', 'DOV', 'KEYS', 'EQR', 'DLTR', 'IFF', 'PHM', 'VLTO', 'WST', 'ZBH', 'ETR', 'IRM', 'DTE', 'WDC', 'NVR', 'BR', 'LYV', 'CAH', 'NTAP', 'FE', 'STT', 'HPE', 'DECK', 'STE', 'WY', 'ROL', 'BF.B', 'APTV', 'PPL', 'TER', 'PTC', 'BALL', 'AXON', 'HUBB', 'ARE', 'ES', 'INVH', 'SBAC', 'TSN', 'K', 'STLD', 'GPC', 'TYL', 'WAT', 'CTRA', 'HBAN', 'BLDR', 'CCL', 'LDOS', 'WRB', 'MOH', 'HRL', 'AEE', 'ALGN', 'MKC', 'STX', 'VTR', 'PFG', 'CNP', 'TDY', 'WBD', 'CBOE', 'COO', 'CPAY', 'CMS', 'OMC', 'CINF', 'AVY', 'ULTA', 'RF', 'EQT', 'SYF', 'DPZ', 'ATO', 'DRI', 'BAX', 'UAL', 'HOLX', 'J', 'VRSN', 'NRG', 'LH', 'NTRS', 'FDS', 'ILMN', 'EG', 'CE', 'ESS', 'TXT', 'L', 'LUV', 'CLX', 'CFG', 'EXPD', 'ZBRA', 'JBHT', 'IEX', 'PKG', 'MAA', 'DGX', 'BBY', 'MAS', 'NWSA', 'NWS', 'AES', 'ALB', 'SWKS', 'MRO', 'FOXA', 'FOX', 'CAG', 'BG', 'SNA', 'AMCR', 'IP', 'KEY', 'AKAM', 'WBA', 'CF', 'POOL', 'JBL', 'EXPE', 'TRMB', 'PNR', 'RVTY', 'NDSN', 'DOC', 'CPB', 'WRK', 'SWK', 'LNT', 'NI', 'UDR', 'VTRS', 'TECH', 'HST', 'INCY', 'KIM', 'PODD', 'MGM', 'EVRG', 'LW', 'AOS', 'BEN', 'JKHY', 'DVA', 'SJM', 'IPG', 'LKQ', 'EMN', 'CRL', 'APA', 'KMX', 'CPT', 'JNPR', 'BBWI', 'REG', 'EPAM', 'TAP', 'WYNN', 'ALLE', 'UHS', 'RL', 'SOLV', 'PAYC', 'FFIV', 'TFX', 'HII', 'CTLT', 'MOS', 'BXP', 'TPR', 'CHRW', 'DAY', 'QRVO', 'HSIC', 'GNRC', 'AAL', 'PNW', 'AIZ', 'HAS', 'PARA', 'FRT', 'BWA', 'BIO', 'MKTX', 'FMC', 'MTCH', 'CZR', 'GL', 'MHK', 'NCLH', 'RHI', 'IVZ', 'CMA']
class MyAlgorithm(QCAlgorithm):

    def change(self, before, after):
        diff = after-before
        return diff/before

    def OnEndOfAlgorithm(self):
        self.Liquidate()

    def Initialize(self):
        self.END_YEAR = 2020
        self.set_start_date(2000, 1, 1)
        self.set_end_date(self.END_YEAR, 1, 10)
        self.set_cash(100000)

        self.symbolData = dict()

        # self.AddEquity("SPY", Resolution.DAILY)

        for ticker in TICKERS:
            symbol = self.AddEquity(ticker, Resolution.DAILY,leverage=1).Symbol

            consolidator_daily = TradeBarConsolidator(timedelta(1))
            consolidator_daily.DataConsolidated += self.OnDailyData
            self.SubscriptionManager.AddConsolidator(symbol, consolidator_daily)

            self.symbolData[symbol] = SymbolData()

        # self.schedule.on(self.date_rules.every_day("SPY"),
        #                 self.time_rules.before_market_close("SPY", 30),
        #                 self.before_close_market)

        self.periodMonths = 3
        self.set_warm_up(timedelta(365))

        self.nextEntryTime = self.Time
        self.nextSellTime = self.Time
        self.DiversificationAmount = 20

	# Add daily bar to daily rolling window
    def OnDailyData(self, sender, bar):
        self.symbolData[bar.Symbol].daily_rw.Add(bar.Close)

    def on_data(self, data: Slice):
        if self.is_warming_up: return
        if self.Time.weekday() >= 5: return
        if self.Time > DateTime(self.END_YEAR,1,1):
            self.Liquidate()
        elif self.nextSellTime <= self.Time:
            self.Liquidate()
            self.Log("SELL")
            self.nextSellTime = add_months(self.Time, self.periodMonths)
            self.nextEntryTime = self.Time + timedelta(2)
        elif self.nextEntryTime <= self.Time:
            # for window in self.windows:
            #     if not window.IsReady:
            #         return
            stocks = {}
            for ticker in self.symbolData:
                window = self.symbolData[ticker].daily_rw
                if window.is_ready:
                    half = window.count // 2
                    closes = list(window)
                    before = closes[half:]
                    after = closes[:half]
                    try:

                        bh = max(before)
                        ah = max(after)

                        diffChange = self.change(bh,ah)
                        stocks[diffChange] = ticker
                    except:
                        pass

            sortedStocks = dict(sorted(stocks.items()))
            sortedStocks = dict(reversed(list(sortedStocks.items())))

            toBuy = list(sortedStocks.values())[:self.DiversificationAmount]
            invStocks = {v: k for k, v in stocks.items()}
            for ticker in toBuy:
                self.Log("BUY " + str(ticker.value) + ": " + str(round(invStocks[ticker], 3)))
                self.SetHoldings(ticker, 1/len(toBuy) * 0.9)
            # self.MarketOrder(self.spy, int(self.Portfolio.Cash / price) )
            self.nextEntryTime = add_months(self.Time, self.periodMonths+1)

    def OnData(self, data):
        pass


class SymbolData(object):
	
    def __init__(self):
        self.daily_rw = RollingWindow[float](66)