Improved graphql middleware

Author: MySecondLanguage

nxtbn/users/auth_middleware.py

@@ -26,22 +26,17 @@ def resolve(self, next, root, info, **args):
         return next(root, info, **args)
 
     def get_user_from_jwt(self, request):
-        """Retrieve the user from the JWT token in the request."""
         token = self.get_token_from_request(request)
         if token:
             return self.jwt_manager.verify_jwt_token(token) or AnonymousUser()
         return AnonymousUser()
 
     def get_user_from_session(self, request):
-        """Retrieve the user from the session, if available."""
         return request.user if request.user.is_authenticated else AnonymousUser()
 
     def get_token_from_request(self, request):
-        """Extract the token from the Authorization header or cookies."""
-        # Check Authorization header
         auth_header = request.headers.get("Authorization")
         if auth_header and auth_header.startswith("Bearer "):
             return auth_header.split(" ")[1]
 
-        # Fallback to cookies
         return request.COOKIES.get("access_token")