在VNPY中,并没有提供实际交易数据库记录功能,虽然可以通过第三方交易软件去读取,但是对于针对单个策略交易效果验证还是不方便,这里说以下为VNPY增加数据库记录交易数据功能。
其实主要还是调用VNPY已经提供的功能。
1.在vnpy.trader.app.ctaStrategy.ctaBase中增加一个新的数据库。
按照你的习惯给名字,比如 DEAL_DB_NAME = 'VnTrader_DEAL_Db'
2.在 vnpy.trader.app.ctaStrategy.ctaTemplate 中,增加一个新的function.
def insertData(self, dbName, collectionName, data): if self.trading: self.ctaEngine.insertData(dbName, collectionName, data)
3. 在策略onTrade function中加入下面代码,用策略名字做collection名字
def onTrade(self, trade):
# 发出状态更新事件
"""收到成交推送(必须由用户继承实现)"""
# 对于无需做细粒度委托控制的策略,可以忽略onOrder
t = trade
self.insertData(DEAL_DB_NAME,self.name,t)
在策略代码中,如果没有DEAL_DB_NAME的引用会不成功,请在策略中加入引用
from vnpy.trader.app.ctaStrategy.ctaBase import DEAL_DB_NAME
或者在
self.insertData(DEAL_DB_NAME,self.name,t)
直接写死,"DEAL_DB_NAME"
-----------------------------------------------2019年01月30日更新--------------------------------------------------------
发现写入数据的交易时间 tradeTime 只有时间,没有日期,不方便分析。开了下源代码,修改 vnpy.trader.app.ctaStrategy.ctaTemplate 比较简单。
如果发现插入数据是VtTradeData对象,修改tradeTime
from vnpy.trader.vtObject import VtTradeData
from datetime import datetime
def insertData(self, dbName, collectionName, data):
if self.trading:
if isinstance(data,VtTradeData):
data.tradeTime = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
self.ctaEngine.insertData(dbName, collectionName, data)
来自 “ ITPUB博客 ” ,链接:http://blog.itpub.net/22259926/viewspace-2564798/,如需转载,请注明出处,否则将追究法律责任。
转载于:http://blog.itpub.net/22259926/viewspace-2564798/