def team_screen(self):
self.window.close_window(self.selected_window)
self.class_window = self.window.create_window(15, 30, 0, 5,
'select your team')
self.log_w = self.window.create_window(50, 8, 0, 42, 'Log')
self.log = ui.Log(self.log_w, 48, 6)
self.log_w.add_elem(self.log)
menu = ui.Control_Menu(self.class_window, 13, 28, 'classes')
self.class_window.add_elem(menu)
self.create_squad()
warrior = Ui.Text_Button(menu, 10, 1, 'warrior', Color.WHITE,
Color.BLACK, self.select_class,
[Classes.Warrior()])
menu.add_elem(warrior)
archer = Ui.Text_Button(menu, 10, 1, 'archer', Color.WHITE,
Color.BLACK, self.select_class,
[Classes.Archer()])
menu.add_elem(archer)
assasin = Ui.Text_Button(menu, 10, 1, 'assasin', Color.WHITE,
Color.BLACK, self.select_class,
[Classes.Assasin()])
menu.add_elem(assasin)
sage = Ui.Text_Button(menu, 10, 1, 'sage', Color.WHITE, Color.BLACK,
self.select_class, [Classes.Sage()])
menu.add_elem(sage)
mage = Ui.Text_Button(menu, 10, 1, 'mage', Color.WHITE, Color.BLACK,
self.select_class, [Classes.Mage()])
menu.add_elem(mage)
start = Ui.Text_Button(menu, 10, 1, 'Begin', Color.LBLUE, Color.GREEN,
self.validate_team, None)
menu.add_elem(start)
self.set_state(2) #4: selection_screen
self.menu_w = self.window.create_window(30, 50, 49, 0, "Menu")
self.charamenu = ui.Soft_Menu(self.menu_w, 28, 48, 'Menus')
self.menu_w.add_elem(self.charamenu)
self.squad_ui = ui.Squad_Ui(self.charamenu, self.squad, self)
self.charamenu.add_elem(self.squad_ui)
self.select_window(self.class_window)
self.window.build('erase')
