fix for order circle color

Lumiwealth · Nov 27, 2024 · e8ec025 · e8ec025
1 parent 650ad42
commit e8ec025
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/lumibot/brokers/interactive_brokers_rest.py b/lumibot/brokers/interactive_brokers_rest.py
@@ -722,6 +722,7 @@ def submit_orders(
                 order.order_class = Order.OrderClass.MULTILEG
                 order.identifier = response[0]["order_id"]
                 order.status=Order.OrderStatus.SUBMITTED
+                order.side = order_data['orders'][0]['side'].lower() if order_data is not None else None
 
                 order.child_orders = orders
                 for n, child_order in enumerate(order.child_orders):

diff --git a/lumibot/entities/order.py b/lumibot/entities/order.py
@@ -364,16 +364,20 @@ def __init__(
         )
 
     def is_buy_order(self):
-        return self.side == self.OrderSide.BUY or \
-            self.side == self.OrderSide.BUY_TO_OPEN or \
-            self.side == self.OrderSide.BUY_TO_COVER or \
-            self.side == self.OrderSide.BUY_TO_CLOSE
+        return self.side is not None and (
+            self.side.lower() == self.OrderSide.BUY or
+            self.side.lower() == self.OrderSide.BUY_TO_OPEN or
+            self.side.lower() == self.OrderSide.BUY_TO_COVER or
+            self.side.lower() == self.OrderSide.BUY_TO_CLOSE
+        )
 
     def is_sell_order(self):
-        return self.side == self.OrderSide.SELL or \
-            self.side == self.OrderSide.SELL_SHORT or \
-            self.side == self.OrderSide.SELL_TO_OPEN or \
-            self.side == self.OrderSide.SELL_TO_CLOSE
+        return self.side is not None and (
+            self.side.lower() == self.OrderSide.SELL or
+            self.side.lower() == self.OrderSide.SELL_SHORT or
+            self.side.lower() == self.OrderSide.SELL_TO_OPEN or
+            self.side.lower() == self.OrderSide.SELL_TO_CLOSE
+        )
 
     def is_parent(self) -> bool:
         """