686 lines
22 KiB
Python
686 lines
22 KiB
Python
# encoding: UTF-8
|
|
|
|
# 修改编码
|
|
from __future__ import print_function
|
|
import sys
|
|
try:
|
|
reload(sys) # Python 2
|
|
sys.setdefaultencoding('utf8')
|
|
except NameError:
|
|
pass # Python 3
|
|
|
|
# 创建主引擎代理对象
|
|
from vnpy.event import EventEngine2
|
|
from vnpy.trader.vtEvent import (EVENT_TICK, EVENT_ORDER, EVENT_TRADE,
|
|
EVENT_ACCOUNT, EVENT_POSITION, EVENT_LOG,
|
|
EVENT_ERROR, EVENT_CONTRACT)
|
|
from vnpy.trader.vtObject import VtSubscribeReq, VtOrderReq, VtCancelOrderReq
|
|
from vnpy.trader.app.rpcService.rsClient import MainEngineProxy
|
|
from vnpy.trader.app.ctaStrategy.ctaBase import EVENT_CTA_LOG, EVENT_CTA_STRATEGY
|
|
|
|
reqAddress = 'tcp://localhost:6688'
|
|
subAddress = 'tcp://localhost:8866'
|
|
|
|
ee = EventEngine2()
|
|
me = MainEngineProxy(ee)
|
|
me.init(reqAddress, subAddress)
|
|
|
|
#----------------------------------------------------------------------
|
|
def printLog(event):
|
|
"""打印日志"""
|
|
log = event.dict_['data']
|
|
print(log.logTime, log.logContent)
|
|
|
|
ee.register(EVENT_LOG, printLog)
|
|
|
|
|
|
# 载入Web密码
|
|
import json
|
|
import base64
|
|
import datetime
|
|
|
|
TODAY = str(datetime.datetime.now().date())
|
|
|
|
with open("WEB_setting.json") as f:
|
|
setting = json.load(f)
|
|
USERNAME = setting['username']
|
|
PASSWORD = setting['password']
|
|
TOKEN = base64.encodestring(TODAY+PASSWORD).replace('\n','')
|
|
|
|
|
|
# 创建Flask对象
|
|
from flask import Flask, send_file
|
|
from flask_restful import Api, Resource, reqparse
|
|
from flask_socketio import SocketIO
|
|
from flask_cors import *
|
|
|
|
app = Flask(__name__)
|
|
CORS(app, supports_credentials=True)
|
|
api = Api(app)
|
|
socketio = SocketIO(app)
|
|
|
|
# 创建资源
|
|
########################################################################
|
|
class Token(Resource):
|
|
"""登录验证"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('username')
|
|
self.parser.add_argument('password')
|
|
super(Token, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
username = args['username']
|
|
password = args['password']
|
|
|
|
if username == USERNAME and password == PASSWORD:
|
|
return {'result_code':'success','data':TOKEN}
|
|
else:
|
|
return {'result_code':'error','message':'wrong username or password'}
|
|
|
|
|
|
########################################################################
|
|
class Gateway(Resource):
|
|
"""接口"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('gatewayName')
|
|
self.parser.add_argument('token')
|
|
|
|
super(Gateway, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
l = me.getAllGatewayDetails()
|
|
return {'result_code':'success','data':l}
|
|
|
|
#----------------------------------------------------------------------
|
|
def post(self):
|
|
"""连接"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
print('token error')
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
gatewayName = args['gatewayName']
|
|
me.connect(gatewayName)
|
|
return {'result_code':'success','data':''}
|
|
|
|
|
|
########################################################################
|
|
class Order(Resource):
|
|
"""委托"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.getParser = reqparse.RequestParser()
|
|
self.getParser.add_argument('token')
|
|
|
|
self.postParser = reqparse.RequestParser()
|
|
self.postParser.add_argument('vtSymbol')
|
|
self.postParser.add_argument('price')
|
|
self.postParser.add_argument('volume')
|
|
self.postParser.add_argument('priceType')
|
|
self.postParser.add_argument('direction')
|
|
self.postParser.add_argument('offset')
|
|
self.postParser.add_argument('token')
|
|
|
|
self.deleteParser = reqparse.RequestParser()
|
|
self.deleteParser.add_argument('vtOrderID')
|
|
self.deleteParser.add_argument('token')
|
|
|
|
super(Order, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.getParser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
data = me.getAllOrders()
|
|
l = [o.__dict__ for o in data]
|
|
return {'result_code':'success','data':l}
|
|
|
|
#----------------------------------------------------------------------
|
|
def post(self):
|
|
"""发单"""
|
|
args = self.postParser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
print(args)
|
|
vtSymbol = args['vtSymbol']
|
|
price = args['price']
|
|
volume = args['volume']
|
|
priceType = args['priceType']
|
|
direction = args['direction']
|
|
offset = args['offset']
|
|
|
|
contract = me.getContract(vtSymbol)
|
|
if not contract:
|
|
return {'result_code':'error','message':'contract error'}
|
|
|
|
priceType_map = {'PRICETYPE_LIMITPRICE' : u'限价','PRICETYPE_MARKETPRICE' : u'市价','PRICETYPE_FAK' : u'FAK','PRICETYPE_FOK' : u'FOK'}
|
|
direction_map = {'DIRECTION_LONG' : u'多','DIRECTION_SHORT' : u'空'}
|
|
offset_map = {'OFFSET_OPEN' : u'开仓', 'OFFSET_CLOSE' : u'平仓','OFFSET_CLOSETODAY' : u'平今','OFFSET_CLOSEYESTERDAY' : u'平昨'}
|
|
|
|
req = VtOrderReq()
|
|
req.symbol = contract.symbol
|
|
req.exchange = contract.exchange
|
|
req.price = float(price)
|
|
req.volume = int(volume)
|
|
req.priceType = priceType_map[ priceType ]
|
|
req.direction = direction_map[ direction ]
|
|
req.offset = offset_map[ offset ]
|
|
vtOrderID = me.sendOrder(req, contract.gatewayName)
|
|
return {'result_code':'success','data':vtOrderID}
|
|
|
|
#----------------------------------------------------------------------
|
|
def delete(self):
|
|
"""撤单"""
|
|
args = self.deleteParser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
vtOrderID = args['vtOrderID']
|
|
|
|
# 撤单某一委托
|
|
if vtOrderID:
|
|
order = me.getOrder(vtOrderID)
|
|
if not order:
|
|
return {'result_code':'error','message':'vtOrderID error'}
|
|
|
|
self.cancel(order)
|
|
# 全撤
|
|
else:
|
|
l = me.getAllWorkingOrders()
|
|
|
|
for order in l:
|
|
self.cancel(order)
|
|
|
|
return {'result_code':'success','data':""}
|
|
|
|
#----------------------------------------------------------------------
|
|
def cancel(self, order):
|
|
"""撤单"""
|
|
req = VtCancelOrderReq()
|
|
req.orderID = order.orderID
|
|
req.exchange = order.exchange
|
|
req.symbol = order.symbol
|
|
req.frontID = order.frontID
|
|
req.sessionID = order.sessionID
|
|
me.cancelOrder(req, order.gatewayName)
|
|
|
|
|
|
########################################################################
|
|
class Trade(Resource):
|
|
"""成交"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
|
|
super(Trade, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
data = me.getAllTrades()
|
|
l = [o.__dict__ for o in data]
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class Account(Resource):
|
|
"""账户"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
|
|
super(Account, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
data = me.getAllAccounts()
|
|
l = [o.__dict__ for o in data]
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class Position(Resource):
|
|
"""持仓"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
|
|
super(Position, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
data = me.getAllPositions()
|
|
print('position',data)
|
|
l = [o.__dict__ for o in data]
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class Contract(Resource):
|
|
"""合约"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
|
|
super(Contract, self).__init__()
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
data = me.getAllContracts()
|
|
print('Contract',data)
|
|
l = [o.__dict__ for o in data]
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class Log(Resource):
|
|
"""日志"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
|
|
super(Log, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
data = me.getLog()
|
|
l = [o.__dict__ for o in data]
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class Error(Resource):
|
|
"""错误"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
|
|
super(Error, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""查询"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
data = me.getError()
|
|
l = [o.__dict__ for o in data]
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class Tick(Resource):
|
|
"""行情"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('vtSymbol')
|
|
self.parser.add_argument('token')
|
|
super(Tick, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def post(self):
|
|
"""订阅"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
vtSymbol = args['vtSymbol']
|
|
|
|
contract = me.getContract(vtSymbol)
|
|
if not contract:
|
|
return {'result_code':'error','message':'contract error'}
|
|
|
|
req = VtSubscribeReq()
|
|
req.symbol = contract.symbol
|
|
req.exchange = contract.exchange
|
|
req.vtSymbol = contract.vtSymbol
|
|
|
|
me.subscribe(req, contract.gatewayName)
|
|
return {'result_code':'success','data':''}
|
|
|
|
|
|
########################################################################
|
|
class CtaStrategyInit(Resource):
|
|
"""初始化策略"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('name')
|
|
self.parser.add_argument('token')
|
|
super(CtaStrategyInit, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def post(self):
|
|
"""订阅"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
name = args['name']
|
|
|
|
engine = me.getApp('CtaStrategy')
|
|
if not name:
|
|
engine.initAll()
|
|
else:
|
|
engine.initStrategy(name)
|
|
return {'result_code':'success','data':''}
|
|
|
|
|
|
########################################################################
|
|
class CtaStrategyStart(Resource):
|
|
"""启动策略"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('name')
|
|
self.parser.add_argument('token')
|
|
super(CtaStrategyStart, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def post(self):
|
|
"""订阅"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
name = args['name']
|
|
|
|
engine = me.getApp('CtaStrategy')
|
|
if not name:
|
|
engine.startAll()
|
|
else:
|
|
engine.startStrategy(name)
|
|
return {'result_code':'success','data':''}
|
|
|
|
|
|
########################################################################
|
|
class CtaStrategyStop(Resource):
|
|
"""停止策略"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('name')
|
|
self.parser.add_argument('token')
|
|
super(CtaStrategyStop, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def post(self):
|
|
"""订阅"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
name = args['name']
|
|
|
|
engine = me.getApp('CtaStrategy')
|
|
if not name:
|
|
engine.stopAll()
|
|
else:
|
|
engine.stopStrategy(name)
|
|
return {'result_code':'success','data':''}
|
|
|
|
|
|
########################################################################
|
|
class CtaStrategyName(Resource):
|
|
"""»ñÈ¡²ßÂÔÃû"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""³õʼ»¯"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
super(CtaStrategyName, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""»ñÈ¡²ßÂÔÃû"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
engine = me.getApp('CtaStrategy')
|
|
l = engine.getStrategyNames()
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class CtaStrategyLoad(Resource):
|
|
"""加载策略"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('token')
|
|
super(CtaStrategyLoad, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def post(self):
|
|
"""订阅"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
engine = me.getApp('CtaStrategy')
|
|
engine.loadSetting()
|
|
l = engine.getStrategyNames()
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class CtaStrategyParam(Resource):
|
|
"""查询策略参数"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('name')
|
|
self.parser.add_argument('token')
|
|
super(CtaStrategyParam, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""订阅"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
name = args['name']
|
|
|
|
engine = me.getApp('CtaStrategy')
|
|
l = engine.getStrategyParam(name)
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
class CtaStrategyVar(Resource):
|
|
"""查询策略变量"""
|
|
|
|
#----------------------------------------------------------------------
|
|
def __init__(self):
|
|
"""初始化"""
|
|
self.parser = reqparse.RequestParser()
|
|
self.parser.add_argument('name')
|
|
self.parser.add_argument('token')
|
|
super(CtaStrategyVar, self).__init__()
|
|
|
|
#----------------------------------------------------------------------
|
|
def get(self):
|
|
"""订阅"""
|
|
args = self.parser.parse_args()
|
|
token = args['token']
|
|
if token != TOKEN:
|
|
return {'result_code':'error','message':'token error'}
|
|
|
|
name = args['name']
|
|
|
|
engine = me.getApp('CtaStrategy')
|
|
l = engine.getStrategyVar(name)
|
|
return {'result_code':'success','data':l}
|
|
|
|
|
|
########################################################################
|
|
@app.route('/')
|
|
def index_html():
|
|
"""首页"""
|
|
return send_file('./templates/index.html')
|
|
|
|
|
|
# 注册资源
|
|
api.add_resource(Token, '/token')
|
|
api.add_resource(Gateway, '/gateway')
|
|
api.add_resource(Order, '/order')
|
|
api.add_resource(Trade, '/trades')
|
|
api.add_resource(Account, '/account')
|
|
api.add_resource(Position, '/position')
|
|
api.add_resource(Contract, '/contract')
|
|
api.add_resource(Log, '/log')
|
|
api.add_resource(Error, '/error')
|
|
api.add_resource(Tick, '/tick')
|
|
|
|
api.add_resource(CtaStrategyLoad, '/ctastrategy/load')
|
|
api.add_resource(CtaStrategyInit, '/ctastrategy/init')
|
|
api.add_resource(CtaStrategyStart, '/ctastrategy/start')
|
|
api.add_resource(CtaStrategyStop, '/ctastrategy/stop')
|
|
api.add_resource(CtaStrategyParam, '/ctastrategy/param')
|
|
api.add_resource(CtaStrategyVar, '/ctastrategy/var')
|
|
api.add_resource(CtaStrategyName, '/ctastrategy/name')
|
|
|
|
|
|
# SocketIO
|
|
#----------------------------------------------------------------------
|
|
def handleEvent(event):
|
|
"""处理事件"""
|
|
eventType = event.type_
|
|
|
|
eventData = event.dict_['data']
|
|
|
|
if not isinstance(eventData, dict):
|
|
eventData = eventData.__dict__
|
|
|
|
if eventType == 'eTick.':
|
|
del eventData['datetime']
|
|
|
|
socketio.emit(eventType, eventData)
|
|
|
|
|
|
ee.register(EVENT_TICK, handleEvent)
|
|
ee.register(EVENT_ORDER, handleEvent)
|
|
ee.register(EVENT_TRADE, handleEvent)
|
|
ee.register(EVENT_ACCOUNT, handleEvent)
|
|
ee.register(EVENT_POSITION, handleEvent)
|
|
ee.register(EVENT_CONTRACT, handleEvent)
|
|
ee.register(EVENT_LOG, handleEvent)
|
|
ee.register(EVENT_ERROR, handleEvent)
|
|
ee.register(EVENT_CTA_LOG, handleEvent)
|
|
ee.register(EVENT_CTA_STRATEGY, handleEvent)
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
def run():
|
|
"""启动Web服务"""
|
|
socketio.run(app,
|
|
debug=True,
|
|
host='0.0.0.0',
|
|
port=5000,
|
|
use_reloader=False)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run()
|