Django通道自定义身份验证中间件__call __()缺少2个必需的位置参数:“接收"和“发送"

问题描述

我正在为Django频道编写自定义身份验证中间件

I am writing a custom authentication middleware for django channels

class TokenAuthMiddleware:
    def __init__(self, inner):
        # Store the ASGI application we were passed
        self.inner = inner

    def __call__(self, scope):

        return TokenAuthMiddlewareInstance(scope, self)


class TokenAuthMiddlewareInstance:

    def __init__(self, scope, middleware):
        self.middleware = middleware
        self.scope = dict(scope)
        self.inner = self.middleware.inner

    async def __call__(self, receive, send):
        ## my logic to get validate user and store the user in user data
        ...
        ...
        ...
        self.scope['user'] = user_data
        inner = self.inner(self.scope)
        return await inner(receive, send)

但是在尝试从前端连接到Web套接字时，出现以下错误

but on trying to connect to web socket from front end I get the following error

TypeError: __call__() missing 2 required positional arguments: 'receive' and 'send'

推荐答案

供您参考: https://channels.readthedocs.io/en/stable/releases/3.0.0.html

从routing.py中更改

change from in routing.py

websocket_urlpatterns = [
    re_path(r'ws/chat/(?P<room_name>\w+)/$', consumers.ChatConsumer),
]

到消费者现在拥有在设置路由时需要调用的as_asgi()类方法:

toConsumers now have an as_asgi() class method you need to call when setting up your routing:

websocket_urlpatterns = [
    re_path(r'ws/chat/(?P<room_name>\w+)/$', consumers.ChatConsumer.as_asgi()),
]

然后如果您需要自定义身份验证，请 https://channels.readthedocs.io/en/stable/topics/authentication.html#custom-authentication

thenif you need custom authentication https://channels.readthedocs.io/en/stable/topics/authentication.html#custom-authentication

from channels.auth import AuthMiddlewareStack
from channels.db import database_sync_to_async
from django.contrib.auth import get_user_model

User = get_user_model()

@database_sync_to_async
def get_user(user_id):
    try:
        return User.objects.get(id=user_id)
    except User.DoesNotExist:
        return AnonymousUser()

class QueryAuthMiddleware:
    """
    Custom middleware (insecure) that takes user IDs from the query string.
    """

    def __init__(self, app):
        # Store the ASGI application we were passed
        self.app = app

    async def __call__(self, scope, receive, send):
        # Look up user from query string (you should also do things like
        # checking if it is a valid user ID, or if scope["user"] is already
        # populated).
        scope['user'] = await get_user(int(scope["query_string"]))

        return await self.app(scope, receive, send)
TokenAuthMiddlewareStack = lambda inner: QueryAuthMiddleware(AuthMiddlewareStack(inner))

Django==3.0.8
djangorestframework==3.11.0
websocket-client==0.57.0
redis==3.5.3
asgiref==3.2.10
channels-redis==2.4.2
channels==3.0.1

这篇关于Django通道自定义身份验证中间件__call __()缺少2个必需的位置参数:“接收"和“发送"的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！