# encoding: UTF-8 import psutil from uiBasicWidget import * from uiCtaWidget import CtaEngineManager ######################################################################## class MainWindow(QtGui.QMainWindow): """主窗口""" # ---------------------------------------------------------------------- def __init__(self, mainEngine, eventEngine, dataEngine): """Constructor""" super(MainWindow, self).__init__() self.mainEngine = mainEngine self.eventEngine = eventEngine self.dataEngine = dataEngine self.widgetDict = {} # 用来保存子窗口的字典 self.initUi() # ---------------------------------------------------------------------- def initUi(self): """初始化界面""" self.setWindowTitle('VnTrader') self.initCentral() self.initMenu() self.initStatusBar() # ---------------------------------------------------------------------- def initCentral(self): """初始化中心区域""" marketM = MarketMonitor(self.eventEngine) logM = LogMonitor(self.eventEngine) errorM = ErrorMonitor(self.eventEngine) tradeM = TradeMonitor(self.eventEngine) orderM = OrderMonitor(self.eventEngine, self.mainEngine) positionM = PositionMonitor(self.eventEngine) accountM = AccountMonitor(self.eventEngine) tradingW = TradingWidget(self.mainEngine, self.mainEngine.eventEngine, self.mainEngine.dataEngine) leftTab = QtGui.QTabWidget() leftTab.addTab(logM, u'日志') leftTab.addTab(errorM, u'错误') leftTab.addTab(accountM, u'账户') rightTab = QtGui.QTabWidget() rightTab.addTab(tradeM, u'成交') rightTab.addTab(orderM, u'委托') rightTab.addTab(positionM, u'持仓') hbox = QtGui.QHBoxLayout() hbox.addWidget(tradingW) hbox.addWidget(marketM) grid = QtGui.QGridLayout() grid.addLayout(hbox, 0, 0, 1, 2) grid.addWidget(leftTab, 1, 0) grid.addWidget(rightTab, 1, 1) central = QtGui.QWidget() central.setLayout(grid) self.setCentralWidget(central) # ---------------------------------------------------------------------- def initMenu(self): """初始化菜单""" # 创建操作 connectCtpProdAction = QtGui.QAction(u'连接生产CTP', self) connectCtpProdAction.triggered.connect(self.connectCtpProd) connectCtpPostAction = QtGui.QAction(u'连接盘后CTP', self) connectCtpPostAction.triggered.connect(self.connectCtpPost) connectCtpTestAction = QtGui.QAction(u'连接测试CTP', self) connectCtpTestAction.triggered.connect(self.connectCtpTest) #connectCtpAction = QtGui.QAction(u'连接CTP', self) #connectCtpAction.triggered.connect(self.connectCtp) connectLtsAction = QtGui.QAction(u'连接LTS', self) connectLtsAction.triggered.connect(self.connectLts) connectWindAction = QtGui.QAction(u'连接Wind', self) connectWindAction.triggered.connect(self.connectWind) connectIbAction = QtGui.QAction(u'连接IB', self) connectIbAction.triggered.connect(self.connectIb) connectDbAction = QtGui.QAction(u'连接数据库', self) connectDbAction.triggered.connect(self.mainEngine.dbConnect) testAction = QtGui.QAction(u'测试', self) testAction.triggered.connect(self.testSubscribe) exitAction = QtGui.QAction(u'退出', self) exitAction.triggered.connect(self.close) aboutAction = QtGui.QAction(u'关于', self) aboutAction.triggered.connect(self.openAbout) contractAction = QtGui.QAction(u'查询合约', self) contractAction.triggered.connect(self.openContract) ctaAction = QtGui.QAction(u'CTA策略', self) ctaAction.triggered.connect(self.openCta) # 创建菜单 menubar = self.menuBar() sysMenu = menubar.addMenu(u'系统') sysMenu.addAction(connectCtpProdAction) sysMenu.addAction(connectCtpPostAction) sysMenu.addAction(connectCtpTestAction) #sysMenu.addAction(connectCtpAction) sysMenu.addAction(connectLtsAction) sysMenu.addAction(connectWindAction) sysMenu.addAction(connectIbAction) sysMenu.addSeparator() sysMenu.addAction(connectDbAction) sysMenu.addSeparator() sysMenu.addAction(testAction) sysMenu.addAction(exitAction) functionMenu = menubar.addMenu(u'功能') functionMenu.addAction(contractAction) # 算法相关 algoMenu = menubar.addMenu(u'算法') algoMenu.addAction(ctaAction) # 帮助 helpMenu = menubar.addMenu(u'帮助') helpMenu.addAction(aboutAction) # ---------------------------------------------------------------------- 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.eventEngine.register(EVENT_TIMER, self.updateStatusBar) # ---------------------------------------------------------------------- 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 u'CPU使用率：%d%% 内存使用率：%d%%' % (cpuPercent, memoryPercent) # ---------------------------------------------------------------------- def connectCtpProd(self): """连接生产环境CTP接口""" self.mainEngine.connect('CTP_Prod') # ---------------------------------------------------------------------- def connectCtpPost(self): """连接盘后CTP接口""" self.mainEngine.connect('CTP_Post') def connectCtpTest(self): """连接测试环境CTP接口""" self.mainEngine.connect('CTP_Test') def connectCtp(self): """连接CTP接口""" self.mainEngine.connect('CTP') # ---------------------------------------------------------------------- def connectLts(self): """连接LTS接口""" self.mainEngine.connect('LTS') # ---------------------------------------------------------------------- def connectWind(self): """连接Wind接口""" self.mainEngine.connect('Wind') # ---------------------------------------------------------------------- def connectIb(self): """连接Ib""" self.mainEngine.connect('IB') # ---------------------------------------------------------------------- def testSubscribe(self): """测试订阅""" api = self.mainEngine.gatewayDict['LTS'].qryApi api.reqID += 1 api.reqQryOFInstrument({}, api.reqID) #req = VtSubscribeReq() #req.symbol = 'GOOG' #req.productClass = PRODUCT_EQUITY #req.exchange = EXCHANGE_SMART #req.currency = CURRENCY_USD #self.mainEngine.subscribe(req, 'IB') #req.symbol = 'AAPL' #self.mainEngine.subscribe(req, 'IB') #req.symbol = 'YHOO' #self.mainEngine.subscribe(req, 'IB') #req.symbol = 'MSFT' #self.mainEngine.subscribe(req, 'IB') #req.symbol = 'GE' #self.mainEngine.subscribe(req, 'IB') # ---------------------------------------------------------------------- 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'] = ContractMonitor(self.mainEngine.dataEngine) self.widgetDict['contractM'].show() # ---------------------------------------------------------------------- def openCta(self): """打开CTA组件""" try: self.widgetDict['ctaM'].show() except KeyError: self.widgetDict['ctaM'] = CtaEngineManager(self.mainEngine.ctaEngine, self.eventEngine) self.widgetDict['ctaM'].show() # ---------------------------------------------------------------------- def closeEvent(self, event): """关闭事件""" reply = QtGui.QMessageBox.question(self, u'退出', u'确认退出?', QtGui.QMessageBox.Yes | QtGui.QMessageBox.No, QtGui.QMessageBox.No) if reply == QtGui.QMessageBox.Yes: for widget in self.widgetDict.values(): widget.close() self.mainEngine.exit() event.accept() else: event.ignore() ######################################################################## class AboutWidget(QtGui.QDialog): """显示关于信息""" # ---------------------------------------------------------------------- def __init__(self, parent=None): """Constructor""" super(AboutWidget, self).__init__(parent) self.initUi() # ---------------------------------------------------------------------- def initUi(self): """""" self.setWindowTitle(u'关于') text = u""" VnTrader 更新日期：2015/9/29 作者：用Python的交易员 License：MIT 主页：vnpy.org Github：github.com/vnpy/vnpy QQ交流群：262656087 开发环境 操作系统：Windows 7 专业版 64位 Python发行版：Python 2.7.6 (Anaconda 1.9.2 Win-32) CTP：vn.ctp 2015/6/1版 图形库：PyQt4 4.11.3 Py2.7-x32 事件驱动引擎：vn.event 开发环境：WingIDE 5.0.6 """ label = QtGui.QLabel() label.setText(text) label.setMinimumWidth(500) vbox = QtGui.QVBoxLayout() vbox.addWidget(label) self.setLayout(vbox)