"Manual or Auto" OrderPlacement declaration (#41)

* variable manual_or_auto OrderPlacement

- Option to set-and-forget at client init
- Option to override on each request in cases of part-auto part-manual applications

* Update client.py

* Update order.py

Author: dhsmyth

async_rithmic/client.py

@@ -9,7 +9,7 @@
 from .plants.order import OrderPlant
 from .plants.pnl import PnlPlant
 from .logger import logger
-from .enums import SysInfraType
+from .enums import OrderPlacement, SysInfraType
 from .objects import ReconnectionSettings, RetrySettings
 
 def _setup_ssl_context():
@@ -31,6 +31,7 @@ def __init__(
         app_name: str,
         app_version: str,
         url: str,
+        manual_or_auto: OrderPlacement = OrderPlacement.MANUAL,
         **kwargs
     ):
         # Connection events
@@ -60,6 +61,9 @@ def __init__(
         if "://" not in url:
             url = f"wss://{url}"
 
+        assert manual_or_auto in [OrderPlacement.MANUAL, OrderPlacement.AUTO], "manual_or_auto must be an instance of .enums.OrderPlacement"
+        self.manual_or_auto = manual_or_auto
+
         self.credentials = dict(
             user=user,
             password=password,

async_rithmic/plants/order.py

@@ -254,6 +254,8 @@ async def submit_order(
             msg_kwargs["cancel_at_ssboe"] = ssboe
             msg_kwargs["cancel_at_usecs"] = usecs
 
+        manual_or_auto = kwargs.get("manual_or_auto", self.client.manual_or_auto)
+
         return await self._send_and_collect(
             template_id=template_id,
             expected_response=dict(template_id=template_id + 1),
@@ -262,7 +264,7 @@ async def submit_order(
             exchange=exchange,
             price_type=order_type,
             quantity=qty,
-            manual_or_auto=OrderPlacement.MANUAL,
+            manual_or_auto=manual_or_auto,
             transaction_type=transaction_type,
             duration=kwargs["duration"],
             **msg_kwargs
@@ -284,10 +286,12 @@ async def cancel_order(self, **kwargs):
             basket_id = order.basket_id
             account_id = order.account_id
 
+        manual_or_auto = kwargs.get("manual_or_auto", self.client.manual_or_auto)
+
         return await self._send_and_collect(
             template_id=316,
             expected_response=dict(template_id=317),
-            manual_or_auto=OrderPlacement.MANUAL,
+            manual_or_auto=manual_or_auto,
             basket_id=basket_id,
             account_id=account_id,
         )
@@ -296,10 +300,12 @@ async def cancel_all_orders(self, **kwargs):
         """
         Cancel all orders
         """
+        manual_or_auto = kwargs.get("manual_or_auto", self.client.manual_or_auto)
+
         return await self._send_and_collect(
             template_id=346,
             expected_response=dict(template_id=347),
-            manual_or_auto=OrderPlacement.MANUAL,
+            manual_or_auto=manual_or_auto,
             account_id=self._get_account_id(**kwargs)
         )
 
@@ -360,10 +366,13 @@ async def modify_order(self, **kwargs):
 
         # Update the actual order
         msg_kwargs = self._validate_price_fields(order_type, raise_exception=False, **kwargs)
+
+        manual_or_auto = kwargs.get("manual_or_auto", self.client.manual_or_auto)
+
         return await self._send_and_collect(
             template_id=314,
             expected_response=dict(template_id=315),
-            manual_or_auto=OrderPlacement.MANUAL,
+            manual_or_auto=manual_or_auto,
             account_id=order.account_id,
             basket_id=order.basket_id,
             symbol=order.symbol,
@@ -404,10 +413,12 @@ async def exit_position(self, **kwargs):
         You can pass `symbol` and `exchange` to target a specific position, otherwise all the account positions will
         be exited.
         """
+        manual_or_auto = kwargs.get("manual_or_auto", self.client.manual_or_auto)
+
         return await self._send_and_collect(
             template_id=3504,
             expected_response=dict(template_id=3505),
-            manual_or_auto=OrderPlacement.MANUAL,
+            manual_or_auto=manual_or_auto,
             **kwargs
         )