add user_id param

Author: tigersk

tigeropen/tiger_open_config.py

@@ -46,6 +46,8 @@ def __init__(self, sandbox_debug=False):
         self._private_key = ''
         # 请求签名类型，推荐RSA2
         self._sign_type = SIGN_TYPE
+        # 用户id
+        self._user_id = ''
 
         # 老虎证券开放平台网关地址
         self._server_url = SERVER_URL
@@ -162,10 +164,18 @@ def timeout(self):
     def timeout(self, value):
         self._timeout = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
 
 def get_client_config(private_key_path, tiger_id, account, standard_account=None, paper_account=None,
                       sandbox_debug=False, sign_type=None, timeout=None, language=None, charset=None,
-                      server_url=None, socket_host_port=None):
+                      server_url=None, socket_host_port=None, user_id=None):
     """
     生成客户端配置
     :param private_key_path: 私钥文件路径, 如 '/Users/tiger/.ssh/rsa_private_key.pem'
@@ -180,6 +190,7 @@ def get_client_config(private_key_path, tiger_id, account, standard_account=None
     :param charset: 字符集编码
     :param server_url: 网关地址
     :param socket_host_port: 推送长连接的域名端口, 值为协议, 域名, 端口构成的三元组
+    :param user_id: 用户id
     :return:
     """
     config = TigerOpenClientConfig(sandbox_debug=sandbox_debug)
@@ -202,5 +213,7 @@ def get_client_config(private_key_path, tiger_id, account, standard_account=None
         config.server_url = server_url
     if socket_host_port:
         config.socket_host_port = socket_host_port
+    if user_id:
+        config.user_id = user_id
     return config
 

tigeropen/trade/domain/order.py

@@ -11,17 +11,18 @@
 ALGO_PARAMS_TAG_MAP = {'noTakeLiq': 'no_take_liq', 'startTime': 'start_time', 'endTime': 'end_time',
                        'participationRate': 'participation_rate', 'allowPastEndTime': 'allow_past_end_time'}
 
+
 class Order(object):
     __slots__ = ["account", "id", "order_id", "parent_id", "order_time", "reason", "trade_time", "contract", "action",
                  "quantity", "filled", "_remaining", "avg_fill_price", "commission", "realized_pnl", "_status",
                  "trail_stop_price", "limit_price", "aux_price", "trailing_percent", "percent_offset", "action",
-                 "order_type", "time_in_force", "outside_rth", "order_legs", "algo_params"]
+                 "order_type", "time_in_force", "outside_rth", "order_legs", "algo_params", "user_id"]
 
     def __init__(self, account, contract, action, order_type, quantity, limit_price=None, aux_price=None,
                  trail_stop_price=None, trailing_percent=None, percent_offset=None, time_in_force=None,
                  outside_rth=None, filled=0, avg_fill_price=0, commission=None, realized_pnl=None,
                  id=None, order_id=None, parent_id=None, order_time=None, trade_time=None, order_legs=None,
-                 algo_params=None):
+                 algo_params=None, user_id=None):
         """
         - account: 订单所属的账户
         - id: 全局订单 id
@@ -49,6 +50,7 @@ def __init__(self, account, contract, action, order_type, quantity, limit_price=
         - remaining: 未成交的数量
         - order_legs: 附加订单列表
         - algo_params: 算法订单参数
+        - user_id: 用户id
         """
 
         self.id = id
