From e98032200e1cbc49c26e423761eb95252111a9ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=BC=BA?= <1206709430@qq.com>
Date: Sat, 8 Apr 2023 20:29:23 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AF=BC=E5=8C=85=E9=97=AE=E9=A2=98=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/application/asgi.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/backend/application/asgi.py b/backend/application/asgi.py
index 586b729..a366be7 100644
--- a/backend/application/asgi.py
+++ b/backend/application/asgi.py
@@ -13,27 +13,23 @@ import jwt
 from channels.db import database_sync_to_async
 from channels.middleware import BaseMiddleware
 from django.contrib.auth import get_user_model
-from django.contrib.auth.models import AnonymousUser
 from django.core.asgi import get_asgi_application
 from channels.auth import AuthMiddlewareStack
 from channels.routing import ProtocolTypeRouter, URLRouter
 from django.db import close_old_connections
-from rest_framework_simplejwt.authentication import AUTH_HEADER_TYPE_BYTES
-from rest_framework_simplejwt.exceptions import InvalidToken, TokenError
-from rest_framework_simplejwt.tokens import UntypedToken
-from dvadmin.system.models import Users
-from application import settings
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'application.settings')
 os.environ["DJANGO_ALLOW_ASYNC_UNSAFE"] = "true"
 
 @database_sync_to_async
 def get_user(validated_token):
+    from dvadmin.system.models import Users
     try:
         user = get_user_model().objects.get(id=validated_token["user_id"])
         # return get_user_model().objects.get(id=toke_id)
         return user
 
     except Users.DoesNotExist:
+        from django.contrib.auth.models import AnonymousUser
         return AnonymousUser()
 
 
@@ -43,6 +39,9 @@ class JwtAuthMiddleware(BaseMiddleware):
 
     async def __call__(self, scope, receive, send):
         # Close old database connections to prevent usage of timed out connections
+        from rest_framework_simplejwt.authentication import AUTH_HEADER_TYPE_BYTES
+        from rest_framework_simplejwt.exceptions import InvalidToken, TokenError
+        from rest_framework_simplejwt.tokens import UntypedToken
         close_old_connections()
         parts = dict(scope['headers']).get(b'authorization', b'').split()
         print("parts",scope)
@@ -69,6 +68,7 @@ class JwtAuthMiddleware(BaseMiddleware):
             return None
         else:
             #  Then token is valid, decode it
+            from application import settings
             decoded_data = jwt.decode(token, settings.SECRET_KEY, algorithms=["HS256"])
             print(decoded_data)
             # Will return a dictionary like -
@@ -97,4 +97,4 @@ application = ProtocolTypeRouter({
             websocket_urlpatterns #指明路由文件是devops/routing.py
         )
     ),
-})
\ No newline at end of file
+})