1

#This library is free software; you can redistribute it and/or

     2

#modify it under the terms of the GNU General Public

     3

#License as published by the Free Software Foundation; either

     4

#version 2.1 of the License, or (at your option) any later version.

     5

#

     6

#This library is distributed in the hope that it will be useful,

     7

#but WITHOUT ANY WARRANTY; without even the implied warranty of

     8

#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

     9

#General Public License for more details.

    10

#

    11

#You should have received a copy of the GNU General Public

    12

#License along with this library; if not, write to the Free Software

    13

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    14

    15

import wx

    16

import wx.grid

    17

    18

class CustomTable(wx.grid.PyGridTableBase):

    19

    20

    """

    21

    A custom wx.grid.Grid Table using user supplied data

    22

    """

    23

    def __init__(self, parent, data, colnames):

    24

        # The base class must be initialized *first*

    25

        wx.grid.PyGridTableBase.__init__(self)

    26

        self.data = data

    27

        self.colnames = colnames

    28

        self.Highlights = {}

    29

        self.Parent = parent

    30

        # XXX

    31

        # we need to store the row length and collength to

    32

        # see if the table has changed size

    33

        self._rows = self.GetNumberRows()

    34

        self._cols = self.GetNumberCols()

    35

    36

    def GetNumberCols(self):

    37

        return len(self.colnames)

    38

    39

    def GetNumberRows(self):

    40

        return len(self.data)

    41

    42

    def GetColLabelValue(self, col, translate=True):

    43

        if col < len(self.colnames):

    44

            if translate:

    45

                return _(self.colnames[col])

    46

            return self.colnames[col]

    47

    48

    def GetRowLabelValue(self, row, translate=True):

    49

        return row

    50

    51

    def GetValue(self, row, col):

    52

        if row < self.GetNumberRows():

    53

            return self.data[row].get(self.GetColLabelValue(col, False), "")

    54

    55

    def SetValue(self, row, col, value):

    56

        if col < len(self.colnames):

    57

            self.data[row][self.GetColLabelValue(col, False)] = value

    58

    59

    def GetValueByName(self, row, colname):

    60

        if row < self.GetNumberRows():

    61

            return self.data[row].get(colname)

    62

    63

    def SetValueByName(self, row, colname, value):

    64

        if row < self.GetNumberRows():

    65

            self.data[row][colname] = value

    66

    67

    def ResetView(self, grid):

    68

        """

    69

        (wx.grid.Grid) -> Reset the grid view.   Call this to

    70

        update the grid if rows and columns have been added or deleted

    71

        """

    72

        grid.BeginBatch()

    73

        for current, new, delmsg, addmsg in [

    74

            (self._rows, self.GetNumberRows(), wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED),

    75

            (self._cols, self.GetNumberCols(), wx.grid.GRIDTABLE_NOTIFY_COLS_DELETED, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED),

    76

        ]:

    77

            if new < current:

    78

                msg = wx.grid.GridTableMessage(self,delmsg,new,current-new)

    79

                grid.ProcessTableMessage(msg)

    80

            elif new > current:

    81

                msg = wx.grid.GridTableMessage(self,addmsg,new-current)

    82

                grid.ProcessTableMessage(msg)

    83

                self.UpdateValues(grid)

    84

        grid.EndBatch()

    85

    86

        self._rows = self.GetNumberRows()

    87

        self._cols = self.GetNumberCols()

    88

        # update the column rendering scheme

    89

        self._updateColAttrs(grid)

    90

    91

        # update the scrollbars and the displayed part of the grid

    92

        grid.AdjustScrollbars()

    93

        grid.ForceRefresh()

    94

    95

    def UpdateValues(self, grid):

    96

        """Update all displayed values"""

    97

        # This sends an event to the grid table to update all of the values

    98

        msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_REQUEST_VIEW_GET_VALUES)

    99

        grid.ProcessTableMessage(msg)

   100

   101

    def _updateColAttrs(self, grid):

   102

        """

   103

        wx.grid.Grid -> update the column attributes to add the

   104

        appropriate renderer given the column name.

   105

   106

        Otherwise default to the default renderer.

   107

        """

   108

        for row in range(self.GetNumberRows()):

   109

            row_highlights = self.Highlights.get(row, {})

   110

            for col in range(self.GetNumberCols()):

   111

                colname = self.GetColLabelValue(col, False)

   112

   113

                grid.SetReadOnly(row, col, True)

   114

                grid.SetCellEditor(row, col, None)

   115

                grid.SetCellRenderer(row, col, None)

   116

   117

                highlight_colours = row_highlights.get(colname.lower(), [(wx.WHITE, wx.BLACK)])[-1]

   118

                grid.SetCellBackgroundColour(row, col, highlight_colours[0])

   119

                grid.SetCellTextColour(row, col, highlight_colours[1])

   120

            self.ResizeRow(grid, row)

   121

   122

    def ResizeRow(self, grid, row):

   123

        if wx.Platform == '__WXMSW__':

   124

            grid.SetRowMinimalHeight(row, 20)

   125

        else:

   126

            grid.SetRowMinimalHeight(row, 28)

   127

        grid.AutoSizeRow(row, False)

   128

   129

    def SetData(self, data):

   130

        self.data = data

   131

   132

    def GetData(self):

   133

        return self.data

   134

   135

    def GetCurrentIndex(self):

   136

        return self.CurrentIndex

   137

   138

    def SetCurrentIndex(self, index):

   139

        self.CurrentIndex = index

   140

   141

    def AppendRow(self, row_content):

   142

        self.data.append(row_content)

   143

   144

    def InsertRow(self, index, row_content):

   145

        self.data.insert(index, row_content)

   146

   147

    def MoveRow(self, row_index, move):

   148

        new_index = max(0, min(row_index + move, len(self.data) - 1))

   149

        if new_index != row_index:

   150

            self.data.insert(new_index, self.data.pop(row_index))

   151

        return new_index

   152

   153

    def RemoveRow(self, row_index):

   154

        self.data.pop(row_index)

   155

   156

    def GetRow(self, row_index):

   157

        return self.data[row_index]

   158

   159

    def Empty(self):

   160

        self.data = []

   161

   162

    def AddHighlight(self, infos, highlight_type):

   163

        row_highlights = self.Highlights.setdefault(infos[0], {})

   164

        col_highlights = row_highlights.setdefault(infos[1], [])

   165

        col_highlights.append(highlight_type)

   166

   167

    def ClearHighlights(self, highlight_type=None):

   168

        if highlight_type is None:

   169

            self.Highlights = {}

   170

        else:

   171

            for row, row_highlights in self.Highlights.iteritems():

   172

                row_items = row_highlights.items()

   173

                for col, col_highlights in row_items:

   174

                    if highlight_type in col_highlights:

   175

                        col_highlights.remove(highlight_type)

   176

                    if len(col_highlights) == 0:

   177

                        row_highlights.pop(col)