github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-06-30 16:52:05 +00:00
Code Issues Projects Releases Wiki Activity

[Bugfix] 修复coco无法查看资产的bug

Browse Source
This commit is contained in:
ibuler 2018-09-13 12:16:49 +08:00
parent 310bc6ad0b
commit 3ce9d01b6d
1 changed files with 21 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
apps/perms/api.py
View File

@ -62,14 +62,15 @@ class UserGrantedAssetsApi(ListAPIView):
""" """
permission_classes = (IsOrgAdminOrAppUser,) permission_classes = (IsOrgAdminOrAppUser,)
serializer_class = AssetGrantedSerializer serializer_class = AssetGrantedSerializer
def dispatch(self, request, *args, **kwargs): def change_org_if_need(self):
if request.user.is_superuser or request.user.is_app or \ if self.request.user.is_superuser or \
self.request.user.is_app or \
self.kwargs.get('pk') is None: self.kwargs.get('pk') is None:
set_to_root_org() set_to_root_org()
return super().dispatch(request, *args, **kwargs)
def get_queryset(self): def get_queryset(self):
self.change_org_if_need()
user_id = self.kwargs.get('pk', '') user_id = self.kwargs.get('pk', '')
queryset = [] queryset = []
@ -97,14 +98,15 @@ class UserGrantedNodesApi(ListAPIView):
""" """
permission_classes = (IsOrgAdmin,) permission_classes = (IsOrgAdmin,)
serializer_class = NodeSerializer serializer_class = NodeSerializer
def dispatch(self, request, *args, **kwargs): def change_org_if_need(self):
if request.user.is_superuser or request.user.is_app or \ if self.request.user.is_superuser or \
self.request.user.is_app or \
self.kwargs.get('pk') is None: self.kwargs.get('pk') is None:
set_to_root_org() set_to_root_org()
return super().dispatch(request, *args, **kwargs)
def get_queryset(self): def get_queryset(self):
self.change_org_if_need()
user_id = self.kwargs.get('pk', '') user_id = self.kwargs.get('pk', '')
if user_id: if user_id:
user = get_object_or_404(User, id=user_id) user = get_object_or_404(User, id=user_id)
@ -126,14 +128,15 @@ class UserGrantedNodesWithAssetsApi(ListAPIView):
""" """
permission_classes = (IsOrgAdminOrAppUser,) permission_classes = (IsOrgAdminOrAppUser,)
serializer_class = NodeGrantedSerializer serializer_class = NodeGrantedSerializer
def dispatch(self, request, *args, **kwargs): def change_org_if_need(self):
if request.user.is_superuser or request.user.is_app or \ if self.request.user.is_superuser or \
self.request.user.is_app or \
self.kwargs.get('pk') is None: self.kwargs.get('pk') is None:
set_to_root_org() set_to_root_org()
return super().dispatch(request, *args, **kwargs)
def get_queryset(self): def get_queryset(self):
self.change_org_if_need()
user_id = self.kwargs.get('pk', '') user_id = self.kwargs.get('pk', '')
queryset = [] queryset = []
if not user_id: if not user_id:
@ -164,14 +167,15 @@ class UserGrantedNodeAssetsApi(ListAPIView):
""" """
permission_classes = (IsOrgAdminOrAppUser,) permission_classes = (IsOrgAdminOrAppUser,)
serializer_class = AssetGrantedSerializer serializer_class = AssetGrantedSerializer
def dispatch(self, request, *args, **kwargs): def change_org_if_need(self):
if request.user.is_superuser or request.user.is_app or \ if self.request.user.is_superuser or \
self.request.user.is_app or \
self.kwargs.get('pk') is None: self.kwargs.get('pk') is None:
set_to_root_org() set_to_root_org()
return super().dispatch(request, *args, **kwargs)
def get_queryset(self): def get_queryset(self):
self.change_org_if_need()
user_id = self.kwargs.get('pk', '') user_id = self.kwargs.get('pk', '')
node_id = self.kwargs.get('node_id') node_id = self.kwargs.get('node_id')