Home
Diffusion Pillar (wip-tags-view-progress) Browse

pillar/pillar/web/utils/jstree.py
e2236864e768wip-tags-view-progress

Last Change
Raw File

jstree.py
View Options

from pillarsdk import Node
from pillarsdk.exceptions import ForbiddenAccess
from pillarsdk.exceptions import ResourceNotFound
from flask_login import current_user
from pillar.web import system_util
GROUP_NODES = {'group', 'storage', 'group_texture', 'group_hdri'}
def jstree_parse_node(node, children=None):
"""Generate JStree node from node object"""
node_type = node.node_type
# Define better the node type
if node_type == 'asset':
node_type = node.properties.content_type
parsed_node = dict(
id="n_{0}".format(node._id),
text=node.name,
type=node_type,
children=False)
# Append children property only if it is a directory type
if node_type in GROUP_NODES:
parsed_node['children'] = True
return parsed_node
def jstree_get_children(node_id, project_id=None):
api = system_util.pillar_api()
children_list = []
lookup = {
'projection': {
'name': 1, 'parent': 1, 'node_type': 1, 'properties.order': 1,
'properties.status': 1, 'properties.content_type': 1, 'user': 1,
'project': 1},
'sort': [('properties.order', 1), ('_created', 1)],
'where': {
'$and': [
{'node_type': {'$regex': '^(?!attract_)'}},
{'node_type': {'$not': {'$in': ['comment', 'post', 'blog']}}},
],
}
}
if node_id:
if node_id.startswith('n_'):
node_id = node_id.split('_')[1]
lookup['where']['parent'] = node_id
elif project_id:
lookup['where']['project'] = project_id
lookup['where']['parent'] = {'$exists': False}
try:
children = Node.all(lookup, api=api)
for child in children['_items']:
is_pub = child.properties.status == 'published'
if is_pub or (current_user.is_authenticated and child.user == current_user.objectid):
children_list.append(jstree_parse_node(child))
except ForbiddenAccess:
pass
return children_list
def jstree_build_children(node):
return dict(
id="n_{0}".format(node._id),
text=node.name,
type=node.node_type,
children=jstree_get_children(node._id)
)
def jstree_build_from_node(node):
"""Give a node, traverse the tree bottom to top and expand the relevant
branches.
:param node: the base node, where tree building starts
"""
api = system_util.pillar_api()
# Parse the node and mark it as selected
child_node = jstree_parse_node(node)
child_node['state'] = dict(selected=True)
# Splice the specified child node between the other project children.
def select_node(x):
if x['id'] == child_node['id']:
return child_node
return x
# Get the parent node
parent = None
if node.parent:
try:
parent = Node.find(node.parent, {
'projection': {
'name': 1,
'node_type': 1,
'parent': 1,
'properties.content_type': 1,
}}, api=api)
# Define the child node of the tree (usually an asset)
except ResourceNotFound:
# If not found, we might be on the top level, in which case we skip the
# while loop and use child_node
pass
except ForbiddenAccess:
pass
while parent:
# Get the parent's parent
parent_parent = jstree_parse_node(parent)
# Get the parent's children (this will also include child_node)
parent_children = [select_node(x) for x in jstree_get_children(parent_parent['id'])]
parent_parent.pop('children', None)
# Overwrite children_node with the current parent
child_node = parent_parent
# Set the node to open so that jstree actually displays the nodes
child_node['state'] = dict(opened=True)
# Push in the computed children into the parent
child_node['children'] = parent_children
# If we have a parent
if parent.parent:
try:
parent = Node.find(parent.parent, {
'projection': {
'name': 1, 'parent': 1, 'project': 1, 'node_type': 1},
}, api=api)
except ResourceNotFound:
parent = None
else:
parent = None
# Get top level nodes for the project
project_children = jstree_get_children(None, node.project)
nodes_list = [select_node(x) for x in project_children]
return nodes_list