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

[Update] sql优化，查询用户资产权限时，使用union替代or (#3681)

pull/3682/head
老广 2020-02-05 12:10:24 +08:00 committed by GitHub
parent ec1f6677ec
commit aa2255a87e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
apps/perms/utils/asset_permission.py
View File

@ -25,12 +25,14 @@ __all__ = [
def get_user_permissions(user, include_group=True):
permissions = AssetPermission.get_queryset_with_prefetch().filter(users=user)
if include_group:
groups = user.groups.all()
arg = Q(users=user) | Q(user_groups__in=groups)
else:
arg = Q(users=user)
return AssetPermission.get_queryset_with_prefetch().filter(arg)
permissions_groups = AssetPermission.get_queryset_with_prefetch().filter(
user_groups__in=groups
)
permissions = permissions.union(permissions_groups)
return permissions
def get_user_group_permissions(user_group):
@ -40,12 +42,14 @@ def get_user_group_permissions(user_group):
def get_asset_permissions(asset, include_node=True):
permissions = AssetPermission.get_queryset_with_prefetch().filter(asset=asset)
if include_node:
nodes = asset.get_all_nodes(flat=True)
arg = Q(assets=asset) | Q(nodes__in=nodes)
else:
arg = Q(assets=asset)
return AssetPermission.objects.valid().filter(arg)
permissions_nodes = AssetPermission.get_queryset_with_prefetch().filter(
nodes__in=nodes
)
permissions = permissions.union(permissions_nodes)
return permissions
def get_node_permissions(node):