github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

[Update] 修改小问题

pull/3342/head
jym503558564 2019-10-17 11:59:18 +08:00
parent ee97144a10
commit 70664d435f
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/perms/api/asset_permission.py
View File

@ -68,18 +68,18 @@ class AssetPermissionViewSet(viewsets.ModelViewSet):
node_id = self.request.query_params.get('node_id') node_id = self.request.query_params.get('node_id')
node_name = self.request.query_params.get('node') node_name = self.request.query_params.get('node')
if node_id: if node_id:
nodes = Node.objects.filter(pk=node_id) _nodes = Node.objects.filter(pk=node_id)
elif node_name: elif node_name:
nodes = Node.objects.filter(value=node_name) _nodes = Node.objects.filter(value=node_name)
else: else:
return queryset return queryset
if not nodes: if not _nodes:
return queryset.none() return queryset.none()
related_nodes = [] nodes = set()
for node in nodes: for node in _nodes:
[related_nodes.append(item) for item in node.get_ancestors(with_self=True)] nodes |= set(node.get_ancestors(with_self=True))
queryset = queryset.filter(nodes__in=related_nodes) queryset = queryset.filter(nodes__in=nodes)
return queryset return queryset
def filter_asset(self, queryset): def filter_asset(self, queryset):