From d5386ccd841d82dec129216965bbef8edf88ea81 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Fri, 1 Sep 2023 09:11:43 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96ws=20=E8=AE=A4?= =?UTF-8?q?=E8=AF=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/utils/middlewares.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/apps/utils/middlewares.py b/apps/utils/middlewares.py index 74be9af3..fdda5e31 100644 --- a/apps/utils/middlewares.py +++ b/apps/utils/middlewares.py @@ -15,6 +15,9 @@ class TokenAuthMiddleware: # checking if it is a valid user ID, or if scope["user"] is already # populated). from urllib.parse import parse_qs - token = parse_qs(str(scope["query_string"], 'UTF-8'))['token'][0] - scope['user'] = await _get_user(token) - return await self.app(scope, receive, send) \ No newline at end of file + token = parse_qs(str(scope["query_string"], 'UTF-8')).get('token', [None])[0] + if token: + user = await _get_user(token) + if user: + scope['user'] = user + return await self.app(scope, receive, send) \ No newline at end of file