conditional await

r0fls · Nov 28, 2024 · 9040459 · 9040459
1 parent be975a4
commit 9040459
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/brokers/base_broker.py b/brokers/base_broker.py
@@ -91,7 +91,10 @@ async def is_order_filled(self, order_id):
         '''Check if an order has been filled'''
         logger.debug('Checking if order has been filled', extra={'order_id': order_id})
         try:
-            return await self._is_order_filled(order_id)
+            if asyncio.iscoroutinefunction(self._is_order_filled):
+                return await self._is_order_filled(order_id)
+            else:
+                return self._is_order_filled(order_id)
         except Exception as e:
             logger.error('Failed to check if order has been filled', extra={'error': str(e)})
             return False