# -*- coding: cp936 -*- import wx import wx.grid import wx.lib.gridmovers as gridmovers import pymssql connect=pymssql.connect(host='wxpython',user='sa',password='',database='THIS4_0807') cursor=connect.cursor() ascordesc=True class LineupTable(wx.grid.PyGridTableBase): def __init__(self,data,fields): wx.grid.PyGridTableBase.__init__(self) self.data=data self.fields=fields ## self.dataTypes = [wx.grid.GRID_VALUE_STRING, ## wx.grid.GRID_VALUE_STRING, ## #gridlib.GRID_VALUE_CHOICE + ':only in a million years!,wish list,minor,normal,major,critical', ## #gridlib.GRID_VALUE_NUMBER + ':1,5', ## #gridlib.GRID_VALUE_CHOICE + ':all,MSW,GTK,other', ## #gridlib.GRID_VALUE_BOOL, ## #gridlib.GRID_VALUE_BOOL, ## #gridlib.GRID_VALUE_BOOL, ## wx.grid.GRID_VALUE_FLOAT + ':6,2', ## ] #---Grid cell attributes self.odd = wx.grid.GridCellAttr() self.odd.SetBackgroundColour("grey") self.odd.SetFont(wx.Font(8, wx.SWISS, wx.NORMAL, wx.BOLD)) self.even = wx.grid.GridCellAttr() self.even.SetBackgroundColour("white") self.even.SetFont(wx.Font(8, wx.SWISS, wx.NORMAL, wx.BOLD)) #---Mandatory constructors for grid def GetNumberRows(self): # if len(self.data)<10: # rowcounts=10 #else: #rowcounts=len(self.data) return len(self.data) def GetNumberCols(self): return len(self.fields) def GetColLabelValue(self, col): return self.fields[col] def IsEmptyCell(self, row, col): if self.data[row][col] == "" or self.data[row][col] is None: return True else: return False # def GetValue(self, row, col): # value = self.data[row][col] # if value is not None: # return value # else: # return '' # # def SetValue(self, row, col, value): # #print col # def innerSetValue(row, col, value): # try: # self.data[row][col] = value # except IndexError: # # add a new row # self.data.append([''] * self.GetNumberCols()) # innerSetValue(row, col, value) # # # tell the grid we've added a row # msg = gridlib.GridTableMessage(self, # The table # gridlib.GRIDTABLE_NOTIFY_ROWS_APPENDED, # what we did to it # 1 # how many # ) # # self.GetView().ProcessTableMessage(msg) # innerSetValue(row, col, value) def GetValue(self, row, col): #id = self.fields[col] return self.data[row][col] def SetValue(self, row, col, value): #id = self.fields[col] self.data[row][col] = value #-------------------------------------------------- # Some optional methods # Called when the grid needs to display column labels # def GetColLabelValue(self, col): # #id = self.fields[col] # return self.fields[col][0] def GetAttr(self, row, col, kind): attr = [self.even, self.odd][row % 2] attr.IncRef() return attr def SortColumn(self, col,ascordesc): """ col -> sort the data based on the column indexed by col """ name = self.fields[col] _data = [] for row in self.data: #print row #rowname, entry = row _data.append((row[col], row)) _data.sort(reverse=ascordesc) self.data = [] for sortvalue, row in _data: self.data.append(row) def AppendRow(self, row):#增加行 #print 'append' entry = [] for name in self.fields: entry.append('A') self.data.append(tuple(entry )) return True def MoveColumn(self,frm,to): grid = self.GetView() if grid: # Move the identifiers old = self.fields[frm] del self.fields[frm] if to > frm: self.fields.insert(to-1,old) else: self.fields.insert(to,old) print self.fields # Notify the grid grid.BeginBatch() msg = wx.grid.GridTableMessage( self, wx.grid.GRIDTABLE_NOTIFY_COLS_INSERTED, to, 1 ) grid.ProcessTableMessage(msg) msg = wx.grid.GridTableMessage( self, wx.grid.GRIDTABLE_NOTIFY_COLS_DELETED, frm, 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 = wx.grid.GridTableMessage( self, wx.grid.GRIDTABLE_NOTIFY_ROWS_INSERTED, to, 1 ) grid.ProcessTableMessage(msg) msg = wx.grid.GridTableMessage( self, wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, frm, 1 ) grid.ProcessTableMessage(msg) grid.EndBatch() class DragableGrid(wx.grid.Grid): def __init__(self, parent): wx.grid.Grid.__init__(self, parent ) cursor.execute('select top 2 blh,hzxm,qrrq from JK_YSQR_DK ') data = cursor.fetchall() fields = [cursor.description[i][0] for i in range(len(cursor.description))] table = LineupTable(data,fields) self.SetTable(table, True) #table = LineupTable() # 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(self.table, True) # Enable Column moving gridmovers.GridColMover(self) self.Bind(gridmovers.EVT_GRID_COL_MOVE, self.OnColMove, self) # Enable Row moving gridmovers.GridRowMover(self) self.Bind(gridmovers.EVT_GRID_ROW_MOVE, self.OnRowMove, self) # 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 MyFrame(wx.Frame): def __init__(self): wx.Frame.__init__(self, parent=None, id=-1, title='wx.grid.PyGridTableBase',size=(900,600)) #---Panel #panel = wx.Panel(self, -1) #---Buttons self.btn_1hr = wx.Button(self, -1, "RUN", pos=(10, 10),size=(100,40)) self.Bind(wx.EVT_BUTTON, self.OnClick, self.btn_1hr) self.btn_2hr = wx.Button(self, -1, "ADD", pos=(10, 35),size=(100,40)) self.Bind(wx.EVT_BUTTON, self.OnADD, self.btn_2hr) self.btn_3hr = wx.Button(self, -1, "DELETE", pos=(10, 60),size=(100,40)) self.btn_4hr = wx.Button(self, -1, "INSERT", pos=(10, 85),size=(100,40)) box = wx.BoxSizer(wx.VERTICAL) box.Add(self.btn_1hr, 1, wx.EXPAND,5) box.Add(self.btn_2hr, 1, wx.EXPAND,5) box.Add(self.btn_3hr, 1, wx.EXPAND,5) box.Add(self.btn_4hr, 1, wx.EXPAND,5) #---Grid self.grid =DragableGrid(self)#, pos=(140, 0), size=(900,400)) self.grid.SetRowLabelSize(40)#设置行标签的宽度 self.Bind(wx.grid.EVT_GRID_LABEL_RIGHT_CLICK, self.OnLabelRightClick) #self.Bind(wx.grid.EVT_GRID_RANGE_SELECT, self.OnRangeSelect) box1 = wx.BoxSizer(wx.VERTICAL) box1.Add(self.grid, 1, wx.GROW|wx.ALL) cursor.execute('select top 2 blh,hzxm,qrrq from JK_YSQR_DK ') data = cursor.fetchall() fields = [cursor.description[i][0] for i in range(len(cursor.description))] self.table = LineupTable(data,fields) self.grid.SetTable(self.table, True) #---Grid properties self.grid.EnableEditing(True)#是否可以编辑 self.grid.SetDefaultCellAlignment(wx.ALIGN_LEFT, wx.ALIGN_RIGHT)#设置CELL的文本对齐方式 self.grid.SetSelectionBackground('red') self.grid.EnableDragColSize(enable=True)#控制列宽是否可以拉动 self.grid.EnableDragRowSize(enable=True)#控制行高是否可以拉动 self.grid.SetLabelBackgroundColour((100, 200, 150)) self.grid.SetLabelTextColour((255, 255, 255)) #---Column Sizes self.grid.AdjustScrollbars() self.grid.Refresh() Hbox = wx.BoxSizer(wx.HORIZONTAL) Hbox.Add(box, 0, wx.EXPAND) Hbox.Add(box1, 1, wx.EXPAND) #Vbox = wx.BoxSizer(wx.VERTICAL) #Vbox.Add(Hbox,0,wx.ALL|wx.EXPAND) self.SetSizer(Hbox) #self.Fit() #self.grid.AutoSize() #---Use below if want to size individual columns (index, size) #---Also have SetRowSize #grid.SetColSize(0, 150) def Reset(self): """reset the view based on the data in the table. Call this when rows are added or destroyed""" self.table.ResetView(self) def OnLabelRightClick(self, event): global ascordesc,row, col #self.SetStatusText('You Have Right-Clicked On Label "%s"!' % event.GetString()) row, col = event.GetRow(), event.GetCol() #print row, col self.table.SortColumn(col,ascordesc) if ascordesc: ascordesc=False else: ascordesc=True self.grid.Refresh() def OnADD(self,event): #self.table.AppendRow(row) #print (self.grid.SelectedRows) self.table.AppendRow(row) #self.grid.SetTable(self.table, True) self.grid.ForceRefresh() def OnClick(self, event): cursor.execute('select top 5 id as "编码",name as "名称" ,cast(memo as numeric(12,2)) as "单价" from YY_SFDXMK where memo>0') data1 = cursor.fetchall() fields1 = [cursor.description[i][0] for i in range(len(cursor.description))] self.table = LineupTable(data1,fields1) self.grid.SetTable(self.table, True) self.grid.EnableDragColSize(enable=True) self.grid.EnableDragRowSize(enable=True) #self.grid.AutoSize() self.grid.AdjustScrollbars() #self.grid.ForceRefresh() self.grid.Refresh() if __name__ == "__main__": app = wx.PySimpleApp() frame = MyFrame() frame.Show() app.MainLoop()