/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/UI/tableview.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/UI/tableview.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/thuban/Thuban/UI/tableview.py revision 2561 by bh, Tue Feb 8 20:34:29 2005 UTC branches/WIP-pyshapelib-bramz/Thuban/UI/tableview.py revision 2755 by dpinte, Thu Apr 12 09:21:58 2007 UTC
# Line 11  import os.path Line 11  import os.path
11    
12  from Thuban import _  from Thuban import _
13    
14  from wxPython.wx import *  import wx
15  from wxPython.grid import *  from wx import grid
16    
17  from Thuban.Lib.connector import Publisher  from Thuban.Lib.connector import Publisher
18  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
# Line 23  from messages import SHAPES_SELECTED, SE Line 23  from messages import SHAPES_SELECTED, SE
23  from Thuban.Model.messages import TABLE_REMOVED, MAP_LAYERS_REMOVED  from Thuban.Model.messages import TABLE_REMOVED, MAP_LAYERS_REMOVED
24  from Thuban.UI.common import ThubanBeginBusyCursor, ThubanEndBusyCursor  from Thuban.UI.common import ThubanBeginBusyCursor, ThubanEndBusyCursor
25    
26  wx_value_type_map = {FIELDTYPE_INT: wxGRID_VALUE_NUMBER,  wx_value_type_map = {FIELDTYPE_INT: grid.GRID_VALUE_NUMBER,
27                       FIELDTYPE_DOUBLE: wxGRID_VALUE_FLOAT,                       FIELDTYPE_DOUBLE: grid.GRID_VALUE_FLOAT,
28                       FIELDTYPE_STRING: wxGRID_VALUE_STRING}                       FIELDTYPE_STRING: grid.GRID_VALUE_STRING}
29    
30  ROW_SELECTED = "ROW_SELECTED"  ROW_SELECTED = "ROW_SELECTED"
31    
32  QUERY_KEY = 'S'  QUERY_KEY = 'S'
33    
34  class DataTable(wxPyGridTableBase):  class DataTable(grid.PyGridTableBase):
35    
36      """Wrapper around a Thuban table object suitable for a wxGrid"""      """Wrapper around a Thuban table object suitable for a wxGrid"""
37    
38      def __init__(self, table = None):      def __init__(self, table = None):
39          wxPyGridTableBase.__init__(self)          grid.PyGridTableBase.__init__(self)
40          self.num_cols = 0          self.num_cols = 0
41          self.num_rows = 0          self.num_rows = 0
42          self.columns = []          self.columns = []
# Line 71  class DataTable(wxPyGridTableBase): Line 71  class DataTable(wxPyGridTableBase):
71      # Renderer understands the type too,) not just strings as in the      # Renderer understands the type too,) not just strings as in the
72      # C++ version.      # C++ version.
73      def GetValue(self, row, col):      def GetValue(self, row, col):
74          record = self.table.ReadRowAsDict(row, row_is_ordinal = 1)          try:
75                record = self.table.ReadRowAsDict(row, row_is_ordinal = 1)
76            except UnicodeError:
77                record = dict()
78                for (key, val) in self.table.ReadRowAsDict(row, \
79                                  row_is_ordinal = 1).items():
80                    record[key] = val
81          return record[self.columns[col][0]]          return record[self.columns[col][0]]
82    
83      def SetValue(self, row, col, value):      def SetValue(self, row, col, value):
# Line 112  class DataTable(wxPyGridTableBase): Line 118  class DataTable(wxPyGridTableBase):
118          return self.table.RowOrdinalToId(ordinal)          return self.table.RowOrdinalToId(ordinal)
119    
120    
121  class NullRenderer(wxPyGridCellRenderer):  class NullRenderer(grid.PyGridCellRenderer):
122    
123      """Renderer that draws NULL as a gray rectangle      """Renderer that draws NULL as a gray rectangle
124    
# Line 121  class NullRenderer(wxPyGridCellRenderer) Line 127  class NullRenderer(wxPyGridCellRenderer)
127      """      """
128    
129      def __init__(self, non_null_renderer):      def __init__(self, non_null_renderer):
130          wxPyGridCellRenderer.__init__(self)          grid.PyGridCellRenderer.__init__(self)
131          self.non_null_renderer = non_null_renderer          self.non_null_renderer = non_null_renderer
132    
133      def Draw(self, grid, attr, dc, rect, row, col, isSelected):      def Draw(self, grid, attr, dc, rect, row, col, isSelected):
134          value = grid.table.GetValue(row, col)          value = grid.table.GetValue(row, col)
135          if value is None:          if value is None:
136              dc.SetBackgroundMode(wxSOLID)              dc.SetBackgroundMode(wx.SOLID)
137              dc.SetBrush(wxBrush(wxColour(192, 192, 192), wxSOLID))              dc.SetBrush(wx.Brush(wx.Colour(192, 192, 192), wx.SOLID))
138              dc.SetPen(wxTRANSPARENT_PEN)              dc.SetPen(wx.TRANSPARENT_PEN)
139              dc.DrawRectangle(rect.x, rect.y, rect.width, rect.height)              dc.DrawRectangle(rect.x, rect.y, rect.width, rect.height)
140          else:          else:
141              self.non_null_renderer.Draw(grid, attr, dc, rect, row, col,              self.non_null_renderer.Draw(grid, attr, dc, rect, row, col,
# Line 142  class NullRenderer(wxPyGridCellRenderer) Line 148  class NullRenderer(wxPyGridCellRenderer)
148          return NullRenderer(self.non_null_renderer)          return NullRenderer(self.non_null_renderer)
149    
150    
151  class TableGrid(wxGrid, Publisher):  class TableGrid(grid.Grid, Publisher):
152    
153      """A grid view for a Thuban table      """A grid view for a Thuban table
154    
# Line 155  class TableGrid(wxGrid, Publisher): Line 161  class TableGrid(wxGrid, Publisher):
161      """      """
162    
163      def __init__(self, parent, table = None):      def __init__(self, parent, table = None):
164          wxGrid.__init__(self, parent, -1)          grid.Grid.__init__(self, parent, -1)
165    
166          self.allow_messages_count = 0          self.allow_messages_count = 0
167    
# Line 178  class TableGrid(wxGrid, Publisher): Line 184  class TableGrid(wxGrid, Publisher):
184          # long time.          # long time.
185          #self.AutoSizeColumns(False)          #self.AutoSizeColumns(False)
186    
187          self.SetSelectionMode(wxGrid.wxGridSelectRows)          self.SetSelectionMode(grid.Grid.wxGridSelectRows)
188    
189          self.ToggleEventListeners(True)          self.ToggleEventListeners(True)
190          EVT_GRID_RANGE_SELECT(self, self.OnRangeSelect)          self.Bind(grid.EVT_GRID_RANGE_SELECT, self.OnRangeSelect)
191          EVT_GRID_SELECT_CELL(self, self.OnSelectCell)          self.Bind(grid.EVT_GRID_SELECT_CELL, self.OnSelectCell)
192    
193          # Replace the normal renderers with our own versions which          # Replace the normal renderers with our own versions which
194          # render NULL/None values specially          # render NULL/None values specially
195          self.RegisterDataType(wxGRID_VALUE_STRING,          self.RegisterDataType(grid.GRID_VALUE_STRING,
196                                NullRenderer(wxGridCellStringRenderer()), None)                                NullRenderer(grid.GridCellStringRenderer()), None)
197          self.RegisterDataType(wxGRID_VALUE_NUMBER,          self.RegisterDataType(grid.GRID_VALUE_NUMBER,
198                                NullRenderer(wxGridCellNumberRenderer()), None)                                NullRenderer(grid.GridCellNumberRenderer()), None)
199          self.RegisterDataType(wxGRID_VALUE_FLOAT,          self.RegisterDataType(grid.GRID_VALUE_FLOAT,
200                                NullRenderer(wxGridCellFloatRenderer()), None)                                NullRenderer(grid.GridCellFloatRenderer()), None)
201    
202          EVT_WINDOW_DESTROY(self, self.OnDestroy)          self.Bind(wx.EVT_WINDOW_DESTROY, self.OnDestroy)
203    
204      def OnDestroy(self, event):      def OnDestroy(self, event):
205          Publisher.Destroy(self)          Publisher.Destroy(self)
# Line 306  class TableFrame(ThubanFrame): Line 312  class TableFrame(ThubanFrame):
312    
313      def __init__(self, parent, name, title, table):      def __init__(self, parent, name, title, table):
314          ThubanFrame.__init__(self, parent, name, title)          ThubanFrame.__init__(self, parent, name, title)
315          self.panel = wxPanel(self, -1)          self.panel = wx.Panel(self, -1)
316    
317          self.table = table          self.table = table
318          self.grid = self.make_grid(self.table)          self.grid = self.make_grid(self.table)
# Line 364  class QueryTableFrame(TableFrame): Line 370  class QueryTableFrame(TableFrame):
370      def __init__(self, parent, name, title, table):      def __init__(self, parent, name, title, table):
371          TableFrame.__init__(self, parent, name, title, table)          TableFrame.__init__(self, parent, name, title, table)
372    
373          self.combo_fields = wxComboBox(self.panel, -1, style=wxCB_READONLY)          self.combo_fields = wx.ComboBox(self.panel, -1, style=wx.CB_READONLY)
374          self.choice_comp = wxChoice(self.panel, -1,          self.choice_comp = wx.Choice(self.panel, -1,
375                                choices=["<", "<=", "==", "!=", ">=", ">"])                                choices=["<", "<=", "==", "!=", ">=", ">"])
376          self.combo_value = wxComboBox(self.panel, ID_COMBOVALUE)          self.combo_value = wx.ComboBox(self.panel, ID_COMBOVALUE)
377          self.choice_action = wxChoice(self.panel, -1,          self.choice_action = wx.Choice(self.panel, -1,
378                                  choices=[_("Replace Selection"),                                  choices=[_("Replace Selection"),
379                                          _("Refine Selection"),                                          _("Refine Selection"),
380                                          _("Add to Selection")])                                          _("Add to Selection")])
381    
382          button_query = wxButton(self.panel, ID_QUERY, _("Query"))          button_query = wx.Button(self.panel, ID_QUERY, _("Query"))
383          button_export = wxButton(self.panel, ID_EXPORT, _("Export"))          button_export = wx.Button(self.panel, ID_EXPORT, _("Export"))
384          button_exportSel = wxButton(self.panel, ID_EXPORTSEL, _("Export Selection"))          button_exportSel = wx.Button(self.panel, ID_EXPORTSEL, _("Export Selection"))
385          button_close = wxButton(self.panel, wxID_CLOSE, _("Close"))          button_close = wx.Button(self.panel, wx.ID_CLOSE, _("Close"))
386    
387          self.CreateStatusBar()          self.CreateStatusBar()
388    
# Line 398  class QueryTableFrame(TableFrame): Line 404  class QueryTableFrame(TableFrame):
404    
405          self.UpdateStatusText()          self.UpdateStatusText()
406    
407          topBox = wxBoxSizer(wxVERTICAL)          topBox = wx.BoxSizer(wx.VERTICAL)
408    
409          sizer = wxStaticBoxSizer(wxStaticBox(self.panel, -1,          sizer = wx.StaticBoxSizer(wx.StaticBox(self.panel, -1,
410                                    _("Selection")),                                    _("Selection")),
411                                    wxHORIZONTAL)                                    wx.HORIZONTAL)
412          sizer.Add(self.combo_fields, 1, wxEXPAND|wxALL, 4)          sizer.Add(self.combo_fields, 1, wx.EXPAND|wx.ALL, 4)
413          sizer.Add(self.choice_comp, 0, wxALL, 4)          sizer.Add(self.choice_comp, 0, wx.ALL, 4)
414          sizer.Add(self.combo_value, 1, wxEXPAND|wxALL, 4)          sizer.Add(self.combo_value, 1, wx.EXPAND|wx.ALL, 4)
415          sizer.Add(self.choice_action, 0, wxALL, 4)          sizer.Add(self.choice_action, 0, wx.ALL, 4)
416          sizer.Add(button_query, 0, wxALL | wxALIGN_CENTER_VERTICAL, 4)          sizer.Add(button_query, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 4)
417          sizer.Add( (40, 20), 0, wxALL, 4)          sizer.Add( (40, 20), 0, wx.ALL, 4)
418    
419          topBox.Add(sizer, 0, wxEXPAND|wxALL, 4)          topBox.Add(sizer, 0, wx.EXPAND|wx.ALL, 4)
420          topBox.Add(self.grid, 1, wxEXPAND|wxALL, 0)          topBox.Add(self.grid, 1, wx.EXPAND|wx.ALL, 0)
421    
422          sizer = wxBoxSizer(wxHORIZONTAL)          sizer = wx.BoxSizer(wx.HORIZONTAL)
423          sizer.Add(button_export, 0, wxALL, 4)          sizer.Add(button_export, 0, wx.ALL, 4)
424          sizer.Add(button_exportSel, 0, wxALL, 4)          sizer.Add(button_exportSel, 0, wx.ALL, 4)
425          sizer.Add( (60, 20), 1, wxALL|wxEXPAND, 4)          sizer.Add( (60, 20), 1, wx.ALL|wx.EXPAND, 4)
426          sizer.Add(button_close, 0, wxALL|wxALIGN_RIGHT, 4)          sizer.Add(button_close, 0, wx.ALL|wx.ALIGN_RIGHT, 4)
427          topBox.Add(sizer, 0, wxALL | wxEXPAND, 4)          topBox.Add(sizer, 0, wx.ALL | wx.EXPAND, 4)
428    
429          self.panel.SetAutoLayout(True)          self.panel.SetAutoLayout(True)
430          self.panel.SetSizer(topBox)          self.panel.SetSizer(topBox)
431          topBox.Fit(self.panel)          topBox.Fit(self.panel)
432          topBox.SetSizeHints(self.panel)          topBox.SetSizeHints(self.panel)
433    
434          panelSizer = wxBoxSizer(wxVERTICAL)          panelSizer = wx.BoxSizer(wx.VERTICAL)
435          panelSizer.Add(self.panel, 1, wxEXPAND, 0)          panelSizer.Add(self.panel, 1, wx.EXPAND, 0)
436          self.SetAutoLayout(True)          self.SetAutoLayout(True)
437          self.SetSizer(panelSizer)          self.SetSizer(panelSizer)
438          panelSizer.Fit(self)          panelSizer.Fit(self)
# Line 434  class QueryTableFrame(TableFrame): Line 440  class QueryTableFrame(TableFrame):
440    
441          self.grid.SetFocus()          self.grid.SetFocus()
442    
443          EVT_BUTTON(self, ID_QUERY, self.OnQuery)          self.Bind(wx.EVT_BUTTON, self.OnQuery, id=ID_QUERY)
444          EVT_BUTTON(self, ID_EXPORT, self.OnExport)          self.Bind(wx.EVT_BUTTON, self.OnExport, id=ID_EXPORT)
445          EVT_BUTTON(self, ID_EXPORTSEL, self.OnExportSel)          self.Bind(wx.EVT_BUTTON, self.OnExportSel, id=ID_EXPORTSEL)
446          EVT_BUTTON(self, wxID_CLOSE, self.OnClose)          self.Bind(wx.EVT_BUTTON, self.OnClose, id=wx.ID_CLOSE)
447          EVT_KEY_DOWN(self.grid, self.OnKeyDown)          self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown, self.grid)
448    
449          self.grid.Subscribe(ROW_SELECTED, self.UpdateStatusText)          self.grid.Subscribe(ROW_SELECTED, self.UpdateStatusText)
450    
451      def UpdateStatusText(self, rows=None):      def UpdateStatusText(self, rows=None):
452          self.SetStatusText(_("%i rows (%i selected), %i columns")          self.SetStatusText(_("%i rows (%i selected), %i columns")
453              % (self.grid.GetNumberRows(),              % (self.grid.GetNumberRows(),
454                 self.grid.GetNumberSelected(),                 self.grid.GetNumberSelected(),
455                 self.grid.GetNumberCols()))                 self.grid.GetNumberCols()))
456    
# Line 468  class QueryTableFrame(TableFrame): Line 474  class QueryTableFrame(TableFrame):
474                  value = self.table.Column(text)                  value = self.table.Column(text)
475    
476              ids = self.table.SimpleQuery(              ids = self.table.SimpleQuery(
477                      self.table.Column(self.combo_fields.GetStringSelection()),                      self.table.Column(self.combo_fields.GetStringSelection()),
478                      self.choice_comp.GetStringSelection(),                      self.choice_comp.GetStringSelection(),
479                      value)                      value)
480    
481              choice = self.choice_action.GetSelection()              choice = self.choice_action.GetSelection()
482                
483              #              #
484              # what used to be nice code got became a bit ugly because              # what used to be nice code got became a bit ugly because
485              # each time we select a row a message is sent to the grid              # each time we select a row a message is sent to the grid
# Line 520  class QueryTableFrame(TableFrame): Line 526  class QueryTableFrame(TableFrame):
526    
527          finally:          finally:
528              ThubanEndBusyCursor()              ThubanEndBusyCursor()
529            
530      def doExport(self, onlySelected):      def doExport(self, onlySelected):
531            
532          dlg = wxFileDialog(self, _("Export Table To"), ".", "",          dlg = wx.FileDialog(self, _("Export Table To"), ".", "",
533                             _("DBF Files (*.dbf)|*.dbf|") +                             _("DBF Files (*.dbf)|*.dbf|") +
534                             _("CSV Files (*.csv)|*.csv|") +                             _("CSV Files (*.csv)|*.csv|") +
535                             _("All Files (*.*)|*.*"),                             _("All Files (*.*)|*.*"),
536                             wxSAVE|wxOVERWRITE_PROMPT)                             wx.SAVE|wx.OVERWRITE_PROMPT)
537          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wx.ID_OK:
538              filename = dlg.GetPath()              filename = dlg.GetPath()
539              type = os.path.basename(filename).split('.')[-1:][0]              type = os.path.basename(filename).split('.')[-1:][0]
540              dlg.Destroy()              dlg.Destroy()
# Line 543  class QueryTableFrame(TableFrame): Line 549  class QueryTableFrame(TableFrame):
549              elif type.upper() == 'CSV':              elif type.upper() == 'CSV':
550                  table_to_csv(self.table, filename, records)                  table_to_csv(self.table, filename, records)
551              else:              else:
552                  dlg = wxMessageDialog(None, "Unsupported format: %s" % type,                  dlg = wx.MessageDialog(None, "Unsupported format: %s" % type,
553                                        "Table Export", wxOK|wxICON_WARNING)                                        "Table Export", wx.OK|wx.ICON_WARNING)
554                  dlg.ShowModal()                  dlg.ShowModal()
555                  dlg.Destroy()                  dlg.Destroy()
556          else:          else:
Legend:
Removed from v.2561  
changed lines
  Added in v.2755
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26