#!/usr/bin/env pythonw
"""
MultiStateButton class
"""

import wx
from wx.lib import buttons

class MultiStateButton(wx.lib.buttons.GenBitmapButton):
    def __init__(self, parent, ID, label="",
                 pos = wx.DefaultPosition, size = wx.DefaultSize,
                 style = 0, validator = wx.DefaultValidator,
                 name = "multistatebutton", multibitmaps = ()):
        wx.GenButton.__init__(self, parent, ID, label, pos, size, style, validator, name)
        self.AddStates(multibitmaps)

    def AddStates(multibitmaps):
        pass

class App(wx.App):
    """Application class."""

    def OnInit(self):
        self.frame = wx.Frame(title='Testing MultiStateButton')
        box = wx.BoxSizer(wx.VERTICAL)
        box.Add(MultiStateButton(self.frame, 1010, multibitmaps=['button1', 'button2']), 0, wx.EXPAND)
        box.Add(MultiStateButton(self.frame, 1010, multibitmaps=['button3', 'button4']), 0, wx.EXPAND)
        self.frame.SetSizer(box)
        self.frame.Show()
        self.SetTopWindow(self.frame)
        return True

def main():
    app = App()
    app.MainLoop()
    print "All done"

if __name__ == '__main__':
    main()