[wxWidgets.git] / wxPython / demo / GridDragable.py

from wxPython.wx import *
from wxPython.grid import *
from wxPython.lib.gridmovers import wxGridColMover, EVT_GRID_COL_MOVE
from wxPython.lib.gridmovers import wxGridRowMover, EVT_GRID_ROW_MOVE

#---------------------------------------------------------------------------

class CustomDataTable(wxPyGridTableBase):
    """
    """

    def __init__(self, log):
        wxPyGridTableBase.__init__(self)
        self.log = log

        self.identifiers = ['id','ds','sv','pr','pl','op','fx','ts']

        self.rowLabels = ['Row1','Row2','Row3']

        self.colLabels = {'id':'ID','ds':'Description','sv':'Severity',
                          'pr':'Priority','pl':'Platform','op':'Opened?',
                          'fx':'Fixed?','ts':'Tested?'}

        self.data = [{'id':1010,
                      'ds':"The foo doesn't bar",
                      'sv':"major",
                      'pr':1,
                      'pl':'MSW',
                      'op':1,
                      'fx':1,
                      'ts':1
                      },
                     {'id':1011,
                      'ds':"I've got a wicket in my wocket",
                      'sv':"wish list",
                      'pr':2,
                      'pl':'other',
                      'op':0,
                      'fx':0,
                      'ts':0
                      },
                     {'id':1012,
                      'ds':"Rectangle() returns a triangle",
                      'sv':"critical",
                      'pr':5,
                      'pl':'all',
                      'op':0,
                      'fx':0,
                      'ts':0
                      }
                     ]

    #--------------------------------------------------
    # required methods for the wxPyGridTableBase interface

    def GetNumberRows(self):
        return len(self.data)

    def GetNumberCols(self):
        return len(self.identifiers)

    def IsEmptyCell(self, row, col):
        id = self.identifiers[col]
        return not self.data[row][id]

    def GetValue(self, row, col):
        id = self.identifiers[col]
        return self.data[row][id]

    def SetValue(self, row, col, value):
        id = self.identifiers[col]
        self.data[row][id] = value

    #--------------------------------------------------
    # Some optional methods

    # Called when the grid needs to display column labels
    def GetColLabelValue(self, col):
        id = self.identifiers[col]
        return self.colLabels[id]

    # Called when the grid needs to display row labels
    def GetRowLabelValue(self,row):
        return self.rowLabels[row]

    #--------------------------------------------------
    # Methods added for demo purposes.

    # The physical moving of the cols/rows is left to the implementer.
    # Because of the dynamic nature of a wxGrid the physical moving of
    # columns differs from implementation to implementation

    # Move the column
    def MoveColumn(self,frm,to):
        grid = self.GetView()
        if grid:
            # Move the identifiers
            old = self.identifiers[frm]
            del self.identifiers[frm]
            if to > frm:
                self.identifiers.insert(to-1,old)
            else:
                self.identifiers.insert(to,old)

            # Notify the grid
            grid.BeginBatch()
            msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_COLS_DELETED,
                                     frm,1)
            grid.ProcessTableMessage(msg)
            msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_COLS_INSERTED,
                                     to,1)
            grid.ProcessTableMessage(msg)
            grid.EndBatch()

    # Move the row
    def MoveRow(self,frm,to):
        grid = self.GetView()
        if grid:
            # Move the rowLabels and data rows
            oldLabel = self.rowLabels[frm]
            oldData = self.data[frm]
            del self.rowLabels[frm]
            del self.data[frm]
            if to > frm:
                self.rowLabels.insert(to-1,oldLabel)
                self.data.insert(to-1,oldData)
            else:
                self.rowLabels.insert(to,oldLabel)
                self.data.insert(to,oldData)

            # Notify the grid
            grid.BeginBatch()
            msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_ROWS_DELETED,
                                     frm,1)
            grid.ProcessTableMessage(msg)
            msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_ROWS_INSERTED,
                                     to,1)
            grid.ProcessTableMessage(msg)
            grid.EndBatch()


#---------------------------------------------------------------------------


