提交 043d24a8 编写于 作者: baltery's avatar baltery 提交者: baltery

fix: bug

上级 1a011f34
...@@ -54,12 +54,3 @@ class UserConnectionTokenApi(RootOrgViewMixin, APIView): ...@@ -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()
...@@ -15,7 +15,11 @@ class CeleryLogWebsocket(JsonWebsocketConsumer): ...@@ -15,7 +15,11 @@ class CeleryLogWebsocket(JsonWebsocketConsumer):
disconnected = False disconnected = False
def connect(self): def connect(self):
self.accept() user = self.scope["user"]
if user.is_authenticated and user.is_org_admin:
self.accept()
else:
self.close()
def receive(self, text_data=None, bytes_data=None, **kwargs): def receive(self, text_data=None, bytes_data=None, **kwargs):
data = json.loads(text_data) data = json.loads(text_data)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册