Laurent@814: #!/usr/bin/env python Laurent@814: # -*- coding: utf-8 -*- Laurent@814: Laurent@814: #This library is free software; you can redistribute it and/or Laurent@814: #modify it under the terms of the GNU General Public Laurent@814: #License as published by the Free Software Foundation; either Laurent@814: #version 2.1 of the License, or (at your option) any later version. Laurent@814: # Laurent@814: #This library is distributed in the hope that it will be useful, Laurent@814: #but WITHOUT ANY WARRANTY; without even the implied warranty of Laurent@814: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Laurent@814: #General Public License for more details. Laurent@814: # Laurent@814: #You should have received a copy of the GNU General Public Laurent@814: #License along with this library; if not, write to the Free Software Laurent@814: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA Laurent@814: Laurent@814: import wx Laurent@814: import wx.grid Laurent@814: Laurent@814: class CustomTable(wx.grid.PyGridTableBase): Laurent@814: Laurent@814: """ Laurent@814: A custom wx.grid.Grid Table using user supplied data Laurent@814: """ Laurent@814: def __init__(self, parent, data, colnames): Laurent@814: # The base class must be initialized *first* Laurent@814: wx.grid.PyGridTableBase.__init__(self) Laurent@814: self.data = data Laurent@814: self.colnames = colnames Laurent@814: self.Highlights = {} Laurent@814: self.Parent = parent Laurent@814: # XXX Laurent@814: # we need to store the row length and collength to Laurent@814: # see if the table has changed size Laurent@814: self._rows = self.GetNumberRows() Laurent@814: self._cols = self.GetNumberCols() Laurent@814: Laurent@814: def GetNumberCols(self): Laurent@814: return len(self.colnames) Laurent@814: Laurent@814: def GetNumberRows(self): Laurent@814: return len(self.data) Laurent@814: Laurent@814: def GetColLabelValue(self, col, translate=True): Laurent@814: if col < len(self.colnames): Laurent@814: if translate: Laurent@814: return _(self.colnames[col]) Laurent@814: return self.colnames[col] Laurent@814: Laurent@814: def GetRowLabelValue(self, row, translate=True): Laurent@814: return row Laurent@814: Laurent@814: def GetValue(self, row, col): Laurent@814: if row < self.GetNumberRows(): Laurent@814: return self.data[row].get(self.GetColLabelValue(col, False), "") Laurent@814: Laurent@814: def SetValue(self, row, col, value): Laurent@814: if col < len(self.colnames): Laurent@814: self.data[row][self.GetColLabelValue(col, False)] = value Laurent@814: Laurent@814: def GetValueByName(self, row, colname): Laurent@814: if row < self.GetNumberRows(): Laurent@814: return self.data[row].get(colname) Laurent@814: Laurent@814: def SetValueByName(self, row, colname, value): Laurent@814: if row < self.GetNumberRows(): Laurent@814: self.data[row][colname] = value Laurent@814: Laurent@814: def ResetView(self, grid): Laurent@814: """ Laurent@814: (wx.grid.Grid) -> Reset the grid view. Call this to Laurent@814: update the grid if rows and columns have been added or deleted Laurent@814: """ Laurent@814: grid.CloseEditControl() Laurent@814: grid.BeginBatch() Laurent@814: for current, new, delmsg, addmsg in [ Laurent@814: (self._rows, self.GetNumberRows(), wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED), Laurent@814: (self._cols, self.GetNumberCols(), wx.grid.GRIDTABLE_NOTIFY_COLS_DELETED, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED), Laurent@814: ]: Laurent@814: if new < current: Laurent@814: msg = wx.grid.GridTableMessage(self,delmsg,new,current-new) Laurent@814: grid.ProcessTableMessage(msg) Laurent@814: elif new > current: Laurent@814: msg = wx.grid.GridTableMessage(self,addmsg,new-current) Laurent@814: grid.ProcessTableMessage(msg) Laurent@814: self.UpdateValues(grid) Laurent@814: grid.EndBatch() Laurent@814: Laurent@814: self._rows = self.GetNumberRows() Laurent@814: self._cols = self.GetNumberCols() Laurent@814: # update the column rendering scheme Laurent@814: self._updateColAttrs(grid) Laurent@814: Laurent@814: # update the scrollbars and the displayed part of the grid Laurent@814: grid.AdjustScrollbars() Laurent@814: grid.ForceRefresh() Laurent@814: Laurent@814: def UpdateValues(self, grid): Laurent@814: """Update all displayed values""" Laurent@814: # This sends an event to the grid table to update all of the values Laurent@814: msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_REQUEST_VIEW_GET_VALUES) Laurent@814: grid.ProcessTableMessage(msg) Laurent@814: Laurent@814: def _updateColAttrs(self, grid): Laurent@814: """ Laurent@814: wx.grid.Grid -> update the column attributes to add the Laurent@814: appropriate renderer given the column name. Laurent@814: Laurent@814: Otherwise default to the default renderer. Laurent@814: """ Laurent@814: for row in range(self.GetNumberRows()): Laurent@814: row_highlights = self.Highlights.get(row, {}) Laurent@814: for col in range(self.GetNumberCols()): Laurent@814: colname = self.GetColLabelValue(col, False) Laurent@814: Laurent@814: grid.SetReadOnly(row, col, True) Laurent@814: grid.SetCellEditor(row, col, None) Laurent@814: grid.SetCellRenderer(row, col, None) Laurent@814: Laurent@814: highlight_colours = row_highlights.get(colname.lower(), [(wx.WHITE, wx.BLACK)])[-1] Laurent@814: grid.SetCellBackgroundColour(row, col, highlight_colours[0]) Laurent@814: grid.SetCellTextColour(row, col, highlight_colours[1]) Laurent@814: self.ResizeRow(grid, row) Laurent@814: Laurent@814: def ResizeRow(self, grid, row): Laurent@814: if wx.Platform == '__WXMSW__': Laurent@814: grid.SetRowMinimalHeight(row, 20) Laurent@814: else: Laurent@814: grid.SetRowMinimalHeight(row, 28) Laurent@814: grid.AutoSizeRow(row, False) Laurent@814: Laurent@814: def SetData(self, data): Laurent@814: self.data = data Laurent@814: Laurent@814: def GetData(self): Laurent@814: return self.data Laurent@814: Laurent@814: def GetCurrentIndex(self): Laurent@814: return self.CurrentIndex Laurent@814: Laurent@814: def SetCurrentIndex(self, index): Laurent@814: self.CurrentIndex = index Laurent@814: Laurent@814: def AppendRow(self, row_content): Laurent@814: self.data.append(row_content) Laurent@814: Laurent@814: def InsertRow(self, index, row_content): Laurent@814: self.data.insert(index, row_content) Laurent@814: Laurent@814: def MoveRow(self, row_index, move): Laurent@814: new_index = max(0, min(row_index + move, len(self.data) - 1)) Laurent@814: if new_index != row_index: Laurent@814: self.data.insert(new_index, self.data.pop(row_index)) Laurent@814: return new_index Laurent@814: Laurent@814: def RemoveRow(self, row_index): Laurent@814: self.data.pop(row_index) Laurent@814: Laurent@814: def GetRow(self, row_index): Laurent@814: return self.data[row_index] Laurent@814: Laurent@814: def Empty(self): Laurent@814: self.data = [] Laurent@814: Laurent@814: def AddHighlight(self, infos, highlight_type): Laurent@814: row_highlights = self.Highlights.setdefault(infos[0], {}) Laurent@814: col_highlights = row_highlights.setdefault(infos[1], []) Laurent@814: col_highlights.append(highlight_type) Laurent@814: Laurent@814: def RemoveHighlight(self, infos, highlight_type): Laurent@814: row_highlights = self.Highlights.get(infos[0]) Laurent@814: if row_highlights is not None: Laurent@814: col_highlights = row_highlights.get(infos[1]) Laurent@814: if col_highlights is not None and highlight_type in col_highlights: Laurent@814: col_highlights.remove(highlight_type) Laurent@814: if len(col_highlights) == 0: Laurent@814: row_highlights.pop(infos[1]) Laurent@814: Laurent@814: def ClearHighlights(self, highlight_type=None): Laurent@814: if highlight_type is None: Laurent@814: self.Highlights = {} Laurent@814: else: Laurent@814: for row, row_highlights in self.Highlights.iteritems(): Laurent@814: row_items = row_highlights.items() Laurent@814: for col, col_highlights in row_items: Laurent@814: if highlight_type in col_highlights: Laurent@814: col_highlights.remove(highlight_type) Laurent@814: if len(col_highlights) == 0: Laurent@814: row_highlights.pop(col)