339 lines
14 KiB
Python
339 lines
14 KiB
Python
|
# encoding: UTF-8
|
|||
|
|
|
|||
|
|
import psutil
|
|||
|
|
|
|||
|
|
# from gateway import GATEWAY_DICT
|
|||
|
|
from uiBasicWidget import *
|
|||
|
|
from ctaStrategy.uiCtaWidget import CtaEngineManager
|
|||
|
|
from dataRecorder.uiDrWidget import DrEngineManager
|
|||
|
|
from riskManager.uiRmWidget import RmEngineManager
|
|||
|
|
|
|||
|
|
|
|||
|
|
########################################################################
|
|||
|
|
class MainWindow(QtGui.QMainWindow):
|
|||
|
|
"""主窗口"""
|
|||
|
|
signalStatusBar = QtCore.pyqtSignal(type(Event()))
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def __init__(self, mainEngine, eventEngine):
|
|||
|
|
"""Constructor"""
|
|||
|
|
super(MainWindow, self).__init__()
|
|||
|
|
|
|||
|
|
self.mainEngine = mainEngine
|
|||
|
|
self.eventEngine = eventEngine
|
|||
|
|
|
|||
|
|
self.widgetDict = {} # 用来保存子窗口的字典
|
|||
|
|
|
|||
|
|
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)
|
|||
|
|
widgetErrorM, dockErrorM = self.createDock(ErrorMonitor, vtText.ERROR, QtCore.Qt.BottomDockWidgetArea)
|
|||
|
|
widgetTradeM, dockTradeM = self.createDock(TradeMonitor, vtText.TRADE, QtCore.Qt.BottomDockWidgetArea)
|
|||
|
|
widgetOrderM, dockOrderM = self.createDock(OrderMonitor, vtText.ORDER, QtCore.Qt.RightDockWidgetArea)
|
|||
|
|
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(dockTradeM, dockErrorM)
|
|||
|
|
self.tabifyDockWidget(dockTradeM, dockLogM)
|
|||
|
|
self.tabifyDockWidget(dockPositionM, dockAccountM)
|
|||
|
|
|
|||
|
|
dockTradeM.raise_()
|
|||
|
|
dockPositionM.raise_()
|
|||
|
|
|
|||
|
|
# 连接组件之间的信号
|
|||
|
|
widgetPositionM.itemDoubleClicked.connect(widgetTradingW.closePosition)
|
|||
|
|
|
|||
|
|
# 保存默认设置
|
|||
|
|
self.saveWindowSettings('default')
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def initMenu(self):
|
|||
|
|
"""初始化菜单"""
|
|||
|
|
# 创建菜单
|
|||
|
|
menubar = self.menuBar()
|
|||
|
|
|
|||
|
|
# 设计为只显示存在的接口
|
|||
|
|
sysMenu = menubar.addMenu(vtText.SYSTEM)
|
|||
|
|
|
|||
|
|
# 鉴于前后端分离，此处不应当直接检查本地的gateway接口可用情况，而是应当向服务端查询
|
|||
|
|
GATEWAY_TYPES = [
|
|||
|
|
GATEWAYTYPE_FUTURES,
|
|||
|
|
GATEWAYTYPE_EQUITY,
|
|||
|
|
GATEWAYTYPE_INTERNATIONAL,
|
|||
|
|
GATEWAYTYPE_BTC,
|
|||
|
|
GATEWAYTYPE_DATA,
|
|||
|
|
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 获取可链接的交易接口
|
|||
|
|
gateways = self.mainEngine.getGateway4sysMenu()
|
|||
|
|
|
|||
|
|
# 模拟一个GateWay 实例
|
|||
|
|
for g in gateways:
|
|||
|
|
sysMenu.addSeparator()
|
|||
|
|
if g['gatewayType'] in GATEWAY_TYPES:
|
|||
|
|
self.addConnectAction(sysMenu, g['gatewayName'],
|
|||
|
|
g['gatewayDisplayName'])
|
|||
|
|
|
|||
|
|
# for gatewayModule in GATEWAY_DICT.values():
|
|||
|
|
# if gatewayModule.gatewayType == GATEWAYTYPE_FUTURES:
|
|||
|
|
# self.addConnectAction(sysMenu, gatewayModule.gatewayName,
|
|||
|
|
# gatewayModule.gatewayDisplayName)
|
|||
|
|
#
|
|||
|
|
# sysMenu.addSeparator()
|
|||
|
|
# for gatewayModule in GATEWAY_DICT.values():
|
|||
|
|
# if gatewayModule.gatewayType == GATEWAYTYPE_EQUITY:
|
|||
|
|
# self.addConnectAction(sysMenu, gatewayModule.gatewayName,
|
|||
|
|
# gatewayModule.gatewayDisplayName)
|
|||
|
|
#
|
|||
|
|
# sysMenu.addSeparator()
|
|||
|
|
# for gatewayModule in GATEWAY_DICT.values():
|
|||
|
|
# if gatewayModule.gatewayType == GATEWAYTYPE_INTERNATIONAL:
|
|||
|
|
# self.addConnectAction(sysMenu, gatewayModule.gatewayName,
|
|||
|
|
# gatewayModule.gatewayDisplayName)
|
|||
|
|
#
|
|||
|
|
# sysMenu.addSeparator()
|
|||
|
|
# for gatewayModule in GATEWAY_DICT.values():
|
|||
|
|
# if gatewayModule.gatewayType == GATEWAYTYPE_BTC:
|
|||
|
|
# self.addConnectAction(sysMenu, gatewayModule.gatewayName,
|
|||
|
|
# gatewayModule.gatewayDisplayName)
|
|||
|
|
#
|
|||
|
|
# sysMenu.addSeparator()
|
|||
|
|
# for gatewayModule in GATEWAY_DICT.values():
|
|||
|
|
# if gatewayModule.gatewayType == GATEWAYTYPE_DATA:
|
|||
|
|
# self.addConnectAction(sysMenu, gatewayModule.gatewayName,
|
|||
|
|
# gatewayModule.gatewayDisplayName)
|
|||
|
|
|
|||
|
|
sysMenu.addSeparator()
|
|||
|
|
sysMenu.addAction(self.createAction(vtText.CONNECT_DATABASE, self.mainEngine.dbConnect))
|
|||
|
|
sysMenu.addSeparator()
|
|||
|
|
sysMenu.addAction(self.createAction(vtText.EXIT, self.close))
|
|||
|
|
|
|||
|
|
# 功能应用
|
|||
|
|
functionMenu = menubar.addMenu(vtText.APPLICATION)
|
|||
|
|
functionMenu.addAction(self.createAction(vtText.CONTRACT_SEARCH, self.openContract))
|
|||
|
|
functionMenu.addAction(self.createAction(vtText.DATA_RECORDER, self.openDr))
|
|||
|
|
functionMenu.addAction(self.createAction(vtText.RISK_MANAGER, self.openRm))
|
|||
|
|
|
|||
|
|
# 算法相关
|
|||
|
|
strategyMenu = menubar.addMenu(vtText.STRATEGY)
|
|||
|
|
strategyMenu.addAction(self.createAction(vtText.CTA_STRATEGY, self.openCta))
|
|||
|
|
|
|||
|
|
# 帮助
|
|||
|
|
helpMenu = menubar.addMenu(vtText.HELP)
|
|||
|
|
helpMenu.addAction(self.createAction(vtText.RESTORE, self.restoreWindow))
|
|||
|
|
helpMenu.addAction(self.createAction(vtText.ABOUT, self.openAbout))
|
|||
|
|
helpMenu.addAction(self.createAction(vtText.TEST, self.test))
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def initStatusBar(self):
|
|||
|
|
"""初始化状态栏"""
|
|||
|
|
self.statusLabel = QtGui.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.mainEngine.getAllGatewayNames():
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
def connect():
|
|||
|
|
self.mainEngine.connect(gatewayName)
|
|||
|
|
|
|||
|
|
if not displayName:
|
|||
|
|
displayName = gatewayName
|
|||
|
|
displayName = gatewayName
|
|||
|
|
actionName = vtText.CONNECT + displayName
|
|||
|
|
|
|||
|
|
menu.addAction(self.createAction(actionName, connect))
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def createAction(self, actionName, function):
|
|||
|
|
"""创建操作功能"""
|
|||
|
|
action = QtGui.QAction(actionName, self)
|
|||
|
|
action.triggered.connect(function)
|
|||
|
|
return action
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
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 openCta(self):
|
|||
|
|
"""打开CTA组件"""
|
|||
|
|
try:
|
|||
|
|
self.widgetDict['ctaM'].showMaximized()
|
|||
|
|
except KeyError:
|
|||
|
|
self.widgetDict['ctaM'] = CtaEngineManager(self.mainEngine.ctaEngine, self.eventEngine)
|
|||
|
|
self.widgetDict['ctaM'].showMaximized()
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def openDr(self):
|
|||
|
|
"""打开行情数据记录组件"""
|
|||
|
|
try:
|
|||
|
|
self.widgetDict['drM'].showMaximized()
|
|||
|
|
except KeyError:
|
|||
|
|
self.widgetDict['drM'] = DrEngineManager(self.mainEngine.drEngine, self.eventEngine)
|
|||
|
|
self.widgetDict['drM'].showMaximized()
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def openRm(self):
|
|||
|
|
"""打开组件"""
|
|||
|
|
try:
|
|||
|
|
self.widgetDict['rmM'].show()
|
|||
|
|
except KeyError:
|
|||
|
|
self.widgetDict['rmM'] = RmEngineManager(self.mainEngine.rmEngine, self.eventEngine)
|
|||
|
|
self.widgetDict['rmM'].show()
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def closeEvent(self, event):
|
|||
|
|
"""关闭事件"""
|
|||
|
|
reply = QtGui.QMessageBox.question(self, vtText.EXIT,
|
|||
|
|
vtText.CONFIRM_EXIT, QtGui.QMessageBox.Yes |
|
|||
|
|
QtGui.QMessageBox.No, QtGui.QMessageBox.No)
|
|||
|
|
|
|||
|
|
if reply == QtGui.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 = QtGui.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)
|
|||
|
|
# 这里由于PyQt4的版本不同，settings.value('state')调用返回的结果可能是：
|
|||
|
|
# 1. None（初次调用，注册表里无相应记录，因此为空）
|
|||
|
|
# 2. QByteArray（比较新的PyQt4）
|
|||
|
|
# 3. QVariant（以下代码正确执行所需的返回结果）
|
|||
|
|
# 所以为了兼容考虑，这里加了一个try...except，如果是1、2的情况就pass
|
|||
|
|
# 可能导致主界面的设置无法载入（每次退出时的保存其实是成功了）
|
|||
|
|
try:
|
|||
|
|
self.restoreState(settings.value('state').toByteArray())
|
|||
|
|
self.restoreGeometry(settings.value('geometry').toByteArray())
|
|||
|
|
except AttributeError:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
#----------------------------------------------------------------------
|
|||
|
|
def restoreWindow(self):
|
|||
|
|
"""还原默认窗口设置（还原停靠组件位置）"""
|
|||
|
|
self.loadWindowSettings('default')
|
|||
|
|
self.showMaximized()
|
|||
|
|
|
|||
|
|
|
|||
|
|
########################################################################
|
|||
|
|
class AboutWidget(QtGui.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 = QtGui.QLabel()
|
|||
|
|
label.setText(text)
|
|||
|
|
label.setMinimumWidth(500)
|
|||
|
|
|
|||
|
|
vbox = QtGui.QVBoxLayout()
|
|||
|
|
vbox.addWidget(label)
|
|||
|
|
|
|||
|
|
self.setLayout(vbox)
|
|||
|
|
|