investGridBase.py

# -*- coding: utf-8 -*-

import wx
import wx.grid as gridlib

class InvestGridBase(gridlib.Grid):
    class Const:
        FONT_SIZE_CONTENT = 14
        FONT_NAME = "微软雅黑"

    FONT_SIZE_SEPARATOR = 12
    COLOR_LIGHT_YELLOW = wx.Colour(255, 255, 180)
    COLOR_BACKGROUND_LIGHT_GRAY = wx.Colour(245, 245, 245)
    COLOR_DARK_GREEN = wx.Colour(0, 128, 64)
    FLOAT_FIELD_WIDTH = 2

    def __init__(self, parent):
        gridlib.Grid.__init__(self, parent, -1)
        self.EnableEditing(False)
        self.EnableDragRowSize(False)
        self.setGridFormat = self.createSetGridFormat()

    def GetColorByUpDown(self, number):
        if (number > 0.0):
            return self.COLOR_DARK_GREEN
        return wx.RED

    def SetCellNumberAndColor(self, row, col, argNumber, fieldWidth = -1):
        textNumber = ''
        floatNumber = 0.0
        if (type(argNumber) is str):
            textNumber = argNumber
            if (argNumber.find('%') != -1):
                argNumber = argNumber.strip('%')
            floatNumber = float(argNumber)
        else:
            floatNumber = argNumber

            if (fieldWidth == -1): # set field with
                fieldWidth = self.FLOAT_FIELD_WIDTH
            textNumber = "{0:.{1}f}".format(floatNumber, fieldWidth)

        if (floatNumber == 0.0):
            return

        self.SetCellValue(row, col, textNumber)
        self.SetCellTextColour(row, col, self.GetColorByUpDown(floatNumber))

    def setCellFont(self, row, fontColor = wx.BLACK, fontSize = Const.FONT_SIZE_CONTENT, fontName = Const.FONT_NAME):
        # 设置字体格式
        attr = wx.grid.GridCellAttr()
        attr.SetTextColour(fontColor)
        font = wx.Font(fontSize, wx.DEFAULT, wx.NORMAL, wx.NORMAL, False, fontName)
        attr.SetFont(font)
        self.SetRowAttr(row, attr)

    # trick: return inner class via a function to pass the outer class instance
    def createSetGridFormat(self):
        gridObj = self
        class SetGridFormat():
            def __init__(self):
                self.toggleBackgroundColor = True

            def finishUpdate(self):
                gridObj.AutoSizeColumns(setAsMin=True)

                # ignore performance; always update
                self.toggleBackgroundColor = True

            def setFormat(self, row):
                # 设置字体格式
                gridObj.setCellFont(row, wx.BLUE)

                for col in range(len(gridObj.Columns)):
                    horizAlign = wx.ALIGN_RIGHT
                    if (col == gridObj.COLUMN.NAME):
                        horizAlign = wx.ALIGN_LEFT
                    gridObj.SetCellAlignment(row, col, horizAlign, wx.ALIGN_CENTRE)

                backgroundColor = gridObj.GetDefaultCellBackgroundColour()
                if self.toggleBackgroundColor:
                    backgroundColor = gridObj.COLOR_BACKGROUND_LIGHT_GRAY
                self.toggleBackgroundColor = not self.toggleBackgroundColor # toggle the background color

                for col in range(len(gridObj.Columns)):
                    gridObj.SetCellBackgroundColour(row, col, backgroundColor)

        return SetGridFormat()