# https://quantpedia.com/strategies/market-seasonality-effect-in-world-equity-indexes/
#
# Be invested in global equity markets during November – April period, stay in cash during May-October period (alternatively go
# long in stocks from countries from northern hemisphere during winter period and long in stocks from countries from southern hemisphere
# during summer period; alternatively go long in cyclical companies during winter period and short defensive stocks and switch positions
# during the summer period)

from AlgorithmImports import *

class SeasonalityInEquitiesAlgorithm(QCAlgorithm):

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

        self.AddEquity("SPY", Resolution.Daily)
        self.AddEquity("SHY", Resolution.Daily)
        
        self.Schedule.On(self.DateRules.MonthStart("SPY"), self.TimeRules.AfterMarketOpen("SPY"), self.Rebalance)
        
    def Rebalance(self):
        if self.Time.month == 5:
            self.Liquidate("SPY")
        if self.Time.month == 11:
            self.SetHoldings("SPY", 1)