class DragableGrid(wxGrid):
    def __init__(self, parent, log):
        wxGrid.__init__(self, parent, -1)

        table = CustomDataTable(log)

        # The second parameter means that the grid is to take ownership of the
        # table and will destroy it when done.  Otherwise you would need to keep
        # a reference to it and call it's Destroy method later.
        self.SetTable(table, True)

        # Enable Column moving
        wxGridColMover(self)
        EVT_GRID_COL_MOVE(self,self.GetId(),self.OnColMove)

        # Enable Row moving
        wxGridRowMover(self)
        EVT_GRID_ROW_MOVE(self,self.GetId(),self.OnRowMove)

    # Event method called when a column move needs to take place
    def OnColMove(self,evt):
        frm = evt.GetMoveColumn()       # Column being moved
        to = evt.GetBeforeColumn()      # Before which column to insert
        self.GetTable().MoveColumn(frm,to)

    # Event method called when a row move needs to take place
    def OnRowMove(self,evt):
        frm = evt.GetMoveRow()          # Row being moved
        to = evt.GetBeforeRow()         # Before which row to insert
        self.GetTable().MoveRow(frm,to)

#---------------------------------------------------------------------------

class TestFrame(wxFrame):
    def __init__(self, parent, log):
        wxFrame.__init__(self, parent, -1, "Custom Table, data driven Grid  Demo", size=(640,480))
        grid = DragableGrid(self, log)


#---------------------------------------------------------------------------

if __name__ == '__main__':
    import sys
    app = wxPySimpleApp()
    frame = TestFrame(None, sys.stdout)
    frame.Show(True)
    app.MainLoop()

#---------------------------------------------------------------------------
Commit	Line	Data
1e4a197e RD	1	from wxPython.wx import *
	2	from wxPython.grid import *
	3	from wxPython.lib.gridmovers import wxGridColMover, EVT_GRID_COL_MOVE
	4	from wxPython.lib.gridmovers import wxGridRowMover, EVT_GRID_ROW_MOVE
	5
	6	#---------------------------------------------------------------------------
	7
	8	class CustomDataTable(wxPyGridTableBase):
	9	"""
	10	"""
	11
	12	def __init__(self, log):
	13	wxPyGridTableBase.__init__(self)
	14	self.log = log
	15
	16	self.identifiers = ['id','ds','sv','pr','pl','op','fx','ts']
	17
	18	self.rowLabels = ['Row1','Row2','Row3']
	19
	20	self.colLabels = {'id':'ID','ds':'Description','sv':'Severity',
	21	'pr':'Priority','pl':'Platform','op':'Opened?',
	22	'fx':'Fixed?','ts':'Tested?'}
	23
	24	self.data = [{'id':1010,
	25	'ds':"The foo doesn't bar",
	26	'sv':"major",
	27	'pr':1,
	28	'pl':'MSW',
	29	'op':1,
	30	'fx':1,
	31	'ts':1
	32	},
	33	{'id':1011,
	34	'ds':"I've got a wicket in my wocket",
	35	'sv':"wish list",
	36	'pr':2,
	37	'pl':'other',
	38	'op':0,
	39	'fx':0,
	40	'ts':0
	41	},
	42	{'id':1012,
	43	'ds':"Rectangle() returns a triangle",
	44	'sv':"critical",
	45	'pr':5,
	46	'pl':'all',
	47	'op':0,
	48	'fx':0,
	49	'ts':0
	50	}
	51	]
	52
	53	#--------------------------------------------------
	54	# required methods for the wxPyGridTableBase interface
	55
	56	def GetNumberRows(self):
	57	return len(self.data)
	58
	59	def GetNumberCols(self):
	60	return len(self.identifiers)
	61
	62	def IsEmptyCell(self, row, col):
	63	id = self.identifiers[col]
	64	return not self.data[row][id]
