def __init__(self, container):
super(WorkflowCycleTreeWidget, self).__init__()
# Composition is used here it's a tree widget with sub trees.
# Subtrees are not implemented in `TreeWidget` class.
self._tree_widget = tree_widget.TreeWidget(
container=container, table_row_cls=WorkflowCycleRow)
def task_group_titles(workflow):
"""Returns task group titles of `workflow`."""
selenium_utils.open_url(url.obj_tab_url(workflow, "Setup"))
return [item.title() for item in tree_widget.TreeWidget().tree_items()]
def _tree_widget(self):
"""Returns tree widget with active cycles."""
return tree_widget.TreeWidget(
container=self._browser.section(id=self._url_fragment()),
table_row_cls=controls_tree_rows.ControlsRow)