From 43584e9030ace5a577bd4181f81d86234cdd4652 Mon Sep 17 00:00:00 2001
From: yutiansut <yutiansut@qq.com>
Date: Wed, 10 Jul 2019 19:15:37 +0800
Subject: [PATCH] #

---
 QATRADER/__init__.py |  5 +++--
 QATRADER/trader.py   | 13 +++++++++----
 2 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/QATRADER/__init__.py b/QATRADER/__init__.py
index 2fe2737..87044b7 100644
--- a/QATRADER/__init__.py
+++ b/QATRADER/__init__.py
@@ -13,7 +13,8 @@
 @click.option('--portfolio', default='default')
 @click.option('--bank_password', default=None)
 @click.option('--capital_password', default=None)
-def single_trade(acc, pwd, wsuri, broker, trade_host, ping_gap, taskid, portfolio, bank_password, capital_password):
+@click.option('--appid', default=None)
+def single_trade(acc, pwd, wsuri, broker, trade_host, ping_gap, taskid, portfolio, bank_password, capital_password, appid):
     QA_TRADER(str(acc), str(pwd), wsuri=wsuri, trade_host=trade_host, portfolio=portfolio, bank_password=bank_password, capital_password=capital_password,
-                  broker_name=broker, ping_gap=ping_gap, taskid=taskid).start()
+                  appid=appid, broker_name=broker, ping_gap=ping_gap, taskid=taskid).start()
 
diff --git a/QATRADER/trader.py b/QATRADER/trader.py
index c80ccff..998266b 100644
--- a/QATRADER/trader.py
+++ b/QATRADER/trader.py
@@ -39,7 +39,7 @@ class QA_TRADER(QA_Thread):
 
     def __init__(self, account_cookie, password, wsuri, broker_name='simnow', portfolio='default',
                  trade_host='localhost', trade_port='5672', pub_host='localhost', pub_port='5672', sig=True, ping_gap=1,
-                 bank_password=None, capital_password=None,
+                 bank_password=None, capital_password=None, appid= None,
                  if_restart=True, taskid=None, database=pymongo.MongoClient()):
 
         super().__init__(name='QATRADER_{}'.format(account_cookie))
@@ -70,6 +70,7 @@ def __init__(self, account_cookie, password, wsuri, broker_name='simnow', portfo
         self.bank_password = bank_password
         self.capital_password = capital_password
         self.tempPass = ''
+        self.appid = appid
 
         self.sub = consumer.subscriber_routing(host=trade_host, port=trade_port, user=trade_server_user, password=trade_server_password,
                                                exchange=trade_server_order_exchange, routing_key=self.account_cookie)
@@ -90,7 +91,7 @@ def on_message(self,  message):
             message, dict) else json.loads(str(message))
         #print(message)
         message = fix_dict(message)
-        #print(message)
+        print(message)
         self.pub.pub(json.dumps(message), routing_key=self.account_cookie)
         """需要在这里维持实时账户逻辑
 
@@ -199,8 +200,12 @@ def generate_websocket(self,):
 
         def _onopen(ws):
             def run():
-                ws.send(login(
-                    name=self.account_cookie, password=self.password, broker=self.broker_name))
+                if self.appid is None:
+                    ws.send(login(
+                        name=self.account_cookie, password=self.password, broker=self.broker_name))
+                else:
+                    ws.send(login(
+                        name=self.account_cookie, password=self.password, broker=self.broker_name, appid=self.appid))
             threading.Thread(target=run, daemon=False).start()
         ws.on_open = _onopen
         return ws