vnpy/vn.trader/uiBasicWidget.py

# encoding: UTF-8

from collections import OrderedDict

from PyQt4 import QtGui, QtCore

from eventEngine import *
  

########################################################################
class BasicMonitor(QtGui.QTableWidget):
    """基础监控"""
    signal = QtCore.pyqtSignal(type(Event()))

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(BasicMonitor, self).__init__(parent)
        
        self.eventEngine = eventEngine
        
        # 保存表头标签用
        self.headerDict = OrderedDict()  # 有序字典，key是英文名，value是对应的中文名
        self.headerList = []             # 对应self.headerDict.keys()
        
        # 保存相关数据用
        self.dataDict = {}  # 字典，key是字段对应的数据，value是保存相关单元格的字典
        self.dataKey = ''   # 字典键对应的数据字段
        
        # 监控的事件类型
        self.eventType = ''
        
        # 字体
        self.font = None
        
    #----------------------------------------------------------------------
    def setHeaderDict(self, headerDict):
        """设置表头有序字典"""
        self.headerDict = headerDict
        self.headerList = headerDict.keys()
        
    #----------------------------------------------------------------------
    def setDataKey(self, dataKey):
        """设置数据字典的键"""
        self.dataKey = dataKey
        
    #----------------------------------------------------------------------
    def setEventType(self, eventType):
        """设置监控的事件类型"""
        self.eventType = eventType
        
    #----------------------------------------------------------------------
    def setFont(self, font):
        """设置字体"""
        self.font = font
        
    #----------------------------------------------------------------------
    def initTable(self):
        """初始化表格"""
        # 设置表格的列数
        col = len(self.labelDict)
        self.setColumnCount(col)
        
        # 设置列表头
        self.setHorizontalHeaderLabels(self.labelDict.values())
        
        # 关闭左边的垂直表头
        self.verticalHeader().setVisible(False)
        
        # 设为不可编辑
        self.setEditTriggers(self.NoEditTriggers)

    #----------------------------------------------------------------------
    def registerEvent(self):
        """注册GUI更新相关的事件监听"""
        self.signal.connect(self.updateData)
        self.eventEngine.register(self.eventType, self.signal.emit)
    
    #----------------------------------------------------------------------
    def updateData(self, event):
        """更新数据"""
        data = event.dict_['data']
        key = data.__getattribute__(self.dataKey)
        
        # 如果设置了dataKey，则采用存量更新模式
        if self.dataKey:
            # 如果键在数据字典中不存在，则先插入新的一行，并创建对应单元格
            if key not in self.dataDict:
                self.insertRow(0)     
                d = {}
                for n, header in enumerate(self.headerList):
                    content = unicode(data.__getattribute__(header))
                    cell = QtGui.QTableWidgetItem(content)
                    if self.font:
                        cell.setFont(font)  # 如果设置了特殊字体，则进行单元格设置
                    self.setItem(0, n, cell)
                    d[header] = cell
                self.dataDict[key] = d
            # 否则如果已经存在，则直接更新相关单元格
            else:
                d = self.dataDict[key]
                for header in self.headerList:
                    content = unicode(data.__getattribute__(header))
                    cell = d[header]
                    cell.setText(content)
        # 否则采用增量更新模式
        else:
            self.insertRow(0)  
            for n, header in enumerate(self.headerList):
                content = unicode(data.__getattribute__(header))
                cell = QtGui.QTableWidgetItem(content)
                if self.font:
                    cell.setFont(font)
                self.setItem(0, n, cell)            
                
        # 调整列宽
        self.resizeColumns()
    
    #----------------------------------------------------------------------
    def resizeColumns(self):
        """调整各列的大小"""
        self.horizontalHeader().resizeSections(QtGui.QHeaderView.ResizeToContents)    
    

########################################################################
class MarketMonitor(BasicMonitor):
    """市场监控组件"""

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(MarketMonitor, self).__init__(eventEngine, parent)
        
        # 设置表头有序字典
        d = OrderedDict()
        d['symbol'] = u'合约代码'
        d['vtSymbol'] = u'系统代码'
        d['lastPrice'] = u'最新价'
        d['volume'] = u'成交量'
        d['openInterest'] = u'持仓量'
        d['openPrice'] = u'开盘价'
        d['highPrice'] = u'最高价'
        d['lowPrice'] = u'最低价'
        d['bidPrice1'] = u'买一价'
        d['bidVolume1'] = u'买一量'
        d['askPrice1'] = u'卖一价'
        d['askVolume1'] = u'卖一量'
        d['tickTime'] = u'时间'
        self.setHeaderDict(d)
        
        # 设置数据键
        self.setDataKey('vtSymbol')
        
        # 设置监控事件类型
        self.setEventType(EVENT_TICK)
        
        # 初始化表格
        self.initTable()
        
        # 注册事件监听
        self.registerEvent()


