perf: 代码优化

pull/6555/head
feng626 2021-07-28 14:33:02 +08:00
parent d0680c3753
commit da588ce0ae
1 changed files with 9 additions and 7 deletions

View File

@ -549,13 +549,15 @@ class UserGrantedNodesQueryUtils(UserGrantedUtilsBase):
return self.get_top_level_nodes()
nodes = PermNode.objects.none()
if key != PermNode.FAVORITE_NODE_KEY:
node = PermNode.objects.get(key=key)
granted_status = node.get_granted_status(self.user)
if granted_status == NodeFrom.granted:
nodes = PermNode.objects.filter(parent_key=key)
elif granted_status in (NodeFrom.asset, NodeFrom.child):
nodes = self.get_indirect_granted_node_children(key)
if key == PermNode.FAVORITE_NODE_KEY:
return nodes
node = PermNode.objects.get(key=key)
granted_status = node.get_granted_status(self.user)
if granted_status == NodeFrom.granted:
nodes = PermNode.objects.filter(parent_key=key)
elif granted_status in (NodeFrom.asset, NodeFrom.child):
nodes = self.get_indirect_granted_node_children(key)
nodes = self.sort(nodes)
return nodes