730 lines
27 KiB
Python
730 lines
27 KiB
Python
# encoding: UTF-8
|
||
|
||
'''
|
||
vn.shzd的gateway接入
|
||
|
||
1. 期权合约数量太多，为了方便起见默认接口只接收期货合约数据
|
||
2. 直达接口的撤单操作也被视作一个独立的委托，但是在vn.trader中选择忽略
|
||
3. 持仓全部平光后，再次查询时会没有该合约的推送（和CTP不同），为了避免最后平仓
|
||
不更新的情况，使用缓存机制来处理
|
||
'''
|
||
|
||
|
||
import os
|
||
import json
|
||
from copy import copy
|
||
from datetime import datetime
|
||
|
||
from vnshzd import ShzdApi
|
||
from vtGateway import *
|
||
|
||
|
||
# 以下为一些VT类型和SHZD类型的映射字典
|
||
# 价格类型映射
|
||
priceTypeMap = {}
|
||
priceTypeMap[PRICETYPE_LIMITPRICE] = '1'
|
||
priceTypeMap[PRICETYPE_MARKETPRICE] = '2'
|
||
priceTypeMapReverse = {v: k for k, v in priceTypeMap.items()}
|
||
|
||
# 方向类型映射
|
||
directionMap = {}
|
||
directionMap[DIRECTION_LONG] = '1'
|
||
directionMap[DIRECTION_SHORT] = '2'
|
||
directionMapReverse = {v: k for k, v in directionMap.items()}
|
||
|
||
# 交易所类型映射
|
||
exchangeMap = {}
|
||
#exchangeMap[EXCHANGE_CFFEX] = 'CFFEX'
|
||
#exchangeMap[EXCHANGE_SHFE] = 'SHFE'
|
||
#exchangeMap[EXCHANGE_CZCE] = 'CZCE'
|
||
#exchangeMap[EXCHANGE_DCE] = 'DCE'
|
||
exchangeMap[EXCHANGE_HKEX] = 'HKEX'
|
||
exchangeMap[EXCHANGE_CME] = 'CME'
|
||
exchangeMap[EXCHANGE_ICE] = 'ICE'
|
||
exchangeMapReverse = {v:k for k,v in exchangeMap.items()}
|
||
|
||
# 产品类型映射
|
||
productClassMap = {}
|
||
productClassMap[PRODUCT_FUTURES] = 'F'
|
||
productClassMap[PRODUCT_OPTION] = 'O'
|
||
productClassMapReverse = {v:k for k,v in productClassMap.items()}
|
||
|
||
# 委托状态映射
|
||
orderStatusMapReverse = {}
|
||
orderStatusMapReverse['2'] = STATUS_NOTTRADED
|
||
orderStatusMapReverse['3'] = STATUS_PARTTRADED
|
||
orderStatusMapReverse['4'] = STATUS_ALLTRADED
|
||
orderStatusMapReverse['5'] = STATUS_CANCELLED
|
||
orderStatusMapReverse['6'] = STATUS_CANCELLED
|
||
|
||
|
||
########################################################################
|
||
class ShzdGateway(VtGateway):
|
||
"""SHZD接口"""
|
||
|
||
#----------------------------------------------------------------------
|
||
def __init__(self, eventEngine, gatewayName='SHZD'):
|
||
"""Constructor"""
|
||
super(ShzdGateway, self).__init__(eventEngine, gatewayName)
|
||
|
||
self.api = ShzdGatewayApi(self)
|
||
|
||
self.qryEnabled = False # 是否要启动循环查询
|
||
|
||
#----------------------------------------------------------------------
|
||
def connect(self):
|
||
"""连接"""
|
||
# 载入json文件
|
||
fileName = self.gatewayName + '_connect.json'
|
||
path = os.path.abspath(os.path.dirname(__file__))
|
||
fileName = os.path.join(path, fileName)
|
||
|
||
try:
|
||
f = file(fileName)
|
||
except IOError:
|
||
log = VtLogData()
|
||
log.gatewayName = self.gatewayName
|
||
log.logContent = u'读取连接配置出错，请检查'
|
||
self.onLog(log)
|
||
return
|
||
|
||
# 解析json文件
|
||
setting = json.load(f)
|
||
try:
|
||
frontAddress = str(setting['frontAddress'])
|
||
frontPort = int(setting['frontPort'])
|
||
marketAddress = str(setting['marketAddress'])
|
||
marketPort = int(setting['marketPort'])
|
||
userId = str(setting['userId'])
|
||
userPwd = str(setting['userPwd'])
|
||
except KeyError:
|
||
self.writeLog(u'连接配置缺少字段，请检查')
|
||
return
|
||
|
||
# 创建行情和交易接口对象
|
||
self.api.connect(userId, userPwd,
|
||
frontAddress, frontPort,
|
||
marketAddress, marketPort)
|
||
|
||
# 初始化并启动查询
|
||
self.initQuery()
|
||
|
||
#----------------------------------------------------------------------
|
||
def writeLog(self, logContent):
|
||
"""记录日志"""
|
||
log = VtLogData()
|
||
log.gatewayName = self.gatewayName
|
||
log.logContent = logContent
|
||
self.onLog(log)
|
||
|
||
#----------------------------------------------------------------------
|
||
def subscribe(self, subscribeReq):
|
||
"""订阅行情"""
|
||
self.api.subscribe(subscribeReq)
|
||
|
||
#----------------------------------------------------------------------
|
||
def sendOrder(self, orderReq):
|
||
"""发单"""
|
||
return self.api.sendOrder(orderReq)
|
||
|
||
#----------------------------------------------------------------------
|
||
def cancelOrder(self, cancelOrderReq):
|
||
"""撤单"""
|
||
self.api.cancelOrder(cancelOrderReq)
|
||
|
||
#----------------------------------------------------------------------
|
||
def qryAccount(self):
|
||
"""查询账户资金"""
|
||
self.api.qryAccount()
|
||
|
||
#----------------------------------------------------------------------
|
||
def qryPosition(self):
|
||
"""查询持仓"""
|
||
self.api.qryPosition()
|
||
|
||
#----------------------------------------------------------------------
|
||
def close(self):
|
||
"""关闭"""
|
||
self.api.close() # 释放接口对象
|
||
|
||
#----------------------------------------------------------------------
|
||
def initQuery(self):
|
||
"""初始化连续查询"""
|
||
if self.qryEnabled:
|
||
# 需要循环的查询函数列表
|
||
self.qryFunctionList = [self.qryAccount, self.qryPosition]
|
||
|
||
self.qryCount = 0 # 查询触发倒计时
|
||
self.qryTrigger = 2 # 查询触发点
|
||
self.qryNextFunction = 0 # 上次运行的查询函数索引
|
||
|
||
self.startQuery()
|
||
|
||
#----------------------------------------------------------------------
|
||
def query(self, event):
|
||
"""注册到事件处理引擎上的查询函数"""
|
||
self.qryCount += 1
|
||
|
||
if self.qryCount > self.qryTrigger:
|
||
# 清空倒计时
|
||
self.qryCount = 0
|
||
|
||
# 执行查询函数
|
||
function = self.qryFunctionList[self.qryNextFunction]
|
||
function()
|
||
|
||
# 计算下次查询函数的索引，如果超过了列表长度，则重新设为0
|
||
self.qryNextFunction += 1
|
||
if self.qryNextFunction == len(self.qryFunctionList):
|
||
self.qryNextFunction = 0
|
||
|
||
#----------------------------------------------------------------------
|
||
def startQuery(self):
|
||
"""启动连续查询"""
|
||
self.eventEngine.register(EVENT_TIMER, self.query)
|
||
|
||
#----------------------------------------------------------------------
|
||
def setQryEnabled(self, qryEnabled):
|
||
"""设置是否要启动循环查询"""
|
||
self.qryEnabled = qryEnabled
|
||
|
||
|
||
########################################################################
|
||
class ShzdGatewayApi(ShzdApi):
|
||
"""直达接口的继承实现"""
|
||
|
||
#----------------------------------------------------------------------
|
||
def __init__(self, gateway):
|
||
"""Constructor"""
|
||
super(ShzdGatewayApi, self).__init__()
|
||
|
||
self.gateway = gateway
|
||
self.gatewayName = gateway.gatewayName
|
||
|
||
self.userId = EMPTY_STRING # 用户名
|
||
self.accountNo = EMPTY_STRING # 查询等用的单一账号
|
||
self.accountNoList = [] # 账号列表
|
||
|
||
self.tradeCallbacks = {} # 交易回调函数映射
|
||
self.marketCallbacks = {} # 行情回调函数映射
|
||
|
||
# 委托相关
|
||
self.localNo = EMPTY_INT # 本地委托号
|
||
self.orderDict = {} # key为str(localNo)，value为委托对象
|
||
self.orderNoDict = {} # key为OrderNo，value为localNo
|
||
self.localNoDict = {} # key为str(localNo)，value为(SystemNo, OrderNo)
|
||
self.cancelDict = {} # key为等待撤单的str(localNo)，value为CancelOrderReq
|
||
|
||
# 委托号前缀
|
||
n = datetime.now()
|
||
self.orderPrefix = n.strftime("%H%M%S.")
|
||
|
||
# 持仓缓存
|
||
self.posDict = {} # key为vtPositionName，value为VtPositionData
|
||
|
||
# 是否进行了初始化
|
||
self.inited = False
|
||
|
||
self.initCallbacks()
|
||
|
||
#----------------------------------------------------------------------
|
||
def initCallbacks(self):
|
||
"""初始化回调函数映射"""
|
||
# 行情推送
|
||
self.marketCallbacks['MA1'] = self.onMarketData
|
||
|
||
# 登录和查询回报
|
||
self.tradeCallbacks['A1'] = self.onLogin
|
||
self.tradeCallbacks['AC1'] = self.onQryAccount
|
||
self.tradeCallbacks['OS1'] = self.onQryPosition
|
||
self.tradeCallbacks['HY'] = self.onQryContract
|
||
self.tradeCallbacks['ORS1'] = self.onQryOrder
|
||
self.tradeCallbacks['FS1'] = self.onTrade
|
||
|
||
# 下单和撤单确认
|
||
self.tradeCallbacks['O1'] = self.onSendOrder
|
||
self.tradeCallbacks['C1'] = self.onCancelOrder
|
||
|
||
# 成交委托推送
|
||
self.tradeCallbacks['O3'] = self.onTrade
|
||
self.tradeCallbacks['OST'] = self.onOrder
|
||
|
||
#----------------------------------------------------------------------
|
||
def onReceiveErrorInfo(self, errcode, errmsg):
|
||
"""错误推送回报"""
|
||
err = VtErrorData()
|
||
err.gatewayName = self.gatewayName
|
||
err.errorID = errcode
|
||
err.errorMsg = errmsg
|
||
self.gateway.onError(err)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onReceiveMarketInfo(self, data):
|
||
"""行情推送回报"""
|
||
func = self.marketCallbacks.get(data['msgtype'], None)
|
||
if func:
|
||
func(data)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onReceiveTradeInfo(self, data):
|
||
"""交易推送回报"""
|
||
func = self.tradeCallbacks.get(data['msgtype'], None)
|
||
if func:
|
||
func(data)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onMarketData(self, data):
|
||
"""行情推送"""
|
||
tick = VtTickData()
|
||
tick.gatewayName = self.gatewayName
|
||
|
||
tick.symbol = data['307']
|
||
tick.exchange = exchangeMapReverse.get(data['306'], EXCHANGE_UNKNOWN)
|
||
tick.vtSymbol = '.'.join([tick.symbol, tick.exchange])
|
||
|
||
tick.volume = int(data['513'])
|
||
tick.openInterest = int(data['514'])
|
||
|
||
dt = data['512'].split(' ')
|
||
tick.time = dt[1]
|
||
tick.date = dt[0].replace('_', '')
|
||
|
||
try:
|
||
tick.lastPrice = float(data['504'])
|
||
tick.openPrice = float(data['508'])
|
||
tick.highPrice = float(data['506'])
|
||
tick.lowPrice = float(data['507'])
|
||
tick.preClosePrice = float(data['509'])
|
||
|
||
# 可以实现5档深度
|
||
tick.bidPrice1 = float(data['500'])
|
||
tick.bidPrice2 = float(data['515'])
|
||
tick.bidPrice3 = float(data['516'])
|
||
tick.bidPrice4 = float(data['517'])
|
||
tick.bidPrice5 = float(data['518'])
|
||
|
||
tick.bidVolume1 = int(data['501'])
|
||
tick.bidVolume2 = int(data['519'])
|
||
tick.bidVolume3 = int(data['520'])
|
||
tick.bidVolume4 = int(data['521'])
|
||
tick.bidVolume5 = int(data['522'])
|
||
|
||
tick.askPrice1 = float(data['502'])
|
||
tick.askPrice2 = float(data['523'])
|
||
tick.askPrice3 = float(data['524'])
|
||
tick.askPrice4 = float(data['525'])
|
||
tick.askPrice5 = float(data['526'])
|
||
|
||
tick.askVolume1 = int(data['503'])
|
||
tick.askVolume2 = int(data['527'])
|
||
tick.askVolume3 = int(data['528'])
|
||
tick.askVolume4 = int(data['529'])
|
||
tick.askVolume5 = int(data['530'])
|
||
|
||
except ValueError:
|
||
pass
|
||
|
||
self.gateway.onTick(tick)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onLogin(self, data):
|
||
"""登录成功推送"""
|
||
if '11' in data:
|
||
self.accountNo = data['11']
|
||
self.accountNoList.append(data['11'])
|
||
|
||
self.loginStatus = True
|
||
self.gateway.writeLog(u'账户%s，结算货币%s' %(data['11'], data['200']))
|
||
|
||
if '410' in data and data['410'] == '1':
|
||
self.gateway.writeLog(u'登录成功')
|
||
#self.qryContract()
|
||
self.qryOrder()
|
||
self.qryTrade()
|
||
|
||
#----------------------------------------------------------------------
|
||
def onSendOrder(self, data):
|
||
"""下单回报"""
|
||
if not data['404'] or data['404'] == '00000':
|
||
order = VtOrderData()
|
||
order.gatewayName = self.gatewayName
|
||
|
||
order.symbol = data['307']
|
||
order.exchange = exchangeMapReverse.get(data['306'], EXCHANGE_UNKNOWN)
|
||
order.vtSymbol = '.'.join([order.symbol, order.exchange])
|
||
|
||
order.orderID = data['305']
|
||
order.vtOrderID = '.'.join([self.gatewayName, order.orderID])
|
||
|
||
order.direction = directionMapReverse.get(data['308'], DIRECTION_UNKNOWN)
|
||
order.price = float(data['310'])
|
||
order.totalVolume = int(data['309'])
|
||
order.status = orderStatusMapReverse.get(data['405'], STATUS_UNKNOWN)
|
||
order.orderTime = data['346']
|
||
|
||
self.orderDict[order.orderID] = order
|
||
self.localNoDict[order.orderID] = (data['300'], data['301'])
|
||
self.orderNoDict[data['301']] = order.orderID
|
||
|
||
# 委托查询
|
||
if '315' in data:
|
||
order.tradedVolume = int(data['315'])
|
||
|
||
self.gateway.onOrder(copy(order))
|
||
|
||
# 检查是否要撤单
|
||
if order.orderID in self.cancelDict:
|
||
self.cancelOrder(self.cancelDict[order.orderID])
|
||
del self.cancelDict[order.orderID]
|
||
else:
|
||
error = VtErrorData()
|
||
error.gatewayName = self.gatewayName
|
||
error.errorID = data['404']
|
||
error.errorMsg = u'委托失败'
|
||
self.gateway.onError(error)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onCancelOrder(self, data):
|
||
"""撤单回报"""
|
||
orderID = self.orderNoDict[data['301']]
|
||
order = self.orderDict[orderID]
|
||
|
||
if not data['404'] or data['404'] == '00000':
|
||
order.status = STATUS_CANCELLED
|
||
order.cancelTime = data['326']
|
||
self.gateway.onOrder(copy(order))
|
||
else:
|
||
error = VtErrorData()
|
||
error.gatewayName = self.gatewayName
|
||
error.errorID = data['404']
|
||
error.errorMsg = u'撤单失败'
|
||
self.gateway.onError(error)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onTrade(self, data):
|
||
"""成交推送"""
|
||
if '307' in data:
|
||
trade = VtTradeData()
|
||
trade.gatewayName = self.gatewayName
|
||
|
||
trade.symbol = data['307']
|
||
trade.exchange = exchangeMapReverse.get(data['306'], EXCHANGE_UNKNOWN)
|
||
trade.vtSymbol = '.'.join([trade.symbol, trade.exchange])
|
||
|
||
trade.tradeID = data['304']
|
||
trade.vtTradeID = '.'.join([self.gatewayName, trade.tradeID])
|
||
|
||
trade.orderID = data['305']
|
||
trade.vtOrderID = '.'.join([self.gatewayName, trade.orderID])
|
||
|
||
trade.direction = directionMapReverse.get(data['308'], DIRECTION_UNKNOWN)
|
||
|
||
trade.price = float(data['313'])
|
||
trade.volume = int(data['315'])
|
||
trade.tradeTime = data['326']
|
||
|
||
self.gateway.onTrade(trade)
|
||
|
||
elif '410' in data and data['410'] == '1':
|
||
self.gateway.writeLog(u'成交查询完成')
|
||
|
||
#----------------------------------------------------------------------
|
||
def onOrder(self, data):
|
||
"""委托变化推送"""
|
||
orderID = self.orderNoDict.get(data['301'], None)
|
||
if orderID:
|
||
order = self.orderDict[orderID]
|
||
order.tradedVolume = int(data['315'])
|
||
|
||
if order.tradedVolume > 0:
|
||
if order.tradedVolume < order.totalVolume:
|
||
order.status = STATUS_PARTTRADED
|
||
else:
|
||
order.status = STATUS_ALLTRADED
|
||
|
||
self.gateway.onOrder(copy(order))
|
||
|
||
#----------------------------------------------------------------------
|
||
def onQryOrder(self, data):
|
||
"""查询委托回报"""
|
||
if '404' in data and data['404'] != '00000':
|
||
error = VtErrorData()
|
||
error.gatewayName = self.gatewayName
|
||
error.errorID = data['404']
|
||
error.errorMsg = u'查询委托失败'
|
||
self.gateway.onError(error)
|
||
|
||
elif '410' not in data and '307' in data:
|
||
order = VtOrderData()
|
||
order.gatewayName = self.gatewayName
|
||
|
||
order.symbol = data['307']
|
||
order.exchange = exchangeMapReverse.get(data['306'], EXCHANGE_UNKNOWN)
|
||
order.vtSymbol = '.'.join([order.symbol, order.exchange])
|
||
|
||
order.orderID = data['305']
|
||
order.vtOrderID = '.'.join([self.gatewayName, order.orderID])
|
||
|
||
order.direction = directionMapReverse.get(data['308'], DIRECTION_UNKNOWN)
|
||
order.price = float(data['310'])
|
||
order.totalVolume = int(data['309'])
|
||
order.status = orderStatusMapReverse.get(data['405'], STATUS_UNKNOWN)
|
||
order.orderTime = data['346']
|
||
order.cancelTime = data['326']
|
||
|
||
self.orderDict[order.orderID] = order
|
||
self.localNoDict[order.orderID] = (data['300'], data['301'])
|
||
self.orderNoDict[data['301']] = order.orderID
|
||
|
||
order.tradedVolume = int(data['315'])
|
||
|
||
self.gateway.onOrder(copy(order))
|
||
|
||
elif '410' in data and data['410'] == '1':
|
||
self.gateway.writeLog(u'委托查询完成')
|
||
|
||
#----------------------------------------------------------------------
|
||
def onQryPosition(self, data):
|
||
"""持仓查询返回"""
|
||
if '307' in data:
|
||
pos = VtPositionData()
|
||
pos.gatewayName = self.gatewayName
|
||
pos.symbol = data['307']
|
||
pos.exchange = exchangeMapReverse.get(data['306'], EXCHANGE_UNKNOWN)
|
||
pos.vtSymbol = '.'.join([pos.symbol, pos.exchange])
|
||
|
||
# 多头仓位
|
||
longPosName = '.'.join([pos.vtSymbol, DIRECTION_LONG])
|
||
try:
|
||
longPos = self.posDict[longPosName]
|
||
except KeyError:
|
||
longPos = copy(pos)
|
||
longPos.direction = DIRECTION_LONG
|
||
longPos.vtPositionName = longPosName
|
||
self.posDict[longPosName] = longPos
|
||
|
||
longPos.position = int(data['442'])
|
||
longPos.price = float(data['443'])
|
||
|
||
# 空头仓位
|
||
shortPosName = '.'.join([pos.vtSymbol, DIRECTION_SHORT])
|
||
try:
|
||
shortPos = self.posDict[shortPosName]
|
||
except KeyError:
|
||
shortPos = copy(pos)
|
||
shortPos.direction = DIRECTION_SHORT
|
||
shortPos.vtPositionName = shortPosName
|
||
self.posDict[shortPosName] = shortPos
|
||
|
||
shortPos.position = int(data['445'])
|
||
shortPos.price = float(data['446'])
|
||
|
||
# 所有持仓数据推送完成后才向事件引擎中更新持仓数据
|
||
if '410' in data and data['410'] == '1':
|
||
for pos in self.posDict.values():
|
||
self.gateway.onPosition(pos)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onQryAccount(self, data):
|
||
"""账户资金查询返回"""
|
||
if '11' in data:
|
||
account = VtAccountData()
|
||
account.gatewayName = self.gatewayName
|
||
|
||
account.accountID = data['11']
|
||
account.vtAccountID = '.'.join([self.gatewayName, account.accountID])
|
||
account.preBalance = float(data['218'])
|
||
account.balance = float(data['203'])
|
||
account.available = float(data['201'])
|
||
account.commission = float(data['221'])
|
||
account.margin = float(data['212'])
|
||
account.closeProfit = float(data['205'])
|
||
account.positionProfit = float(data['216'])
|
||
|
||
self.gateway.onAccount(account)
|
||
|
||
#----------------------------------------------------------------------
|
||
def onQryContract(self, data):
|
||
"""查询合约推送"""
|
||
if '306' in data and data['306'] in exchangeMapReverse:
|
||
contract = VtContractData()
|
||
contract.gatewayName = self.gatewayName
|
||
|
||
contract.symbol = data['333'] + data['307']
|
||
contract.exchange = exchangeMapReverse.get(data['306'], EXCHANGE_UNKNOWN)
|
||
contract.vtSymbol = '.'.join([contract.symbol, contract.exchange])
|
||
|
||
contract.name = data['332'].decode('GBK')
|
||
contract.productClass = productClassMapReverse.get(data['335'], '')
|
||
contract.size = float(data['336'])
|
||
contract.priceTick = float(data['337'])
|
||
|
||
# 期权合约数量太多，为了方便起见默认接口只接收期货合约数据
|
||
if contract.productClass == PRODUCT_FUTURES:
|
||
self.gateway.onContract(contract)
|
||
|
||
if '410' in data and data['410'] == '1':
|
||
self.gateway.writeLog(u'合约查询完成')
|
||
|
||
#----------------------------------------------------------------------
|
||
def connect(self, userId, userPwd,
|
||
frontAddress, frontPort,
|
||
marketAddress, marketPort):
|
||
"""连接"""
|
||
self.userId = userId
|
||
|
||
# 初始化接口
|
||
n = self.initShZdServer()
|
||
if n:
|
||
self.gateway.writeLog(u'接口初始化失败，原因%s' %n)
|
||
return
|
||
else:
|
||
self.gateway.writeLog(u'接口初始化成功')
|
||
self.inited = True
|
||
|
||
# 连接交易服务器
|
||
n = self.registerFront(frontAddress, frontPort)
|
||
if n:
|
||
self.gateway.writeLog(u'交易服务器连接失败，原因%s' %n)
|
||
return
|
||
else:
|
||
self.gateway.writeLog(u'交易服务器连接成功')
|
||
|
||
# 连接行情服务器
|
||
n = self.registerMarket(marketAddress, marketPort)
|
||
if n:
|
||
self.gateway.writeLog(u'行情服务器连接失败，原因%s' %n)
|
||
return
|
||
else:
|
||
self.gateway.writeLog(u'行情服务器连接成功')
|
||
|
||
# 登录
|
||
req = {}
|
||
req['msgtype'] = 'A'
|
||
req['12'] = 'demo000604'
|
||
req['16'] = '888888'
|
||
self.shzdSendInfoToTrade(req)
|
||
|
||
#----------------------------------------------------------------------
|
||
def subscribe(self, subscribeReq):
|
||
"""订阅行情"""
|
||
req = {}
|
||
req['msgtype'] = 'MA'
|
||
req['11'] = self.accountNo
|
||
req['201'] = '+'
|
||
req['307'] = ','.join([exchangeMap[subscribeReq.exchange], subscribeReq.symbol])
|
||
self.shzdSendInfoToMarket(req)
|
||
|
||
#----------------------------------------------------------------------
|
||
def sendOrder(self, orderReq):
|
||
"""发单"""
|
||
req = {}
|
||
req['msgtype'] = 'O'
|
||
req['12'] = self.userId
|
||
req['11'] = self.accountNo
|
||
req['306'] = exchangeMap.get(orderReq.exchange, '')
|
||
req['307'] = orderReq.symbol
|
||
req['308'] = directionMap.get(orderReq.direction, '')
|
||
req['309'] = str(orderReq.volume)
|
||
req['310'] = str(orderReq.price)
|
||
req['401'] = priceTypeMap.get(orderReq.priceType, '')
|
||
|
||
self.localNo += 1
|
||
req['305'] = self.orderPrefix + str(self.localNo).rjust(10, '0')
|
||
|
||
self.shzdSendInfoToTrade(req)
|
||
|
||
vtOrderID = '.'.join([self.gatewayName, str(self.localNo)])
|
||
return vtOrderID
|
||
|
||
#----------------------------------------------------------------------
|
||
def cancelOrder(self, cancelReq):
|
||
"""撤单"""
|
||
tmp = self.localNoDict.get(cancelReq.orderID, None)
|
||
|
||
if tmp:
|
||
systemNo = tmp[0]
|
||
orderNo = tmp[1]
|
||
order = self.orderDict[cancelReq.orderID]
|
||
|
||
req = {}
|
||
req['msgtype'] = 'C'
|
||
req['12'] = self.userId
|
||
req['11'] = self.accountNo
|
||
req['300'] = systemNo
|
||
req['301'] = orderNo
|
||
req['306'] = exchangeMap.get(order.exchange, '')
|
||
req['307'] = order.symbol
|
||
req['308'] = directionMap.get(order.direction, '')
|
||
req['309'] = str(order.totalVolume)
|
||
req['310'] = str(order.price)
|
||
req['315'] = str(order.tradedVolume)
|
||
|
||
self.shzdSendInfoToTrade(req)
|
||
else:
|
||
self.cancelSet.add(cancelReq)
|
||
|
||
#----------------------------------------------------------------------
|
||
def qryAccount(self):
|
||
"""查询账户"""
|
||
req = {}
|
||
req['msgtype'] = 'AC'
|
||
req['12'] = self.userId
|
||
req['11'] = self.accountNo
|
||
self.shzdSendInfoToTrade(req)
|
||
|
||
#----------------------------------------------------------------------
|
||
def qryPosition(self):
|
||
"""持仓查询"""
|
||
req = {}
|
||
req['msgtype'] = 'OS'
|
||
req['12'] = self.userId
|
||
req['11'] = self.accountNo
|
||
self.shzdSendInfoToTrade(req)
|
||
|
||
# 清空持仓数据
|
||
for pos in self.posDict.values():
|
||
pos.price = 0
|
||
pos.position = 0
|
||
|
||
#----------------------------------------------------------------------
|
||
def qryContract(self):
|
||
"""合约查询"""
|
||
req = {}
|
||
req['msgtype'] = 'HY'
|
||
req['11'] = self.accountNo
|
||
self.shzdSendInfoToTrade(req)
|
||
|
||
#----------------------------------------------------------------------
|
||
def qryTrade(self):
|
||
"""成交查询"""
|
||
req = {}
|
||
req['msgtype'] = 'FS'
|
||
req['12'] = self.userId
|
||
req['11'] = self.accountNo
|
||
self.shzdSendInfoToTrade(req)
|
||
|
||
#----------------------------------------------------------------------
|
||
def qryOrder(self):
|
||
"""委托查询"""
|
||
req = {}
|
||
req['msgtype'] = 'ORS'
|
||
req['12'] = self.userId
|
||
req['11'] = self.accountNo
|
||
self.shzdSendInfoToTrade(req)
|
||
|
||
#----------------------------------------------------------------------
|
||
def close(self):
|
||
"""关闭接口"""
|
||
if self.inited:
|
||
self.release()
|
||
|
||
#----------------------------------------------------------------------
|
||
def printDict(d):
|
||
"""打印字典"""
|
||
print '-' * 50
|
||
l = d.keys()
|
||
l.sort()
|
||
for k in l:
|
||
print k, ':', d[k]
|
||
|