########################################################################
class LogMonitor(BasicMonitor):
    """日志监控"""

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(LogMonitor, self).__init__(eventEngine, parent)
        
        d = OrderedDict()
        d['gatewayName'] = u'接口'
        d['logTime'] = u'时间'
        d['logContent'] = u'内容'
        self.setHeaderDict(d)
        
        self.setEventType(EVENT_LOG)
        self.initTable()
        self.registerEvent()


########################################################################
class ErrorMonitor(BasicMonitor):
    """错误监控"""

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(ErrorMonitor, self).__init__(eventEngine, parent)
        
        d = OrderedDict()
        d['gatewayName'] = u'接口'
        d['errorID'] = u'错误代码'
        d['errorMsg'] = u'错误信息'
        self.setHeaderDict(d)
        
        self.setEventType(EVENT_ERROR)
        self.initTable()
        self.registerEvent()


########################################################################
class TradeMonitor(BasicMonitor):
    """成交监控"""

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(TradeMonitor, self).__init__(eventEngine, parent)
        
        d = OrderedDict()
        d['gatewayName'] = u'接口'
        d['tradeID'] = u'成交编号'
        d['orderID'] = u'委托编号'
        d['symbol'] = u'合约代码'
        d['direction'] = u'方向'
        d['offset'] = u'开平'
        d['price'] = u'价格'
        d['volume'] = u'数量'
        d['tradeTime'] = u'成交时间'
        self.setHeaderDict(d)
        
        self.setEventType(EVENT_TRADE)
        self.initTable()
        self.registerEvent()


########################################################################
class OrderMonitor(BasicMonitor):
    """委托监控"""

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(OrderMonitor, self).__init__(eventEngine, parent)
        
        d = OrderedDict()
        d['gatewayName'] = u'接口'
        d['orderID'] = u'委托编号'
        d['symbol'] = u'合约代码'
        d['direction'] = u'方向'
        d['offset'] = u'开平'
        d['price'] = u'价格'
        d['totalVolume'] = u'委托数量'
        d['tradedVolume'] = u'成交数量'
        d['status'] = u'状态'
        d['orderTime'] = u'委托时间'
        d['tradeTime'] = u'撤销时间'
        d['frontID'] = u'前置编号'
        d['sessionID'] = u'会话编号'
        self.setHeaderDict(d)
        
        self.setDataKey('vtOrderID')
        self.setEventType(EVENT_ORDER)
        self.initTable()
        self.registerEvent()


########################################################################
class PositionMonitor(BasicMonitor):
    """持仓监控"""

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(PositionMonitor, self).__init__(eventEngine, parent)
        
        d = OrderedDict()
        d['gatewayName'] = u'接口'
        d['symbol'] = u'合约代码'
        d['direction'] = u'方向'
        d['position'] = u'持仓量'
        d['frozen'] = u'冻结量'
        d['price'] = u'价格'
        self.setHeaderDict(d)
        
        self.setDataKey('vtPositionName')
        self.setEventType(EVENT_POSITION)
        self.initTable()
        self.registerEvent()


########################################################################
class AccountMonitor(BasicMonitor):
    """账户监控"""

    #----------------------------------------------------------------------
    def __init__(self, eventEngine, parent=None):
        """Constructor"""
        super(AccountMonitor, self).__init__(eventEngine, parent)
        
        d = OrderedDict()
        d['gatewayName'] = u'接口'
        d['accountID'] = u'账户'
        d['preBalance'] = u'昨结'
        d['balance'] = u'净值'
        d['available'] = u'可用'
        d['commission'] = u'手续费'
        d['margin'] = u'保证金'
        d['closeProfit'] = u'平仓盈亏'
        d['positionProfit'] = u'持仓盈亏'
        self.setHeaderDict(d)
        
        self.setDataKey('vtAccountID')
        self.setEventType(EVENT_ACCOUNT)
        self.initTable()
        self.registerEvent()