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

fix: bug

pull/5435/head
ibuler 2021-01-14 10:10:10 +08:00 committed by 老广
parent 83d12d02fb
commit f04e2fa090
2 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apps/authentication/api/auth.py
View File

@ -54,12 +54,3 @@ class UserConnectionTokenApi(RootOrgViewMixin, APIView):
return Response(value) return Response(value)
else: else:
return Response({'user': value['user']}) return Response({'user': value['user']})
def get_permissions(self):
if self.request.query_params.get('user-only', None):
self.permission_classes = (AllowAny,)
return super().get_permissions()

4
apps/ops/ws.py
View File

@ -21,7 +21,11 @@ class TaskLogWebsocket(JsonWebsocketConsumer):
} }
def connect(self): def connect(self):
user = self.scope["user"]
if user.is_authenticated and user.is_org_admin:
self.accept() self.accept()
else:
self.close()
def get_log_path(self, task_id): def get_log_path(self, task_id):
func = self.log_types.get(self.log_type) func = self.log_types.get(self.log_type)