怎樣為VNPY增加數據庫記錄交易數據功能

本篇文章為大家展示了怎樣為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名字

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"

發現寫入數據的交易時間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)

上述內容就是怎樣為VNPY增加數據庫記錄交易數據功能，你們學到知識或技能了嗎？如果還想學到更多技能或者豐富自己的知識儲備，歡迎關注億速云行業資訊頻道。