def _group_tree_branch(root_group, highlight_group_name=None, type="group"):
"""Returns a branch of the group tree hierarchy, rooted in the given group.
:param root_group_id: group object at the top of the part of the tree
:param highlight_group_name: group name that is to be flagged 'highlighted'
:returns: the top GroupTreeNode of the tree
"""
nodes = {} # group_id: GroupTreeNode()
root_node = nodes[root_group.id] = GroupTreeNode({
"id": root_group.id,
"name": root_group.name,
"title": root_group.title,
})
if root_group.name == highlight_group_name:
nodes[root_group.id].highlight()
highlight_group_name = None
for (
group_id,
group_name,
group_title,
parent_id,
) in root_group.get_children_group_hierarchy(type=type):
node = GroupTreeNode({
"id": group_id,
"name": group_name,
"title": group_title
})
nodes[parent_id].add_child_node(node)
if highlight_group_name and group_name == highlight_group_name:
node.highlight()
nodes[group_id] = node
return root_node
def _group_tree_branch(root_group, highlight_group_name=None, type='group', group_counts=None):
'''Returns a branch of the group tree hierarchy, rooted in the given group.
:param root_group_id: group object at the top of the part of the tree
:param highlight_group_name: group name that is to be flagged 'highlighted'
:returns: the top GroupTreeNode of the tree
'''
nodes = {} # group_id: GroupTreeNode()
root_node = nodes[root_group.id] = GroupTreeNode(
{'id': root_group.id,
'name': root_group.name,
'title': root_group.title})
if root_group.name == highlight_group_name:
nodes[root_group.id].highlight()
highlight_group_name = None
for group_id, group_name, group_title, parent_id in \
root_group.get_children_group_hierarchy(type=type):
node = GroupTreeNode({'id': group_id,
'name': group_name,
'title': group_title})
if not group_counts or group_id in group_counts:
nodes[parent_id].add_child_node(node)
if highlight_group_name and group_name == highlight_group_name:
node.highlight()
nodes[group_id] = node
return root_node
def _nest_group_tree_list(group_tree_list, group_tree_leaf):
'''Returns a tree branch composed by nesting the groups in the list.
:param group_tree_list: list of groups to build a tree, first is root
:returns: the top GroupTreeNode of the tree
'''
root_node = None
last_node = None
log.debug(group_tree_list)
for group in group_tree_list:
log.debug(group)
node = GroupTreeNode({
'id': group.id,
'name': group.name,
'description': group.description,
#'image_display_url': group.image_display_url,
'title': group.title
})
if not root_node:
root_node = last_node = node
else:
last_node.add_child_node(node)
last_node = node
last_node.add_child_node(group_tree_leaf)
return root_node
def _group_tree_branch(root_group, pkg_count, highlight_group_name=None, type='group'):
'''Returns a branch of the group tree hierarchy, rooted in the given group.
:param root_group_id: group object at the top of the part of the tree
:param highlight_group_name: group name that is to be flagged 'highlighted'
:returns: the top GroupTreeNode of the tree
'''
import pprint
nodes = {} # group_id: GroupTreeNode()
#Calculate package count for root orgs
root_count = pkg_count.get(root_group.name, 0)
for group_id, group_name, group_title, parent_id in root_group.get_children_group_hierarchy(type=type):
root_count += pkg_count.get(group_name, 0)
root_node = nodes[root_group.id] = GroupTreeNode(
{'id': root_group.id,
'pkg_num': str(root_count),
'name': root_group.name,
'title': root_group.title,
'is_top_org' : True})
if root_group.name == highlight_group_name:
nodes[root_group.id].highlight()
highlight_group_name = None
for group_id, group_name, group_title, parent_id in \
root_group.get_children_group_hierarchy(type=type):
pkg_num = pkg_count.get(group_name, 0)
node = GroupTreeNode({'id': group_id,
'pkg_num': str(pkg_num),
'name': group_name,
'title': group_title,
'is_top_org' : False})
nodes[parent_id].add_child_node(node)
if highlight_group_name and group_name == highlight_group_name:
node.highlight()
nodes[group_id] = node
return root_node
def _nest_group_tree_list(group_tree_list, group_tree_leaf):
"""Returns a tree branch composed by nesting the groups in the list.
:param group_tree_list: list of groups to build a tree, first is root
:returns: the top GroupTreeNode of the tree
"""
root_node = None
last_node = None
for group in group_tree_list:
node = GroupTreeNode({
"id": group.id,
"name": group.name,
"title": group.title
})
if not root_node:
root_node = last_node = node
else:
last_node.add_child_node(node)
last_node = node
last_node.add_child_node(group_tree_leaf)
return root_node
def gtn_from_group(grp):
return GroupTreeNode(group_dictize(grp))