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.CloseEditControl()

    73

        grid.BeginBatch()

    74

        for current, new, delmsg, addmsg in [

    75

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

    76

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

    77

        ]:

    78

            if new < current:

    79

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

    80

                grid.ProcessTableMessage(msg)

    81

            elif new > current:

    82

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

    83

                grid.ProcessTableMessage(msg)

    84

                self.UpdateValues(grid)

    85

        grid.EndBatch()

    86

    87

        self._rows = self.GetNumberRows()

    88

        self._cols = self.GetNumberCols()

    89

        # update the column rendering scheme

    90

        self._updateColAttrs(grid)

    91

    92

        # update the scrollbars and the displayed part of the grid

    93

        grid.AdjustScrollbars()

    94

        grid.ForceRefresh()

    95

    96

    def UpdateValues(self, grid):

    97

        """Update all displayed values"""

    98

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

    99

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

   100

        grid.ProcessTableMessage(msg)

   101

   102

    def _updateColAttrs(self, grid):

   103

        """

   104

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

   105

        appropriate renderer given the column name.

   106

   107

        Otherwise default to the default renderer.

   108

        """

   109

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

   110

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

   111

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

   112

                colname = self.GetColLabelValue(col, False)

   113

   114

                grid.SetReadOnly(row, col, True)

   115

                grid.SetCellEditor(row, col, None)

   116

                grid.SetCellRenderer(row, col, None)

   117

   118

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

   119

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

   120

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

   121

            self.ResizeRow(grid, row)

   122

   123

    def ResizeRow(self, grid, row):

   124

        if wx.Platform == '__WXMSW__':

   125

            grid.SetRowMinimalHeight(row, 20)

   126

        else:

   127

            grid.SetRowMinimalHeight(row, 28)

   128

        grid.AutoSizeRow(row, False)

   129

   130

    def SetData(self, data):

   131

        self.data = data

   132

   133

    def GetData(self):

   134

        return self.data

   135

   136

    def GetCurrentIndex(self):

   137

        return self.CurrentIndex

   138

   139

    def SetCurrentIndex(self, index):

   140

        self.CurrentIndex = index

   141

   142

    def AppendRow(self, row_content):

   143

        self.data.append(row_content)

   144

   145

    def InsertRow(self, index, row_content):

   146

        self.data.insert(index, row_content)

   147

   148

    def MoveRow(self, row_index, move):

   149

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

   150

        if new_index != row_index:

   151

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

   152

        return new_index

   153

   154

    def RemoveRow(self, row_index):

   155

        self.data.pop(row_index)

   156

   157

    def GetRow(self, row_index):

   158

        return self.data[row_index]

   159

   160

    def Empty(self):

   161

        self.data = []

   162

   163

    def AddHighlight(self, infos, highlight_type):

   164

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

   165

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

   166

        col_highlights.append(highlight_type)

   167

   168

    def ClearHighlights(self, highlight_type=None):

   169

        if highlight_type is None:

   170

            self.Highlights = {}

   171

        else:

   172

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

   173

                row_items = row_highlights.items()

   174

                for col, col_highlights in row_items:

   175

                    if highlight_type in col_highlights:

   176

                        col_highlights.remove(highlight_type)

   177

                    if len(col_highlights) == 0:

   178

                        row_highlights.pop(col)