bug fix

vnpy/api/okex/vnokex.py

@@ -149,13 +149,16 @@ class OkexApi(object):
             self.thread.join()
 
     #----------------------------------------------------------------------
-    def onMessage(self, ws, evt):
+    def onMessage(self, *args):
         """
         信息推送事件
         :param ws:  接口
         :param evt: 事件
         :return:
         """
+        evt = None
+        if len(args) == 2:
+            evt = args[-1]
         print(u'vnokex.onMessage:{}'.format(evt))
         
     #----------------------------------------------------------------------
@@ -572,7 +575,7 @@ class WsFuturesApi(object):
         print('OkexContractApi.onClose')
 
     # ----------------------------------------------------------------------
-    def onOpen(self, ws):
+    def onOpen(self, *args):
         """接口打开"""
         print('OkexContractApi.onOpen')
 

vnpy/trader/gateway/okexGateway/okexGateway.py

@@ -103,10 +103,11 @@ class OkexGateway(VtGateway):
 
         # 解析json文件
         setting = json.load(f)
+        trace = False
         try:
             self.apiKey = str(setting['apiKey'])
             self.secretKey = str(setting['secretKey'])
-            trace = setting['trace']
+            trace = setting.get('trace',False)
             self.leverage = setting.get('leverage', 1)
             spot_connect = setting['spot_connect']
             futures_connect = setting['futures_connect']
@@ -430,13 +431,16 @@ class OkexSpotApi(WsSpotApi):
     '''
 
     # ----------------------------------------------------------------------
-    def onMessage(self, ws, evt):
+    def onMessage(self, *args):
         """
         响应信息处理，包括心跳响应、请求响应、数据推送
         :param ws: websocket接口
         :param evt: 消息体
         :return:
         """
+        if len(args)==0:
+            return
+        evt = args[-1]
         # str => json
         decmp_evt = self.inflate(evt)
         ws_data = self.readData(decmp_evt)
@@ -613,7 +617,7 @@ class OkexSpotApi(WsSpotApi):
         #        self.spotOrderInfo(symbol_pair, orderId)
 
     # ----------------------------------------------------------------------
-    def onOpen(self, ws):
+    def onOpen(self, *args):
         """
         ws连接成功事件回调函数
         :param ws:
@@ -2537,7 +2541,6 @@ h_this_week_usd'}
                     # 持仓盈亏
                     account.positionProfit += contract.get('unprofit',0.0)
 
-
                     if account.balance > 0 or account.available > 0:
                         for contractType in CONTRACT_TYPE:
                             self.query_future_position_4fix(symbol=symbol, contractType=contractType)