# https://quantpedia.com/strategies/paired-switching/
#
# This strategy is very flexible. Investors could use stocks, funds, or ETFs as an investment vehicle. We show simple trading rules for a sample strategy
# from the source research paper. The investor uses two Vanguard funds as his investment vehicles – one equity fund (VFINX) and one government bond 
from AlgorithmImports import *

# fund (VUSTX). These two funds have a negative correlation as they are proxies for two negatively correlated asset classes. The investor looks at the
# performance of the two funds over the prior quarter and buys the fund that has a higher return during the ranking period. The position is held for one 
# quarter (the investment period). At the end of the investment period, the cycle is repeated.

class PairedSwitching(QCAlgorithm):
    
    def Initialize(self):
        self.SetStartDate(2004, 1, 1)
        self.SetCash(100000)
        
        self.first_symbol = self.AddEquity("SPY", Resolution.Daily).Symbol
        self.second_symbol = self.AddEquity("AGG", Resolution.Daily).Symbol
        self.recent_month = -1

    def OnData(self, data):
        if self.Time.month == self.recent_month:
            return
        self.recent_month = self.Time.month
        
        if(self.recent_month % 3 == 0):
            if self.first_symbol in data and self.second_symbol in data:
                history_call = self.History([self.first_symbol, self.second_symbol], timedelta(days=90))
                if not history_call.empty:
                    first_bars = history_call.loc[self.first_symbol.Value]
                    last_p1 = first_bars["close"].iloc[0]
    
                    second_bars = history_call.loc[self.second_symbol.Value]
                    last_p2 = second_bars["close"].iloc[0]
    
                    # Calculates performance of funds over the prior quarter.
                    first_performance = (float(self.Securities[self.first_symbol].Price) - float(last_p1)) / (float(self.Securities[self.first_symbol].Price))
                    second_performance = (float(self.Securities[self.second_symbol].Price) - float(last_p2)) / (float(self.Securities[self.second_symbol].Price))
                    
                    # Buys the fund that has the higher return during the period.
                    if(first_performance > second_performance):
                        if(self.Securities[self.second_symbol].Invested):
                            self.Liquidate(self.second_symbol)
                        self.SetHoldings(self.first_symbol, 1)
                    else:
                        if(self.Securities[self.first_symbol].Invested):
                            self.Liquidate(self.first_symbol)
                        self.SetHoldings(self.second_symbol, 1)