# https://quantpedia.com/strategies/january-barometer/
#
# Invest in the equity market in each January. Stay invested in equity markets (via ETF, fund, or futures) only if January return is positive; otherwise, switch investments to T-Bills.

from AlgorithmImports import *


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

        data = self.AddEquity("SPY", Resolution.Daily)
        data.SetLeverage(10)
        self.market = data.Symbol

        data = self.AddEquity("BIL", Resolution.Daily)
        data.SetLeverage(10)
        self.t_bills = 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.market].GetLastData()
            and self.Securities[self.t_bills].GetLastData()
        ):
            if (
                self.Time.date()
                - self.Securities[self.market].GetLastData().Time.date()
            ).days < 5 and (
                self.Time.date()
                - self.Securities[self.t_bills].GetLastData().Time.date()
            ).days < 5:
                if self.Time.month == 1:
                    self.Liquidate(self.t_bills)
                    self.SetHoldings(self.market, 1)

                    self.start_price = self.Securities[self.market].Price

                if self.Time.month == 2 and self.start_price:
                    returns = (
                        self.Securities[self.market].Price - self.start_price
                    ) / self.start_price
                    if returns > 0:
                        self.SetHoldings(self.market, 1)
                    else:
                        self.start_price = None
                        self.Liquidate(self.market)
                        self.SetHoldings(self.t_bills, 1)
            else:
                self.Liquidate()
        else:
            self.Liquidate()