wxPython/demo/wxListCtrl.py

   1 #!/bin/env python
   2 #----------------------------------------------------------------------------
   3 # Name:         ListCtrl.py
   4 # Purpose:      Testing lots of stuff, controls, window types, etc.
   5 #
   6 # Author:       Robin Dunn & Gary Dumer
   7 #
   8 # Created:
   9 # RCS-ID:       $Id$
  10 # Copyright:    (c) 1998 by Total Control Software
  11 # Licence:      wxWindows license
  12 #----------------------------------------------------------------------------
  13
  14 from wxPython.wx import *
  15
  16 #---------------------------------------------------------------------------
  17
  18 musicdata = {
  19 1 : ("Bad English", "The Price Of Love", "Rock"),
  20 2 : ("DNA featuring Suzanne Vega", "Tom's Diner", "Rock"),
  21 3 : ("George Michael", "Praying For Time", "Rock"),
  22 4 : ("Gloria Estefan", "Here We Are", "Rock"),
  23 5 : ("Linda Ronstadt", "Don't Know Much", "Rock"),
  24 6 : ("Michael Bolton", "How Am I Supposed To Live Without You", "Blues"),
  25 ## 7 : ("Paul Young", "Oh Girl", "Rock"),
  26 ## 8 : ("Paula Abdul", "Opposites Attract", "Rock"),
  27 ## 9 : ("Richard Marx", "Should've Known Better", "Rock"),
  28 ## 10: ("Rod Stewart", "Forever Young", "Rock"),
  29 ## 11: ("Roxette", "Dangerous", "Rock"),
  30 ## 12: ("Sheena Easton", "The Lover In Me", "Rock"),
  31 ## 13: ("Sinead O'Connor", "Nothing Compares 2 U", "Rock"),
  32 ## 14: ("Stevie B.", "Because I Love You", "Rock"),
  33 ## 15: ("Taylor Dayne", "Love Will Lead You Back", "Rock"),
  34 ## 16: ("The Bangles", "Eternal Flame", "Rock"),
  35 ## 17: ("Wilson Phillips", "Release Me", "Rock"),
  36 ## 18: ("Billy Joel", "Blonde Over Blue", "Rock"),
  37 ## 19: ("Billy Joel", "Famous Last Words", "Rock"),
  38 ## 20: ("Billy Joel", "Lullabye (Goodnight, My Angel)", "Rock"),
  39 ## 21: ("Billy Joel", "The River Of Dreams", "Rock"),
  40 ## 22: ("Billy Joel", "Two Thousand Years", "Rock"),
  41 ## 23: ("Janet Jackson", "Alright", "Rock"),
  42 ## 24: ("Janet Jackson", "Black Cat", "Rock"),
  43 ## 25: ("Janet Jackson", "Come Back To Me", "Rock"),
  44 ## 26: ("Janet Jackson", "Escapade", "Rock"),
  45 ## 27: ("Janet Jackson", "Love Will Never Do (Without You)", "Rock"),
  46 ## 28: ("Janet Jackson", "Miss You Much", "Rock"),
  47 ## 29: ("Janet Jackson", "Rhythm Nation", "Rock"),
  48 ## 30: ("Janet Jackson", "State Of The World", "Rock"),
  49 ## 31: ("Janet Jackson", "The Knowledge", "Rock"),
  50 ## 32: ("Spyro Gyra", "End of Romanticism", "Jazz"),
  51 ## 33: ("Spyro Gyra", "Heliopolis", "Jazz"),
  52 ## 34: ("Spyro Gyra", "Jubilee", "Jazz"),
  53 ## 35: ("Spyro Gyra", "Little Linda", "Jazz"),
  54 ## 36: ("Spyro Gyra", "Morning Dance", "Jazz"),
  55 ## 37: ("Spyro Gyra", "Song for Lorraine", "Jazz"),
  56 ## 38: ("Yes", "Owner Of A Lonely Heart", "Rock"),
  57 ## 39: ("Yes", "Rhythm Of Love", "Rock"),
  58 }
  59
  60 import images
  61
  62 class TestListCtrlPanel(wxPanel):
  63     def __init__(self, parent, log):
  64         wxPanel.__init__(self, parent, -1, style=wxWANTS_CHARS)
  65
  66         self.log = log
  67         tID = wxNewId()
  68
  69         self.il = wxImageList(16, 16)
  70         bmp = images.getSmilesBitmap()
  71         #idx1 = self.il.AddWithColourMask(bmp, wxWHITE)
  72         idx1 = self.il.Add(bmp)
  73
  74         self.list = wxListCtrl(self, tID,
  75                                style=wxLC_REPORT|wxSUNKEN_BORDER)#|wxLC_VRULES|wxLC_HRULES)
  76         self.list.SetImageList(self.il, wxIMAGE_LIST_SMALL)
  77
  78         #  Why doesn't this show up on MSW???
  79         self.list.SetToolTip(wxToolTip("This is a ToolTip!"))
  80
  81         self.list.InsertColumn(0, "Artist")
  82         self.list.InsertColumn(1, "Title", wxLIST_FORMAT_RIGHT)
  83         self.list.InsertColumn(2, "Genre")
  84         items = musicdata.items()
  85         for x in range(len(items)):
  86             key, data = items[x]
  87             self.list.InsertImageStringItem(x, data[0], idx1)
  88             self.list.SetStringItem(x, 1, data[1])
  89             self.list.SetStringItem(x, 2, data[2])
  90             self.list.SetItemData(x, key)
  91
  92         self.list.SetColumnWidth(0, wxLIST_AUTOSIZE)
  93         self.list.SetColumnWidth(1, wxLIST_AUTOSIZE)
  94         ##self.list.SetColumnWidth(2, wxLIST_AUTOSIZE)
  95
  96         self.list.SetItemState(5, wxLIST_STATE_SELECTED, wxLIST_STATE_SELECTED)
  97
  98         #self.list.SetItemState(25, wxLIST_STATE_SELECTED, wxLIST_STATE_SELECTED)
  99         #self.list.EnsureVisible(25)
 100
 101         # show how to change the colour of an item
 102         item = self.list.GetItem(1)
 103         item.SetTextColour(wxBLUE)
 104         self.list.SetItem(item)
 105
 106         item = self.list.GetItem(4)
 107         item.SetTextColour(wxRED)
 108         self.list.SetItem(item)
 109
 110         self.currentItem = 0
 111         EVT_SIZE(self, self.OnSize)
 112         EVT_LIST_ITEM_SELECTED(self, tID, self.OnItemSelected)
 113         EVT_LIST_ITEM_ACTIVATED(self, tID, self.OnItemActivated)
 114         EVT_LIST_DELETE_ITEM(self, tID, self.OnItemDelete)
 115         EVT_LIST_COL_CLICK(self, tID, self.OnColClick)
 116         EVT_LEFT_DCLICK(self.list, self.OnDoubleClick)
 117         EVT_RIGHT_DOWN(self.list, self.OnRightDown)
 118
 119         # for wxMSW
 120         EVT_COMMAND_RIGHT_CLICK(self.list, tID, self.OnRightClick)
 121
 122         # for wxGTK
 123         EVT_RIGHT_UP(self.list, self.OnRightClick)
 124
 125
 126     def OnRightDown(self, event):
 127         self.x = event.GetX()
 128         self.y = event.GetY()
 129         self.log.WriteText("x, y = %s\n" % str((self.x, self.y)))
 130         event.Skip()
 131
 132
 133     def getColumnText(self, index, col):
 134         item = self.list.GetItem(index, col)
 135         return item.GetText()
 136
 137
 138     def OnItemSelected(self, event):
 139         self.currentItem = event.m_itemIndex
 140         self.log.WriteText("OnItemSelected: %s, %s, %s, %s\n" %
 141                            (self.currentItem,
 142                             self.list.GetItemText(self.currentItem),
 143                             self.getColumnText(self.currentItem, 1),
 144                             self.getColumnText(self.currentItem, 2)))
 145         if self.currentItem == 10:
 146             self.log.WriteText("OnItemSelected: Veto'd selection\n")
 147             #event.Veto()  # doesn't work
 148             # this does
 149             self.list.SetItemState(10, 0, wxLIST_STATE_SELECTED)
 150
 151     def OnItemActivated(self, event):
 152         self.currentItem = event.m_itemIndex
 153         self.log.WriteText("OnItemActivated: %s\n" % self.list.GetItemText(self.currentItem))
 154
 155     def OnItemDelete(self, event):
 156         self.log.WriteText("OnItemDelete\n")
 157
 158     def OnColClick(self, event):
 159         self.log.WriteText("OnColClick: %d\n" % event.m_col)
 160         self.col = event.m_col
 161         self.list.SortItems(self.ColumnSorter)
 162
 163     def ColumnSorter(self, key1, key2):
 164         item1 = musicdata[key1][self.col]
 165         item2 = musicdata[key2][self.col]
 166
 167         # when the items are identical, compare someting else to make the sort key unique...
 168         if item1 == item2:  return key1 - key2
 169         elif item1 < item2: return -1
 170         else:               return 1
 171
 172
 173     def OnDoubleClick(self, event):
 174         self.log.WriteText("OnDoubleClick item %s\n" % self.list.GetItemText(self.currentItem))
 175         event.Skip()
 176
 177     def OnRightClick(self, event):
 178         self.log.WriteText("OnRightClick %s\n" % self.list.GetItemText(self.currentItem))
 179         menu = wxMenu()
 180         tPopupID1 = 0
 181         tPopupID2 = 1
 182         tPopupID3 = 2
 183         tPopupID4 = 3
 184         tPopupID5 = 5
 185         #menu.Append(tPopupID1, "One")
 186         item = wxMenuItem(menu, tPopupID1,"One")
 187         item.SetBitmap(images.getSmilesBitmap())
 188         menu.AppendItem(item)
 189         menu.Append(tPopupID2, "Two")
 190         menu.Append(tPopupID3, "Three")
 191         menu.Append(tPopupID4, "DeleteAllItems")
 192         menu.Append(tPopupID5, "GetItem")
 193         EVT_MENU(self, tPopupID1, self.OnPopupOne)
 194         EVT_MENU(self, tPopupID2, self.OnPopupTwo)
 195         EVT_MENU(self, tPopupID3, self.OnPopupThree)
 196         EVT_MENU(self, tPopupID4, self.OnPopupFour)
 197         EVT_MENU(self, tPopupID5, self.OnPopupFive)
 198         self.PopupMenu(menu, wxPoint(self.x, self.y))
 199         menu.Destroy()
 200         event.Skip()
 201
 202     def OnPopupOne(self, event):
 203         self.log.WriteText("Popup one\n")
 204
 205     def OnPopupTwo(self, event):
 206         self.log.WriteText("Popup two\n")
 207
 208     def OnPopupThree(self, event):
 209         self.log.WriteText("Popup three\n")
 210
 211     def OnPopupFour(self, event):
 212         self.list.DeleteAllItems()
 213
 214     def OnPopupFive(self, event):
 215         item = self.list.GetItem(self.currentItem)
 216         print item.m_text, item.m_itemId, self.list.GetItemData(self.currentItem)
 217
 218     def OnSize(self, event):
 219         w,h = self.GetClientSizeTuple()
 220         self.list.SetDimensions(0, 0, w, h)
 221
 222
 223
 224
 225
 226
 227 #---------------------------------------------------------------------------
 228
 229 def runTest(frame, nb, log):
 230     win = TestListCtrlPanel(nb, log)
 231     return win
 232
 233 #---------------------------------------------------------------------------
 234
 235
 236
 237
 238
 239
 240
 241
 242
 243
 244
 245
 246
 247
 248
 249
 250 overview = """\
 251 A list control presents lists in a number of formats: list view, report view, icon view and small icon view. Elements are numbered from zero.
 252
 253 """