"""MultiStateButton class@copyright: Copyright (c) 2006 Open Source Applications Foundation@license: http://osafoundation.org/Chandler_0.1_license_terms.htm"""import wxclass MultiStateButton(wx.lib.buttons.GenBitmapButton): """ A bitmap button that has more than one possible bitmap in its default state. These bitmaps are passed in as a list of names, which are also the names of the PNG bitmaps. foo = MultiStateButton(view, 100, multibitmaps=["MailState", "NoMailState"]) will create a button """ kBitmapFileExtensions = ["png", "gif", "jpg", "tiff", "psd"] def __init__(self, parent, ID, label="", pos = wx.DefaultPosition, size = wx.DefaultSize, style = 0, validator = wx.DefaultValidator, name = "multistatebutton", multibitmaps = ()): firstBitmap = (len(multibitmaps) > 0 and multibitmaps[0] or "") super(wx.lib.buttons.GenBitmapButton, self).__init__(self, parent, ID, firstBitmap, pos, size, style, validator, name) self.AddStates(multibitmaps) def AddStates(multibitmaps): for name in multibitmaps: found = False for ext in kBitmapFileExtensions: try: img = wx.Image('/Users/rae/wokr/osaf/rae-button/' + name + '.' + ext) self.bits[name] = img.ConvertToBitmap() Found = True break except IOError: pass if not found: ##Reid # should raise an error here somehow pass def SetState(state): bitmap = self.bits.get(state) if bitmap != None: self.SetBitmapLabel(bitmap)theWindow = wx.Frame(parent=None, id=-1, title="MultiButton testing")box = wx.BoxSizer(wx.Horizontal)box.Add(MultiStateButton(theWindow, 1010, multibitmaps=['button1', 'button2']))box.Add(MultiStateButton(theWindow, 1010, multibitmaps=['button3', 'button4']))box.Add(MultiStateButton(theWindow, 1010, multibitmaps=['button1', 'button2', 'button3', 'button4', 'button5']))theWindow.Show()