[Add]新增针对数字货币的CryptoTrader

examples/CryptoTrader/run.py

@@ -17,7 +17,6 @@ system = platform.system()
 from vnpy.event import EventEngine
 from vnpy.trader.vtEngine import MainEngine
 from vnpy.trader.uiQt import createQApp
-from vnpy.trader.uiMainWindow import MainWindow
 
 # 加载底层接口
 from vnpy.trader.gateway import (huobiGateway, okexGateway,
@@ -29,6 +28,8 @@ from vnpy.trader.gateway import (huobiGateway, okexGateway,
 # 加载上层应用
 from vnpy.trader.app import (algoTrading)
 
+# 当前目录组件
+from uiCryptoWindow import MainWindow
 
 #----------------------------------------------------------------------
 def main():

examples/CryptoTrader/uiCryptoWindow.py

@@ -0,0 +1,331 @@
+# encoding: UTF-8
+
+import psutil
+import traceback
+
+from vnpy.trader.vtFunction import loadIconPath
+from vnpy.trader.vtGlobal import globalSetting
+
+from uiCryptoWidget import *
+
+
+########################################################################
+class MainWindow(QtWidgets.QMainWindow):
+    """主窗口"""  
+
+    signalStatusBar = QtCore.Signal(type(Event()))
+
+    #----------------------------------------------------------------------
+    def __init__(self, mainEngine, eventEngine):
+        """Constructor"""          
+        super(MainWindow, self).__init__()
+        
+        self.mainEngine = mainEngine
+        self.eventEngine = eventEngine
+        
+        l = self.mainEngine.getAllGatewayDetails()
+        self.gatewayNameList = [d['gatewayName'] for d in l]        
+        
+        self.widgetDict = {}    # 用来保存子窗口的字典
+        
+        # 获取主引擎中的上层应用信息
+        self.appDetailList = self.mainEngine.getAllAppDetails()
+        
+        self.initUi()
+        self.loadWindowSettings('custom')
+        
+    #----------------------------------------------------------------------
+    def initUi(self):
+        """初始化界面"""
+        self.setWindowTitle('VnTrader')
+        self.initCentral()
+        self.initMenu()
+        self.initStatusBar()
+        
+    #----------------------------------------------------------------------
+    def initCentral(self):
+        """初始化中心区域"""
+        widgetMarketM, dockMarketM = self.createDock(MarketMonitor, vtText.MARKET_DATA, QtCore.Qt.RightDockWidgetArea)
+        widgetLogM, dockLogM = self.createDock(LogMonitor, vtText.LOG, QtCore.Qt.BottomDockWidgetArea)
+        widgetTradeM, dockTradeM = self.createDock(TradeMonitor, vtText.TRADE, QtCore.Qt.BottomDockWidgetArea)
+        widgetOrderM, dockOrderM = self.createDock(OrderMonitor, vtText.ORDER, QtCore.Qt.RightDockWidgetArea)
+        widgetWorkingOrderM, dockWorkingOrderM = self.createDock(WorkingOrderMonitor, vtText.WORKING_ORDER, QtCore.Qt.BottomDockWidgetArea)
+        widgetPositionM, dockPositionM = self.createDock(PositionMonitor, vtText.POSITION, QtCore.Qt.BottomDockWidgetArea)
+        widgetAccountM, dockAccountM = self.createDock(AccountMonitor, vtText.ACCOUNT, QtCore.Qt.BottomDockWidgetArea)
+        widgetTradingW, dockTradingW = self.createDock(TradingWidget, vtText.TRADING, QtCore.Qt.LeftDockWidgetArea)
+    
+        self.tabifyDockWidget(dockOrderM, dockWorkingOrderM)
+        self.tabifyDockWidget(dockPositionM, dockAccountM)
+        
+        # 保存默认设置
+        self.saveWindowSettings('default')
+        
+    #----------------------------------------------------------------------
+    def initMenu(self):
+        """初始化菜单"""
+        # 创建菜单
+        menubar = self.menuBar()
+        
+        # 设计为只显示存在的接口
+        gatewayDetails = self.mainEngine.getAllGatewayDetails()
+        
+        sysMenu = menubar.addMenu(vtText.SYSTEM)
+        
+        for d in gatewayDetails:
+            if d['gatewayType'] == GATEWAYTYPE_FUTURES:
+                self.addConnectAction(sysMenu, d['gatewayName'], d['gatewayDisplayName'])
+        sysMenu.addSeparator()
+        
+        for d in gatewayDetails:
+            if d['gatewayType'] == GATEWAYTYPE_EQUITY:
+                self.addConnectAction(sysMenu, d['gatewayName'], d['gatewayDisplayName'])
+        sysMenu.addSeparator()
+        
+        for d in gatewayDetails:
+            if d['gatewayType'] == GATEWAYTYPE_INTERNATIONAL:
+                self.addConnectAction(sysMenu, d['gatewayName'], d['gatewayDisplayName'])       
+        sysMenu.addSeparator()
+                
+        for d in gatewayDetails:
+            if d['gatewayType'] == GATEWAYTYPE_BTC:
+                self.addConnectAction(sysMenu, d['gatewayName'], d['gatewayDisplayName'])
+        sysMenu.addSeparator()
+                
+        for d in gatewayDetails:
+            if d['gatewayType'] == GATEWAYTYPE_DATA:
+                self.addConnectAction(sysMenu, d['gatewayName'], d['gatewayDisplayName'])
+        
+        sysMenu.addSeparator()
+        sysMenu.addAction(self.createAction(vtText.CONNECT_DATABASE, self.mainEngine.dbConnect, loadIconPath('database.ico')))
+        sysMenu.addSeparator()
+        sysMenu.addAction(self.createAction(vtText.EXIT, self.close, loadIconPath('exit.ico')))
+        
+        # 功能应用
+        appMenu = menubar.addMenu(vtText.APPLICATION)
+        
+        for appDetail in self.appDetailList:
+            # 如果没有应用界面，则不添加菜单按钮
+            if not appDetail['appWidget']:
+                continue
+            
+            function = self.createOpenAppFunction(appDetail)
+            action = self.createAction(appDetail['appDisplayName'], function, loadIconPath(appDetail['appIco']))
+            appMenu.addAction(action)
+        
+        # 帮助
+        helpMenu = menubar.addMenu(vtText.HELP)
+        helpMenu.addAction(self.createAction(vtText.CONTRACT_SEARCH, self.openContract, loadIconPath('contract.ico')))
+        helpMenu.addAction(self.createAction(vtText.EDIT_SETTING, self.openSettingEditor, loadIconPath('editor.ico')))
+        helpMenu.addSeparator()
+        helpMenu.addAction(self.createAction(vtText.RESTORE, self.restoreWindow, loadIconPath('restore.ico')))
+        helpMenu.addAction(self.createAction(vtText.ABOUT, self.openAbout, loadIconPath('about.ico')))
+        helpMenu.addSeparator()
+        helpMenu.addAction(self.createAction(vtText.TEST, self.test, loadIconPath('test.ico')))
+    
+    #----------------------------------------------------------------------
+    def initStatusBar(self):
+        """初始化状态栏"""
+        self.statusLabel = QtWidgets.QLabel()
+        self.statusLabel.setAlignment(QtCore.Qt.AlignLeft)
+        
+        self.statusBar().addPermanentWidget(self.statusLabel)
+        self.statusLabel.setText(self.getCpuMemory())
+        
+        self.sbCount = 0
+        self.sbTrigger = 10     # 10秒刷新一次
+        self.signalStatusBar.connect(self.updateStatusBar)
+        self.eventEngine.register(EVENT_TIMER, self.signalStatusBar.emit)
+        
+    #----------------------------------------------------------------------
+    def updateStatusBar(self, event):
+        """在状态栏更新CPU和内存信息"""
+        self.sbCount += 1
+        
+        if self.sbCount == self.sbTrigger:
+            self.sbCount = 0
+            self.statusLabel.setText(self.getCpuMemory())
+    
+    #----------------------------------------------------------------------
+    def getCpuMemory(self):
+        """获取CPU和内存状态信息"""
+        cpuPercent = psutil.cpu_percent()
+        memoryPercent = psutil.virtual_memory().percent
+        return vtText.CPU_MEMORY_INFO.format(cpu=cpuPercent, memory=memoryPercent)
+        
+    #----------------------------------------------------------------------
+    def addConnectAction(self, menu, gatewayName, displayName=''):
+        """增加连接功能"""
+        if gatewayName not in self.gatewayNameList:
+            return
+        
+        def connect():
+            self.mainEngine.connect(gatewayName)
+            
+        if not displayName:
+            displayName = gatewayName
+        
+        actionName = vtText.CONNECT + displayName
+        connectAction = self.createAction(actionName, connect, 
+                                          loadIconPath('connect.ico'))
+        menu.addAction(connectAction)
+        
+    #----------------------------------------------------------------------
+    def createAction(self, actionName, function, iconPath=''):
+        """创建操作功能"""
+        action = QtWidgets.QAction(actionName, self)
+        action.triggered.connect(function)
+        
+        if iconPath:
+            icon = QtGui.QIcon(iconPath)
+            action.setIcon(icon)
+            
+        return action
+    
+    #----------------------------------------------------------------------
+    def createOpenAppFunction(self, appDetail):
+        """创建打开应用UI的函数"""
+        def openAppFunction():
+            appName = appDetail['appName']
+            try:
+                self.widgetDict[appName].show()
+            except KeyError:
+                appEngine = self.mainEngine.getApp(appName)
+                self.widgetDict[appName] = appDetail['appWidget'](appEngine, self.eventEngine)
+                self.widgetDict[appName].show()
+                
+        return openAppFunction
+        
+    #----------------------------------------------------------------------
+    def test(self):
+        """测试按钮用的函数"""
+        # 有需要使用手动触发的测试函数可以写在这里
+        pass
+
+    #----------------------------------------------------------------------
+    def openAbout(self):
+        """打开关于"""
+        try:
+            self.widgetDict['aboutW'].show()
+        except KeyError:
+            self.widgetDict['aboutW'] = AboutWidget(self)
+            self.widgetDict['aboutW'].show()
+    
+    #----------------------------------------------------------------------
+    def openContract(self):
+        """打开合约查询"""
+        try:
+            self.widgetDict['contractM'].show()
+        except KeyError:
+            self.widgetDict['contractM'] = ContractManager(self.mainEngine)
+            self.widgetDict['contractM'].show()
+            
+    #----------------------------------------------------------------------
+    def openSettingEditor(self):
+        """打开配置编辑"""
+        try:
+            self.widgetDict['settingEditor'].show()
+        except KeyError:
+            self.widgetDict['settingEditor'] = SettingEditor(self.mainEngine)
+            self.widgetDict['settingEditor'].show()    
+    
+    #----------------------------------------------------------------------
+    def closeEvent(self, event):
+        """关闭事件"""
+        reply = QtWidgets.QMessageBox.question(self, vtText.EXIT,
+                                           vtText.CONFIRM_EXIT, QtWidgets.QMessageBox.Yes | 
+                                           QtWidgets.QMessageBox.No, QtWidgets.QMessageBox.No)
+
+        if reply == QtWidgets.QMessageBox.Yes: 
+            for widget in self.widgetDict.values():
+                widget.close()
+            self.saveWindowSettings('custom')
+            
+            self.mainEngine.exit()
+            event.accept()
+        else:
+            event.ignore()
+            
+    #----------------------------------------------------------------------
+    def createDock(self, widgetClass, widgetName, widgetArea):
+        """创建停靠组件"""
+        widget = widgetClass(self.mainEngine, self.eventEngine)
+        dock = QtWidgets.QDockWidget(widgetName)
+        dock.setWidget(widget)
+        dock.setObjectName(widgetName)
+        dock.setFeatures(dock.DockWidgetFloatable|dock.DockWidgetMovable)
+        self.addDockWidget(widgetArea, dock)
+        return widget, dock
+    
+    #----------------------------------------------------------------------
+    def saveWindowSettings(self, settingName):
+        """保存窗口设置"""
+        settings = QtCore.QSettings('vn.trader', settingName)
+        settings.setValue('state', self.saveState())
+        settings.setValue('geometry', self.saveGeometry())
+        
+    #----------------------------------------------------------------------
+    def loadWindowSettings(self, settingName):
+        """载入窗口设置"""
+        settings = QtCore.QSettings('vn.trader', settingName)           
+        state = settings.value('state')
+        geometry = settings.value('geometry')
+        
+        # 尚未初始化
+        if state is None:
+            return
+        # 老版PyQt
+        elif isinstance(state, QtCore.QVariant):
+            self.restoreState(state.toByteArray())
+            self.restoreGeometry(geometry.toByteArray())
+        # 新版PyQt
+        elif isinstance(state, QtCore.QByteArray):
+            self.restoreState(state)
+            self.restoreGeometry(geometry)
+        # 异常
+        else:
+            content = u'载入窗口配置异常，请检查'
+            self.mainEngine.writeLog(content)
+        
+    #----------------------------------------------------------------------
+    def restoreWindow(self):
+        """还原默认窗口设置（还原停靠组件位置）"""
+        self.loadWindowSettings('default')
+        self.showMaximized()
+
+
+########################################################################
+class AboutWidget(QtWidgets.QDialog):
+    """显示关于信息"""
+
+    #----------------------------------------------------------------------
+    def __init__(self, parent=None):
+        """Constructor"""
+        super(AboutWidget, self).__init__(parent)
+
+        self.initUi()
+
+    #----------------------------------------------------------------------
+    def initUi(self):
+        """"""
+        self.setWindowTitle(vtText.ABOUT + 'VnTrader')
+
+        text = u"""
+            Developed by Traders, for Traders.
+
+            License：MIT
+            
+            Website：www.vnpy.org
+
+            Github：www.github.com/vnpy/vnpy
+            
+            """
+
+        label = QtWidgets.QLabel()
+        label.setText(text)
+        label.setMinimumWidth(500)
+
+        vbox = QtWidgets.QVBoxLayout()
+        vbox.addWidget(label)
+
+        self.setLayout(vbox)
+