fix balance in place order

r0fls · Jun 18, 2024 · 3ef719e · 3ef719e
1 parent 32344d0
commit 3ef719e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/brokers/base_broker.py b/brokers/base_broker.py
@@ -154,7 +154,7 @@ async def place_order(self, symbol, quantity, order_type, strategy, price=None):
 
                 # Fetch the latest cash balance for the strategy
                 latest_balance = session.query(Balance).filter_by(
-                    strategy=strategy, type='cash').order_by(Balance.timestamp.desc()).first()
+                    broker=self.broker_name, strategy=strategy, type='cash').order_by(Balance.timestamp.desc()).first()
                 if latest_balance:
                     # Calculate the order cost
                     order_cost = trade.executed_price * quantity
@@ -175,6 +175,8 @@ async def place_order(self, symbol, quantity, order_type, strategy, price=None):
                     )
                     session.add(new_balance)
                     session.commit()
+                else:
+                    logger.info('No balance records found for {strategy} in {self.broker_name}')
 
             return response
         except Exception as e: