from AlgorithmImports import *

class MacdReversionStrategy(QCAlgorithm):
    def Initialize(self):
        self.SetStartDate(2005, 1, 1)
        self.SetEndDate(2025, 1, 31)
        self.SetCash(1000000)  # Set initial capital to 1 million dollars
        
        self.spxl = self.AddEquity("SPXL", Resolution.Daily).Symbol

        self.fast_period = 12
        self.slow_period = 26
        self.signal_period = 9

        self.macd = self.MACD(self.spxl, self.fast_period, self.slow_period, self.signal_period, MovingAverageType.Wilders, Resolution.Daily, Field.Close)
        self.sma150 = self.SMA(self.spxl, 150, Resolution.Daily)  # 150-day moving average
        self.sma10 = self.SMA(self.spxl, 10, Resolution.Daily)  # 10-day moving average

        self.previous_histogram = None
        self.entry_price = None
        self.stop_loss = 50
        self.SetWarmUp(150)

    def OnData(self, data):
        if self.IsWarmingUp:
            return
        if not self.macd.IsReady or not self.sma150.IsReady or not self.sma10.IsReady:
            return

        price = self.Securities[self.spxl].Close
        macd_histogram = self.macd.Current.Value - self.macd.Signal.Current.Value
        close_above_sma150 = price > self.sma150.Current.Value
        close_below_sma10 = price < self.sma10.Current.Value

        if close_above_sma150 and close_below_sma10 and self.previous_histogram is not None:
            if macd_histogram > self.previous_histogram and macd_histogram < 0:
                self.SetHoldings(self.spxl, 1)  # Use full portfolio on every trade
                self.entry_price = price
        
        self.previous_histogram = macd_histogram

        # Exit condition: closing candle $10 above entry price or $50 below entry price (stop loss)
        if self.Portfolio[self.spxl].Invested and self.entry_price is not None:
            if price >= self.entry_price + 10 or price <= self.entry_price - self.stop_loss:
                self.Liquidate(self.spxl)