为VNPY增加数据库记录交易数据功能
在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名字
1 2 3 4 5 6 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 或者在
1 self.insertData(DEAL_DB_NAME,数据self.name,t)直接写死,"DEAL_DB_NAME"
-----------------------------------------------2019年01月30日更新--------------------------------------------------------
发现写入数据的加数据库记录交易交易时间 tradeTime 只有时间,没有日期,不方便分析。开了下源代码,修改 vnpy.trader.app.ctaStrategy.ctaTemplate 比较简单。
如果发现插入数据是VtTradeData对象,香港云服务器修改tradeTime
1 2 3 4 5 6 7 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)