demoMenu.py

'''
    Demonstrates the menu manager API.
'''
import MaxPlus

def outputMenuItem(item, recurse = True, indent = ''):
    text = item.GetTitle() 
    print indent, text if text else "----"
    if item.HasSubMenu and recurse:
        outputMenu(item.SubMenu, recurse, indent + '   ')
    
def outputMenu(menu, recurse = True, indent = ''):
    for i in menu.Items:
        outputMenuItem(i, recurse, indent)
  
somethingHappened = False
def doSomething():
    global somethingHappened 
    somethingHappened = True
    print 'Something happened'
    
action = MaxPlus.ActionFactory.Create('Do something', 'Python demos', doSomething)

def createTestMenu(name):
    if not MaxPlus.MenuManager.MenuExists(name):
        mb = MaxPlus.MenuBuilder(name)
        if action._IsValidWrapper():
            print "Created action"
        else:
            print "Failed to create action"
        mb.AddItem(action)
        mb.AddSeparator()
        menu = mb.Create(MaxPlus.MenuManager.GetMainMenu())
        print 'menu created', menu.Title 
    else:
        print 'The menu ', name, ' already exists'

def getLastMenuItem(menu = MaxPlus.MenuManager.GetMainMenu()):
    return list(menu.Items)[-1]
    
def testLastItem(text):
    assert(getLastMenuItem().Title  == text)

def main():
    print "Removing any previously left 'menu items'"
    MaxPlus.MenuManager.UnregisterMenu(u"Test")
    
    print "Creating a new menu"
    testLastItem(u"&Help")
    outputMenu(MaxPlus.MenuManager.GetMainMenu(), False)
    
    print "Creating a new menu"
    createTestMenu(u"Test")
    outputMenu(MaxPlus.MenuManager.GetMainMenu(), False)
    testLastItem(u"Test")
    
    assert(not somethingHappened)
    mi = getLastMenuItem()
    mi = list(mi.SubMenu.Items)[0] 
    ai = mi.ActionItem
    ai.Execute()
    assert(somethingHappened)    

    print "Unregistering the 'test' menu"
    MaxPlus.MenuManager.UnregisterMenu(u"Test")
    outputMenu(MaxPlus.MenuManager.GetMainMenu(), False)
    testLastItem(u"&Help")

if __name__ == '__main__':
    main()