From 668fa3497fdc17af2dff149309dac43fd9b00f4d Mon Sep 17 00:00:00 2001 From: vapao Date: Sun, 31 Oct 2021 00:13:40 +0800 Subject: [PATCH] =?UTF-8?q?U=20=E8=AE=B0=E5=BD=95websocket=E5=A4=B1?= =?UTF-8?q?=E8=B4=A5=E6=97=A5=E5=BF=97=E4=BF=A1=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spug_api/consumer/middleware.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/spug_api/consumer/middleware.py b/spug_api/consumer/middleware.py index fc7cfd2..af4a48e 100644 --- a/spug_api/consumer/middleware.py +++ b/spug_api/consumer/middleware.py @@ -20,9 +20,11 @@ class AuthMiddleware: "You cannot use AuthMiddleware on a non-WebSocket connection" ) headers = dict(scope.get('headers', [])) - if self.verify_user(scope, headers): + is_ok, message = self.verify_user(scope, headers) + if is_ok: return self.application(scope) else: + print(message) return WebsocketDenier(scope) def get_real_ip(self, headers): @@ -41,4 +43,6 @@ class AuthMiddleware: user = User.objects.filter(access_token=token).first() if user and x_real_ip == user.last_ip and user.token_expired >= time.time() and user.is_active: scope['user'] = user - return True + return True, None + return False, f'IP verify failed: {x_real_ip} <> {user.last_ip}' + return False, 'Token is invalid'