59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
# encoding: UTF-8
|
|
|
|
'''
|
|
行情记录模块相关的GUI控制组件
|
|
'''
|
|
from PyQt4.QtCore import QDir
|
|
from PyQt4.QtGui import QFileSystemModel
|
|
from PyQt4.QtGui import QTreeView
|
|
|
|
from uiBasicWidget import QtGui, QtCore
|
|
from eventEngine import *
|
|
|
|
from eventType import EVENT_CONTRACT
|
|
|
|
|
|
class DrEditWidget(QtGui.QWidget):
|
|
"""行情数据记录引擎管理组件"""
|
|
signal = QtCore.pyqtSignal(type(Event()))
|
|
|
|
def __init__(self, drEngine, eventEngine, parent=None):
|
|
"""Constructor"""
|
|
super(DrEditWidget, self).__init__(parent)
|
|
|
|
self.drEngine = drEngine
|
|
self.eventEngine = eventEngine
|
|
|
|
# 保存合约详细信息的字典
|
|
self.contractDict = {}
|
|
|
|
self.initUi()
|
|
self.updateSetting()
|
|
self.registerEvent()
|
|
|
|
def initUi(self):
|
|
self.qTreeView = QTreeView()
|
|
model = QFileSystemModel()
|
|
model.setRootPath(QDir.currentPath())
|
|
self.qTreeView.setModel(model)
|
|
self.qTreeView.setRootIndex(model.index(QDir.currentPath()))
|
|
|
|
vbox = QtGui.QVBoxLayout()
|
|
vbox.addWidget(self.qTreeView)
|
|
self.setLayout(vbox)
|
|
|
|
def updateSetting(self):
|
|
pass
|
|
|
|
def updateContract(self, event):
|
|
"""更新合约数据"""
|
|
contract = event.dict_['data']
|
|
self.contractDict[contract.vtSymbol] = contract
|
|
self.contractDict[contract.symbol] = contract # 使用常规代码（不包括交易所）可能导致重复
|
|
|
|
# ----------------------------------------------------------------------
|
|
def registerEvent(self):
|
|
"""注册事件监听"""
|
|
self.signal.connect(self.updateContract)
|
|
self.eventEngine.register(EVENT_CONTRACT, self.signal.emit)
|