# https://quantpedia.com/strategies/january-effect-in-stocks/
#
# Invest in small-cap stocks at the beginning of each January. Stay invested in large-cap stocks for the rest of the year.
#
# QC implementation changes:
#   - Small cap etf (IWM) and large cap etf (SPY) are used as investment assets.

from AlgorithmImports import *

class JanuaryEffectInStocks(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2000, 1, 1)  
        self.SetCash(100000) 

        data = self.AddEquity("SPY", Resolution.Daily)
        data.SetLeverage(10)
        self.large_cap = data.Symbol
        
        data = self.AddEquity("IWM", Resolution.Daily)
        data.SetLeverage(10)
        self.small_cap = data.Symbol

        self.start_price = None
        self.recent_month = -1
        
    def OnData(self, data):
        if self.recent_month == self.Time.month:
            return
        self.recent_month = self.Time.month

        if self.Securities[self.large_cap].GetLastData() and self.Securities[self.small_cap].GetLastData():
            if (self.Time.date() - self.Securities[self.large_cap].GetLastData().Time.date()).days < 5 and (self.Time.date() - self.Securities[self.small_cap].GetLastData().Time.date()).days < 5:
                if self.Time.month == 1:
                    if self.Portfolio[self.large_cap].Invested:
                        self.Liquidate(self.large_cap)
                    self.SetHoldings(self.small_cap, 1)
                else:
                    if self.Portfolio[self.small_cap].Invested:
                        self.Liquidate(self.small_cap)
                    self.SetHoldings(self.large_cap, 1)
            else:
                self.Liquidate()
        else:
            self.Liquidate()