# region imports
from AlgorithmImports import *
# endregion

class BiotechStrategy(QCAlgorithm):
    def Initialize(self):
        self.SetStartDate(2024, 2, 1)
        self.SetEndDate(2024, 2, 25)
        self.SetCash(100000)

        # Add symbols
        self.symbols = ['ACON', 'SGLY', 'SLRN', 'PEV', 'SOPA']
        self.order_tickets = {}  # Store order tickets for TP and SL

        for symbol in self.symbols:
            equity = self.AddEquity(symbol, Resolution.Minute)
            equity.SetDataNormalizationMode(DataNormalizationMode.Raw)

    def OnData(self, data):
        for symbol in self.symbols:
            if not self.Portfolio[symbol].Invested:
                # Example entry logic
                quantity = 1
                entry_ticket = self.MarketOrder(symbol, quantity)
                
                # Define TP and SL prices
                fill_price = entry_ticket.AverageFillPrice
                tp_price = round(fill_price * 1.10, 2)  # Take profit at +10%
                sl_price = round(fill_price * 0.95, 2)  # Stop loss at -5%
                
                # Place TP and SL orders
                tp_ticket = self.LimitOrder(symbol, -quantity, tp_price)
                sl_ticket = self.StopMarketOrder(symbol, -quantity, sl_price)
                
                # Store tickets for cancellation logic
                self.order_tickets[symbol] = {"tp": tp_ticket, "sl": sl_ticket}

    def OnOrderEvent(self, order_event):
        if order_event.Status == OrderStatus.Filled:
            symbol = str(order_event.Symbol)
            
            # Check if there are active tickets for this symbol
            if symbol not in self.order_tickets or self.order_tickets[symbol] is None:
                return  # No active tickets to process
            
            tickets = self.order_tickets[symbol]
            
            # Check if Take Profit was filled
            if tickets["tp"] is not None and order_event.OrderId == tickets["tp"].OrderId:
                # TP filled; cancel SL
                if tickets["sl"] is not None:
                    tickets["sl"].Cancel()
                    self.Debug(f"Take Profit filled for {symbol}. Stop Loss canceled.")
                self.order_tickets[symbol] = None  # Clear tickets after TP fills
            
            # Check if Stop Loss was filled
            elif tickets["sl"] is not None and order_event.OrderId == tickets["sl"].OrderId:
                # SL filled; cancel TP
                if tickets["tp"] is not None:
                    tickets["tp"].Cancel()
                    self.Debug(f"Stop Loss filled for {symbol}. Take Profit canceled.")
                self.order_tickets[symbol] = None  # Clear tickets after SL fills