# 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.set_start_date(1999, 1, 1)  
        self.set_cash(100000) 

        self.symbol: Symbol = self.add_equity("SPY", Resolution.DAILY).symbol
        
        self.schedule.on(self.date_rules.month_start(self.symbol), self.time_rules.after_market_open(self.symbol), self.rebalance)
        
    def rebalance(self) -> None:
        if self.time.month == 5:
            self.liquidate(self.symbol)
        
        if self.time.month == 11:
            self.set_holdings(self.symbol, 1)