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

U 记录websocket失败日志信息

pull/410/head
vapao 2021-10-31 00:13:40 +08:00
parent f384374db7
commit 668fa3497f
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spug_api/consumer/middleware.py
View File

@ -20,9 +20,11 @@ class AuthMiddleware:
"You cannot use AuthMiddleware on a non-WebSocket connection" "You cannot use AuthMiddleware on a non-WebSocket connection"
) )
headers = dict(scope.get('headers', [])) 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) return self.application(scope)
else: else:
print(message)
return WebsocketDenier(scope) return WebsocketDenier(scope)
def get_real_ip(self, headers): def get_real_ip(self, headers):
@ -41,4 +43,6 @@ class AuthMiddleware:
user = User.objects.filter(access_token=token).first() 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: if user and x_real_ip == user.last_ip and user.token_expired >= time.time() and user.is_active:
scope['user'] = user 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'