Merge pull request #113 from alpacahq/add-configs-and-replace-order

Add configs and replace order

Author: ttt733

alpaca_trade_api/entity.py

@@ -39,6 +39,10 @@ class Account(Entity):
     pass
 
 
+class AccountConfigurations(Entity):
+    pass
+
+
 class Asset(Entity):
     pass
 

alpaca_trade_api/rest.py

@@ -10,8 +10,8 @@
     get_api_version,
 )
 from .entity import (
-    Account, Asset, Order, Position,
-    BarSet, Clock, Calendar,
+    Account, AccountConfigurations, Asset,
+    Order, Position, BarSet, Clock, Calendar,
 )
 from . import polygon
 
@@ -151,6 +151,9 @@ def get(self, path, data=None):
     def post(self, path, data=None):
         return self._request('POST', path, data)
 
+    def patch(self, path, data=None):
+        return self._request('PATCH', path, data)
+
     def delete(self, path, data=None):
         return self._request('DELETE', path, data)
 
@@ -165,6 +168,31 @@ def get_account(self):
         resp = self.get('/account')
         return Account(resp)
 
+    def get_account_configurations(self):
+        '''Get account configs'''
+        resp = self.get('/account/configurations')
+        return AccountConfigurations(resp)
+
+    def update_account_configurations(
+        self,
+        no_shorting=None,
+        dtbp_check=None,
+        trade_confirm_email=None,
+        suspend_trade=None
+    ):
+        '''Update account configs'''
+        params = {}
+        if no_shorting is not None:
+            params['no_shorting'] = no_shorting
+        if dtbp_check is not None:
+            params['dtbp_check'] = dtbp_check
+        if trade_confirm_email is not None:
+            params['trade_confirm_email'] = trade_confirm_email
+        if suspend_trade is not None:
+            params['suspend_trade'] = suspend_trade
+        resp = self.patch('/account/configurations', params)
+        return AccountConfigurations(resp)
+
     def list_orders(self, status=None, limit=None, after=None, until=None,
                     direction=None, params=None):
         '''
@@ -220,6 +248,29 @@ def get_order(self, order_id):
         resp = self.get('/orders/{}'.format(order_id))
         return Order(resp)
 
+    def replace_order(
+        self,
+        order_id,
+        qty=None,
+        limit_price=None,
+        stop_price=None,
+        time_in_force=None,
+        client_order_id=None
+    ):
+        params = {}
+        if qty is not None:
+            params['qty'] = qty
+        if limit_price is not None:
+            params['limit_price'] = limit_price
+        if stop_price is not None:
+            params['stop_price'] = stop_price
+        if time_in_force is not None:
+            params['time_in_force'] = time_in_force
+        if client_order_id is not None:
+            params['client_order_id'] = client_order_id
+        resp = self.patch('/orders/{}'.format(order_id), params)
+        return Order(resp)
+
     def cancel_order(self, order_id):
         '''Cancel an order'''
         self.delete('/orders/{}'.format(order_id))