TESTING-NewTree.py

#!/usr/bin/env python
RETURN = []
import npyscreen
class TestApp(npyscreen.NPSApp):
    def main(self):
        F = npyscreen.Form(name = "Testing Tree class",)
        #wgtree = F.add(npyscreen.MLTree)
        wgtree = F.add(npyscreen.MLTreeMultiSelect)

        treedata = npyscreen.TreeData(content='Root', selectable=True, ignore_root=False)
        c1 = treedata.new_child(content='Child 1', selectable=True, selected=True)
        c2 = treedata.new_child(content='Child 2', selectable=True)
        g1 = c1.new_child(content='Grand-child 1', selectable=True)
        g2 = c1.new_child(content='Grand-child 2', selectable=True)
        g3 = c1.new_child(content='Grand-child 3')
        gg1 = g1.new_child(content='Great Grand-child 1', selectable=True)
        gg2 = g1.new_child(content='Great Grand-child 2', selectable=True)
        gg3 = g1.new_child(content='Great Grand-child 3')
        wgtree.values = treedata

        F.edit()

        global RETURN
        #RETURN = wgtree.values
        RETURN = wgtree.get_selected_objects()




if __name__ == "__main__":
    App = TestApp()
    App.run()
    for v in RETURN:
        print v