@@ -77,6 +79,7 @@ def __init__(self, account, contract, action, order_type, quantity, limit_price=
         self.trade_time = trade_time
         self.order_legs = order_legs
         self.algo_params = algo_params
+        self.user_id = user_id
 
     def to_dict(self):
         dct = {name: getattr(self, name) for name in self.__slots__ if name not in ORDER_FIELDS_TO_IGNORE}

tigeropen/trade/request/model.py

@@ -9,6 +9,7 @@
 class AccountsParams(object):
     def __init__(self):
         self._account = None
+        self._user_id = None
 
     @property
     def account(self):
@@ -18,17 +19,27 @@ def account(self):
     def account(self, value):
         self._account = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
     def to_openapi_dict(self):
         params = dict()
         if self.account:
             params['account'] = self.account
-
+        if self.user_id:
+            params['user_id'] = self.user_id
         return params
 
 
 class AssetParams(object):
     def __init__(self):
         self._account = None
+        self._user_id = None
         self._segment = False
         self._market_value = False
         self._sub_accounts = None
@@ -41,6 +52,14 @@ def account(self):
     def account(self, value):
         self._account = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
     @property
     def segment(self):
         return self._segment
@@ -70,6 +89,9 @@ def to_openapi_dict(self):
         if self.account:
             params['account'] = self.account
 
+        if self.user_id:
+            params['user_id'] = self.user_id
+
         if self.segment:
             params['segment'] = self.segment
 
@@ -85,6 +107,7 @@ def to_openapi_dict(self):
 class PositionParams(object):
     def __init__(self):
         self._account = None
+        self._user_id = None
         self._symbol = None
         self._sec_type = None
         self._currency = None
@@ -99,6 +122,14 @@ def account(self):
     def account(self, value):
         self._account = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
     @property
     def symbol(self):
         return self._symbol
@@ -144,6 +175,9 @@ def to_openapi_dict(self):
         if self.account:
             params['account'] = self.account
 
+        if self.user_id:
+            params['user_id'] = self.user_id
+
         if self.symbol:
             params['symbol'] = self.symbol
 
@@ -165,6 +199,7 @@ def to_openapi_dict(self):
 class ContractParams(object):
     def __init__(self):
         self._account = None
+        self._user_id = None
         self._symbol = None
         self._symbols = None
         self._sec_type = None
@@ -182,6 +217,14 @@ def account(self):
     def account(self, value):
         self._account = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
     @property
     def symbol(self):
         return self._symbol
@@ -251,6 +294,9 @@ def to_openapi_dict(self):
         if self.account:
             params['account'] = self.account
 
+        if self.user_id:
+            params['user_id'] = self.user_id
+
         if self.symbol:
             params['symbol'] = self.symbol
 
@@ -281,6 +327,7 @@ def to_openapi_dict(self):
 class OrderParams(object):
     def __init__(self):
         self._account = None  # 账户
+        self._user_id = None
         self._id = None  # 订单号(全局)
         self._order_id = None  # 订单号(账户维度)
         self._is_brief = None
@@ -294,6 +341,14 @@ def account(self):
     def account(self, value):
         self._account = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
     @property
     def order_id(self):
         return self._order_id
@@ -331,6 +386,9 @@ def to_openapi_dict(self):
         if self.account:
             params['account'] = self.account
 
+        if self.user_id:
+            params['user_id'] = self.user_id
+
         if self.order_id:
             params['order_id'] = self.order_id
 
@@ -349,6 +407,7 @@ def to_openapi_dict(self):
 class OrdersParams(object):
     def __init__(self):
         self._account = None  # 账户
+        self._user_id = None
         self._market = None  # 市场
         self._sec_type = None  # 合约类型
         self._symbol = None  # 合约代码
@@ -368,6 +427,14 @@ def account(self):
     def account(self, value):
         self._account = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
     @property
     def market(self):
         return self._market
@@ -453,6 +520,9 @@ def to_openapi_dict(self):
         if self.account:
             params['account'] = self.account
 
+        if self.user_id:
+            params['user_id'] = self.user_id
+
         if self.market:
             params['market'] = self.market
 
@@ -489,6 +559,7 @@ def to_openapi_dict(self):
 class PlaceModifyOrderParams(object):
     def __init__(self):
         self.account = None
+        self.user_id = None
         self.id = None
         self.order_id = None
         self.contract = None
@@ -530,6 +601,8 @@ def to_openapi_dict(self):
 
             if self.account:
                 params['account'] = self.account
+            if self.user_id:
+                params['user_id'] = self.user_id
 
             if self.order_id:
                 params['order_id'] = self.order_id
@@ -591,6 +664,7 @@ def to_openapi_dict(self):
 class CancelOrderParams(object):
     def __init__(self):
         self._account = None
+        self._user_id = None
         self._id = None
         self._order_id = None
 
@@ -602,6 +676,14 @@ def account(self):
     def account(self, value):
         self._account = value
 
+    @property
+    def user_id(self):
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, value):
+        self._user_id = value
+
     @property
     def order_id(self):
         return self._order_id
@@ -623,6 +705,9 @@ def to_openapi_dict(self):
         if self.account:
             params['account'] = self.account
 
+        if self.user_id:
+            params['user_id'] = self.user_id
+
         if self.order_id:
             params['order_id'] = self.order_id
 

tigeropen/trade/response/orders_response.py

@@ -31,7 +31,7 @@ def __init__(self):
         self.orders = []
         self._is_success = None
 
-    def parse_response_content(self, response_content):
+    def parse_response_content(self, response_content, user_id=None):
         response = super(OrdersResponse, self).parse_response_content(response_content)
         if 'is_success' in response:
             self._is_success = response['is_success']
@@ -40,16 +40,16 @@ def parse_response_content(self, response_content):
             data_json = json.loads(self.data)
             if 'items' in data_json:
                 for item in data_json['items']:
-                    order = OrdersResponse.parse_order(item)
+                    order = OrdersResponse.parse_order(item, user_id)
                     if order:
                         self.orders.append(order)
             elif 'symbol' in data_json:
-                order = OrdersResponse.parse_order(data_json)
+                order = OrdersResponse.parse_order(data_json, user_id)
                 if order:
                     self.orders.append(order)
 
     @staticmethod
-    def parse_order(item):
+    def parse_order(item, user_id=None):
         contract_fields = {}
         order_fields = {}
         for key, value in item.items():
@@ -110,7 +110,8 @@ def parse_order(item):
             order.trade_time = order_fields.get('trade_time')
         if 'reason' in order_fields:
             order.reason = order_fields.get('reason')
-
+        if user_id is not None:
+            order.user_id = user_id
         order.status = status
 
         return order