65
66	def GetValue(self, row, col):
67	id = self.identifiers[col]
68	return self.data[row][id]
69
70	def SetValue(self, row, col, value):
71	id = self.identifiers[col]
72	self.data[row][id] = value
73
74	#--------------------------------------------------
75	# Some optional methods
76
77	# Called when the grid needs to display column labels
78	def GetColLabelValue(self, col):
79	id = self.identifiers[col]
80	return self.colLabels[id]
81
82	# Called when the grid needs to display row labels
83	def GetRowLabelValue(self,row):
84	return self.rowLabels[row]
85
86	#--------------------------------------------------
87	# Methods added for demo purposes.
88
89	# The physical moving of the cols/rows is left to the implementer.
90	# Because of the dynamic nature of a wxGrid the physical moving of
91	# columns differs from implementation to implementation
92
93	# Move the column
94	def MoveColumn(self,frm,to):
95	grid = self.GetView()
96	if grid:
97	# Move the identifiers
98	old = self.identifiers[frm]
99	del self.identifiers[frm]
100	if to > frm:
101	self.identifiers.insert(to-1,old)
102	else:
103	self.identifiers.insert(to,old)
104
105	# Notify the grid
106	grid.BeginBatch()
107	msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_COLS_DELETED,
108	frm,1)
109	grid.ProcessTableMessage(msg)
110	msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_COLS_INSERTED,
111	to,1)
112	grid.ProcessTableMessage(msg)
113	grid.EndBatch()
114
115	# Move the row
116	def MoveRow(self,frm,to):
117	grid = self.GetView()
118	if grid:
119	# Move the rowLabels and data rows
120	oldLabel = self.rowLabels[frm]
121	oldData = self.data[frm]
122	del self.rowLabels[frm]
123	del self.data[frm]
124	if to > frm:
125	self.rowLabels.insert(to-1,oldLabel)
126	self.data.insert(to-1,oldData)
127	else:
128	self.rowLabels.insert(to,oldLabel)
129	self.data.insert(to,oldData)
130
131	# Notify the grid
132	grid.BeginBatch()
133	msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_ROWS_DELETED,
134	frm,1)
135	grid.ProcessTableMessage(msg)
136	msg = wxGridTableMessage(self,wxGRIDTABLE_NOTIFY_ROWS_INSERTED,
137	to,1)
138	grid.ProcessTableMessage(msg)
139	grid.EndBatch()
140
141
142	#---------------------------------------------------------------------------
143
144
145	class DragableGrid(wxGrid):
146	def __init__(self, parent, log):
147	wxGrid.__init__(self, parent, -1)
148
149	table = CustomDataTable(log)
150
151	# The second parameter means that the grid is to take ownership of the
152	# table and will destroy it when done. Otherwise you would need to keep
153	# a reference to it and call it's Destroy method later.
154	self.SetTable(table, True)
155
156	# Enable Column moving
157	wxGridColMover(self)
158	EVT_GRID_COL_MOVE(self,self.GetId(),self.OnColMove)
159
160	# Enable Row moving
161	wxGridRowMover(self)
162	EVT_GRID_ROW_MOVE(self,self.GetId(),self.OnRowMove)
163
164	# Event method called when a column move needs to take place
165	def OnColMove(self,evt):
166	frm = evt.GetMoveColumn() # Column being moved
167	to = evt.GetBeforeColumn() # Before which column to insert
168	self.GetTable().MoveColumn(frm,to)
169
170	# Event method called when a row move needs to take place
171	def OnRowMove(self,evt):
172	frm = evt.GetMoveRow() # Row being moved
173	to = evt.GetBeforeRow() # Before which row to insert
174	self.GetTable().MoveRow(frm,to)
175
176	#---------------------------------------------------------------------------
177
178	class TestFrame(wxFrame):
179	def __init__(self, parent, log):
180	wxFrame.__init__(self, parent, -1, "Custom Table, data driven Grid Demo", size=(640,480))
181	grid = DragableGrid(self, log)
182
183
184	#---------------------------------------------------------------------------
185
186	if __name__ == '__main__':
187	import sys
188	app = wxPySimpleApp()
189	frame = TestFrame(None, sys.stdout)
190	frame.Show(True)
191	app.MainLoop()
192
193	#---------------------------------------------------------------------------