/[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

revision 1146 by bh, Tue Jun 10 12:00:16 2003 UTC revision 1662 by bh, Wed Aug 27 13:51:01 2003 UTC
# Line 17  from wxPython.grid import * Line 17  from wxPython.grid import *
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, \
19       FIELDTYPE_STRING, table_to_dbf, table_to_csv       FIELDTYPE_STRING, table_to_dbf, table_to_csv
20  import view  from dialogs import ThubanFrame
 from dialogs import NonModalNonParentDialog  
21    
22  from messages import SHAPES_SELECTED, SESSION_REPLACED  from messages import SHAPES_SELECTED, SESSION_REPLACED
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
25    
26  wx_value_type_map = {FIELDTYPE_INT: wxGRID_VALUE_NUMBER,  wx_value_type_map = {FIELDTYPE_INT: wxGRID_VALUE_NUMBER,
27                       FIELDTYPE_DOUBLE: wxGRID_VALUE_FLOAT,                       FIELDTYPE_DOUBLE: wxGRID_VALUE_FLOAT,
# 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)          record = self.table.ReadRowAsDict(row, row_is_ordinal = 1)
75          return record[self.columns[col][0]]          return record[self.columns[col][0]]
76    
77      def SetValue(self, row, col, value):      def SetValue(self, row, col, value):
# Line 102  class DataTable(wxPyGridTableBase): Line 102  class DataTable(wxPyGridTableBase):
102          return self.CanGetValueAs(row, col, typeName)          return self.CanGetValueAs(row, col, typeName)
103    
104    
105        #
106        def RowIdToOrdinal(self, rowid):
107            """Return the ordinal of the row given by its id"""
108            return self.table.RowIdToOrdinal(rowid)
109    
110        def RowOrdinalToId(self, ordinal):
111            """Return the id of the row given by its ordinal"""
112            return self.table.RowOrdinalToId(ordinal)
113    
114    
115  class NullRenderer(wxPyGridCellRenderer):  class NullRenderer(wxPyGridCellRenderer):
116    
# Line 150  class TableGrid(wxGrid, Publisher): Line 159  class TableGrid(wxGrid, Publisher):
159    
160          self.allow_messages_count = 0          self.allow_messages_count = 0
161    
162            # keep track of which rows are selected.
163            self.rows = {}
164    
165          self.table = DataTable(table)          self.table = DataTable(table)
166    
167          # The second parameter means that the grid is to take ownership          # The second parameter means that the grid is to take ownership
# Line 169  class TableGrid(wxGrid, Publisher): Line 181  class TableGrid(wxGrid, Publisher):
181          self.SetSelectionMode(wxGrid.wxGridSelectRows)          self.SetSelectionMode(wxGrid.wxGridSelectRows)
182    
183          self.ToggleEventListeners(True)          self.ToggleEventListeners(True)
184            EVT_GRID_RANGE_SELECT(self, self.OnRangeSelect)
185            EVT_GRID_SELECT_CELL(self, self.OnSelectCell)
186    
187          # Replace the normal renderers with our own versions which          # Replace the normal renderers with our own versions which
188          # render NULL/None values specially          # render NULL/None values specially
# Line 183  class TableGrid(wxGrid, Publisher): Line 197  class TableGrid(wxGrid, Publisher):
197          self.table.SetTable(table)          self.table.SetTable(table)
198    
199      def OnRangeSelect(self, event):      def OnRangeSelect(self, event):
200          rows = dict([(i, 0) for i in self.GetSelectedRows()])          to_id = self.table.RowOrdinalToId
201            if self.handleSelectEvents:
202                self.rows = dict([(to_id(i), 0) for i in self.GetSelectedRows()])
203    
204                # if we're selecting we need to include the selected range and
205                # make sure that the current row is also included, which may
206                # not be the case if you just click on a single row!
207                if event.Selecting():
208                    for i in range(event.GetTopRow(), event.GetBottomRow() + 1):
209                        self.rows[to_id(i)] = 0
210                    self.rows[to_id(event.GetTopLeftCoords().GetRow())] = 0
211    
212          # if we're selecting we need to include the selected range and              self.issue(ROW_SELECTED, self.rows.keys())
         # make sure that the current row is also included, which may  
         # not be the case if you just click on a single row!  
         if event.Selecting():  
             for i in range(event.GetTopRow(), event.GetBottomRow() + 1):  
                 rows[i] = 0  
             rows[event.GetTopLeftCoords().GetRow()] = 0  
213    
         self.issue(ROW_SELECTED, rows.keys())  
214          event.Skip()          event.Skip()
215    
216      def OnSelectCell(self, event):      def OnSelectCell(self, event):
217          self.issue(ROW_SELECTED, self.GetSelectedRows())          to_id = self.table.RowOrdinalToId
218            if self.handleSelectEvents:
219                self.issue(ROW_SELECTED,
220                           [to_id(i) for i in self.GetSelectedRows()])
221          event.Skip()          event.Skip()
222    
223      def ToggleEventListeners(self, on):      def ToggleEventListeners(self, on):
224          if on:          self.handleSelectEvents = on
225              EVT_GRID_RANGE_SELECT(self, self.OnRangeSelect)  
226              EVT_GRID_SELECT_CELL(self, self.OnSelectCell)      def GetNumberSelected(self):
227          else:          return len(self.rows)
228              EVT_GRID_RANGE_SELECT(self, None)  
             EVT_GRID_SELECT_CELL(self, None)  
               
229      def disallow_messages(self):      def disallow_messages(self):
230          """Disallow messages to be send.          """Disallow messages to be send.
231    
# Line 234  class TableGrid(wxGrid, Publisher): Line 252  class TableGrid(wxGrid, Publisher):
252          if self.allow_messages_count == 0:          if self.allow_messages_count == 0:
253              Publisher.issue(self, *args)              Publisher.issue(self, *args)
254    
255        def SelectRowById(self, rowid, do_select):
256            """Select row with the id rowid"""
257            self.SelectRow(self.table.RowIdToOrdinal(rowid), do_select)
258    
259    
260  class LayerTableGrid(TableGrid):  class LayerTableGrid(TableGrid):
261    
# Line 250  class LayerTableGrid(TableGrid): Line 272  class LayerTableGrid(TableGrid):
272          nothing. If shape or layer is None also do nothing.          nothing. If shape or layer is None also do nothing.
273          """          """
274          if layer is not None \          if layer is not None \
275              and layer.table is self.table.table:              and layer.ShapeStore().Table() is self.table.table:
276    
277              self.disallow_messages()              self.disallow_messages()
278              try:              try:
279                  self.ClearSelection()                  self.ClearSelection()
280                  if len(shapes) > 0:                  if len(shapes) > 0:
281                      #                      # keep track of the lowest id so we can make it the
282                      # keep track of the lowest id so we can make it                      # first visible item
283                      # the first visible item                      first = -1
                     #  
                     first = shapes[0]  
284    
285                        to_ordinal = self.table.RowIdToOrdinal
286                      for shape in shapes:                      for shape in shapes:
287                          self.SelectRow(shape, True)                          row = to_ordinal(shape)
288                          if shape < first:                          self.SelectRow(row, True)
289                              first = shape                          if row < first:
290                                first = row
291    
292                      self.SetGridCursor(first, 0)                      self.SetGridCursor(first, 0)
293                      self.MakeCellVisible(first, 0)                      self.MakeCellVisible(first, 0)
# Line 273  class LayerTableGrid(TableGrid): Line 295  class LayerTableGrid(TableGrid):
295                  self.allow_messages()                  self.allow_messages()
296    
297    
298  class TableFrame(NonModalNonParentDialog):  class TableFrame(ThubanFrame):
299    
300      """Frame that displays a Thuban table in a grid view"""      """Frame that displays a Thuban table in a grid view"""
301    
302      def __init__(self, parent, name, title, table):      def __init__(self, parent, name, title, table):
303          NonModalNonParentDialog.__init__(self, parent, name, title)          ThubanFrame.__init__(self, parent, name, title)
304            self.panel = wxPanel(self, -1)
305    
306          self.table = table          self.table = table
307          self.grid = self.make_grid(self.table)          self.grid = self.make_grid(self.table)
308          self.app = self.parent.application          self.app = self.parent.application
# Line 286  class TableFrame(NonModalNonParentDialog Line 310  class TableFrame(NonModalNonParentDialog
310          self.session = self.app.Session()          self.session = self.app.Session()
311          self.session.Subscribe(TABLE_REMOVED, self.close_on_table_removed)          self.session.Subscribe(TABLE_REMOVED, self.close_on_table_removed)
312    
313    
314      def make_grid(self, table):      def make_grid(self, table):
315          """Return the table grid to use in the frame.          """Return the table grid to use in the frame.
316    
# Line 297  class TableFrame(NonModalNonParentDialog Line 322  class TableFrame(NonModalNonParentDialog
322      def OnClose(self, event):      def OnClose(self, event):
323          self.app.Unsubscribe(SESSION_REPLACED, self.close_on_session_replaced)          self.app.Unsubscribe(SESSION_REPLACED, self.close_on_session_replaced)
324          self.session.Unsubscribe(TABLE_REMOVED, self.close_on_table_removed)          self.session.Unsubscribe(TABLE_REMOVED, self.close_on_table_removed)
325          NonModalNonParentDialog.OnClose(self, event)          ThubanFrame.OnClose(self, event)
326    
327      def close_on_session_replaced(self, *args):      def close_on_session_replaced(self, *args):
328          """Subscriber for the SESSION_REPLACED messages.          """Subscriber for the SESSION_REPLACED messages.
# Line 319  class TableFrame(NonModalNonParentDialog Line 344  class TableFrame(NonModalNonParentDialog
344    
345  ID_QUERY = 4001  ID_QUERY = 4001
346  ID_EXPORT = 4002  ID_EXPORT = 4002
347    ID_COMBOVALUE = 4003
348    ID_EXPORTSEL = 4004
349    
350  class QueryTableFrame(TableFrame):  class QueryTableFrame(TableFrame):
351    
# Line 332  class QueryTableFrame(TableFrame): Line 359  class QueryTableFrame(TableFrame):
359      def __init__(self, parent, name, title, table):      def __init__(self, parent, name, title, table):
360          TableFrame.__init__(self, parent, name, title, table)          TableFrame.__init__(self, parent, name, title, table)
361    
362          self.combo_fields = wxComboBox(self, -1, style=wxCB_READONLY)          self.combo_fields = wxComboBox(self.panel, -1, style=wxCB_READONLY)
363          self.choice_comp = wxChoice(self, -1,          self.choice_comp = wxChoice(self.panel, -1,
364                                choices=["<", "<=", "==", "!=", ">=", ">"])                                choices=["<", "<=", "==", "!=", ">=", ">"])
365          self.combo_value = wxComboBox(self, -1)          self.combo_value = wxComboBox(self.panel, ID_COMBOVALUE)
366          self.choice_action = wxChoice(self, -1,          self.choice_action = wxChoice(self.panel, -1,
367                                  choices=[_("Replace Selection"),                                  choices=[_("Replace Selection"),
368                                          _("Refine Selection"),                                          _("Refine Selection"),
369                                          _("Add to Selection")])                                          _("Add to Selection")])
370    
371          button_query = wxButton(self, ID_QUERY, _("Query"))          button_query = wxButton(self.panel, ID_QUERY, _("Query"))
372          button_saveas = wxButton(self, ID_EXPORT, _("Export"))          button_export = wxButton(self.panel, ID_EXPORT, _("Export"))
373            button_exportSel = wxButton(self.panel, ID_EXPORTSEL, _("Export Selection"))
374            button_close = wxButton(self.panel, wxID_CLOSE, _("Close"))
375    
376            self.CreateStatusBar()
377    
378          self.grid.SetSize((400, 200))          self.grid.SetSize((400, 200))
379    
# Line 355  class QueryTableFrame(TableFrame): Line 386  class QueryTableFrame(TableFrame):
386          # assume at least one field?          # assume at least one field?
387          self.combo_fields.SetSelection(0)          self.combo_fields.SetSelection(0)
388          self.combo_value.SetSelection(0)          self.combo_value.SetSelection(0)
389            self.choice_action.SetSelection(0)
390            self.choice_comp.SetSelection(0)
391    
392            self.grid.Reparent(self.panel)
393    
394            self.UpdateStatusText()
395    
396          topBox = wxBoxSizer(wxVERTICAL)          topBox = wxBoxSizer(wxVERTICAL)
397    
398          sizer = wxStaticBoxSizer(wxStaticBox(self, -1,          sizer = wxStaticBoxSizer(wxStaticBox(self.panel, -1,
399                                    _("Selection")),                                    _("Selection")),
400                                    wxHORIZONTAL)                                    wxHORIZONTAL)
401          sizer.Add(self.combo_fields, 1, wxEXPAND|wxALL, 4)          sizer.Add(self.combo_fields, 1, wxEXPAND|wxALL, 4)
# Line 367  class QueryTableFrame(TableFrame): Line 404  class QueryTableFrame(TableFrame):
404          sizer.Add(self.choice_action, 0, wxALL, 4)          sizer.Add(self.choice_action, 0, wxALL, 4)
405          sizer.Add(button_query, 0, wxALL | wxALIGN_CENTER_VERTICAL, 4)          sizer.Add(button_query, 0, wxALL | wxALIGN_CENTER_VERTICAL, 4)
406          sizer.Add(40, 20, 0, wxALL, 4)          sizer.Add(40, 20, 0, wxALL, 4)
         sizer.Add(button_saveas, 0, wxALL | wxALIGN_CENTER_VERTICAL, 4)  
407    
408          topBox.Add(sizer, 0, wxEXPAND|wxALL, 4)          topBox.Add(sizer, 0, wxEXPAND|wxALL, 4)
409          topBox.Add(self.grid, 1, wxEXPAND|wxALL, 0)          topBox.Add(self.grid, 1, wxEXPAND|wxALL, 0)
410    
411            sizer = wxBoxSizer(wxHORIZONTAL)
412            sizer.Add(button_export, 0, wxALL, 4)
413            sizer.Add(button_exportSel, 0, wxALL, 4)
414            sizer.Add(60, 20, 1, wxALL|wxEXPAND, 4)
415            sizer.Add(button_close, 0, wxALL|wxALIGN_RIGHT, 4)
416            topBox.Add(sizer, 0, wxALL | wxEXPAND, 4)
417    
418            self.panel.SetAutoLayout(True)
419            self.panel.SetSizer(topBox)
420            topBox.Fit(self.panel)
421            topBox.SetSizeHints(self.panel)
422    
423            panelSizer = wxBoxSizer(wxVERTICAL)
424            panelSizer.Add(self.panel, 1, wxEXPAND, 0)
425          self.SetAutoLayout(True)          self.SetAutoLayout(True)
426          self.SetSizer(topBox)          self.SetSizer(panelSizer)
427          topBox.Fit(self)          panelSizer.Fit(self)
428          topBox.SetSizeHints(self)          panelSizer.SetSizeHints(self)
429    
430          self.grid.SetFocus()          self.grid.SetFocus()
431    
432          EVT_BUTTON(self, ID_QUERY, self.OnQuery)          EVT_BUTTON(self, ID_QUERY, self.OnQuery)
433          EVT_BUTTON(self, ID_EXPORT, self.OnSaveAs)          EVT_BUTTON(self, ID_EXPORT, self.OnExport)
434            EVT_BUTTON(self, ID_EXPORTSEL, self.OnExportSel)
435            EVT_BUTTON(self, wxID_CLOSE, self.OnClose)
436          EVT_KEY_DOWN(self.grid, self.OnKeyDown)          EVT_KEY_DOWN(self.grid, self.OnKeyDown)
437    
438            self.grid.Subscribe(ROW_SELECTED, self.UpdateStatusText)
439    
440        def UpdateStatusText(self, rows=None):
441            self.SetStatusText(_("%i rows (%i selected), %i columns")
442                % (self.grid.GetNumberRows(),
443                   self.grid.GetNumberSelected(),
444                   self.grid.GetNumberCols()))
445    
446      def OnKeyDown(self, event):      def OnKeyDown(self, event):
447          """Catch query key from grid"""          """Catch query key from grid"""
448          if event.AltDown() and event.GetKeyCode() == ord(QUERY_KEY):          if event.AltDown() and event.GetKeyCode() == ord(QUERY_KEY):
# Line 390  class QueryTableFrame(TableFrame): Line 451  class QueryTableFrame(TableFrame):
451          else:          else:
452              event.Skip()              event.Skip()
453    
   
454      def OnQuery(self, event):      def OnQuery(self, event):
455          wxBeginBusyCursor()          ThubanBeginBusyCursor()
456          try:          try:
457    
458              if self.combo_value.GetSelection() < 1:              text = self.combo_value.GetValue()
459                  value = self.combo_value.GetValue()              if self.combo_value.GetSelection() < 1 \
460                    or self.combo_value.FindString(text) == -1:
461                    value = text
462              else:              else:
463                  value = self.table.Column(self.combo_value.GetValue())                  value = self.table.Column(text)
464    
465              ids = self.table.SimpleQuery(              ids = self.table.SimpleQuery(
466                      self.table.Column(self.combo_fields.GetStringSelection()),                      self.table.Column(self.combo_fields.GetStringSelection()),
# Line 418  class QueryTableFrame(TableFrame): Line 480  class QueryTableFrame(TableFrame):
480              # the first element, which causes everything to be              # the first element, which causes everything to be
481              # updated properly.              # updated properly.
482              #              #
483              self.grid.ToggleEventListeners(False)              if ids:
484                    self.grid.ToggleEventListeners(False)
485    
486              if choice == 0:              if choice == 0:
487                  # Replace Selection                  # Replace Selection
# Line 440  class QueryTableFrame(TableFrame): Line 503  class QueryTableFrame(TableFrame):
503                  if firsttime:                  if firsttime:
504                      firsttime = False                      firsttime = False
505                  else:                  else:
506                      self.grid.SelectRow(id, True)                      self.grid.SelectRowById(id, True)
507    
508              self.grid.ToggleEventListeners(True)              self.grid.ToggleEventListeners(True)
509    
# Line 448  class QueryTableFrame(TableFrame): Line 511  class QueryTableFrame(TableFrame):
511              # select the first row              # select the first row
512              #              #
513              if ids:              if ids:
514                  self.grid.SelectRow(ids[0], True)                  self.grid.SelectRowById(ids[0], True)
515    
516          finally:          finally:
517              wxEndBusyCursor()              ThubanEndBusyCursor()
518            
519        def doExport(self, onlySelected):
520                    
     def OnSaveAs(self, event):  
521          dlg = wxFileDialog(self, _("Export Table To"), ".", "",          dlg = wxFileDialog(self, _("Export Table To"), ".", "",
522                             _("DBF Files (*.dbf)|*.dbf|") +                             _("DBF Files (*.dbf)|*.dbf|") +
523                             _("CSV Files (*.csv)|*.csv|") +                             _("CSV Files (*.csv)|*.csv|") +
# Line 462  class QueryTableFrame(TableFrame): Line 527  class QueryTableFrame(TableFrame):
527              filename = dlg.GetPath()              filename = dlg.GetPath()
528              type = os.path.basename(filename).split('.')[-1:][0]              type = os.path.basename(filename).split('.')[-1:][0]
529              dlg.Destroy()              dlg.Destroy()
530    
531                if onlySelected:
532                    records = self.grid.GetSelectedRows()
533                else:
534                    records = None
535    
536              if type.upper() == "DBF":              if type.upper() == "DBF":
537                  table_to_dbf(self.table, filename)                  table_to_dbf(self.table, filename, records)
538              elif type.upper() == 'CSV':              elif type.upper() == 'CSV':
539                  table_to_csv(self.table, filename)                  table_to_csv(self.table, filename, records)
540              else:              else:
541                  dlg = wxMessageDialog(None, "Unsupported format: %s" % type,                  dlg = wxMessageDialog(None, "Unsupported format: %s" % type,
542                                        "Table Export", wxOK|wxICON_WARNING)                                        "Table Export", wxOK|wxICON_WARNING)
# Line 474  class QueryTableFrame(TableFrame): Line 545  class QueryTableFrame(TableFrame):
545          else:          else:
546              dlg.Destroy()              dlg.Destroy()
547    
548        def OnExport(self, event):
549            self.doExport(False)
550    
551        def OnExportSel(self, event):
552            self.doExport(True)
553    
554      def OnClose(self, event):      def OnClose(self, event):
555          TableFrame.OnClose(self, event)          TableFrame.OnClose(self, event)
556    
557      def get_selected(self):      def get_selected(self):
558          """Return a dictionary of the selected rows.          """Return a dictionary of the selected rows.
559            
560          The dictionary has sthe indexes as keys."""          The dictionary has the indexes as keys."""
561          return dict([(i, 0) for i in self.grid.GetSelectedRows()])          to_id = self.table.RowOrdinalToId
562            return dict([(to_id(i), 0) for i in self.grid.GetSelectedRows()])
563    
564    
565  class LayerTableFrame(QueryTableFrame):  class LayerTableFrame(QueryTableFrame):
566    
# Line 503  class LayerTableFrame(QueryTableFrame): Line 582  class LayerTableFrame(QueryTableFrame):
582          # accordingly          # accordingly
583          sel = self.get_selected().keys()          sel = self.get_selected().keys()
584          for i in sel:          for i in sel:
585              self.grid.SelectRow(i, True)              self.grid.SelectRowById(i, True)
586    
587      def make_grid(self, table):      def make_grid(self, table):
588          """Override the derived method to return a LayerTableGrid.          """Override the derived method to return a LayerTableGrid.
Legend:
Removed from v.1146  
changed lines
  Added in v.1662
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26