Django Channels 1.1.8如何自定义代码拒绝WebSocket连接？

嘿，我刚好折腾过Channels 1.x的WebSocket处理，太懂你遇到的这个痛点了！确实，Channels 1.1.8里默认的message.reply_channel({'accept': False})只会返回403，没法自定义状态码，而close方法又得先接受连接，完全卡壳。不过咱们可以绕开封装，直接操作底层的ASGI接口来解决这个问题。

核心思路：直接返回自定义HTTP响应 #

WebSocket的握手本质是一个HTTP升级请求，所以在连接阶段（websocket_connect处理函数里），我们不需要用Channels封装的accept逻辑，而是直接构造一个带自定义状态码的HTTP响应发送回去，这样客户端就能拿到你指定的状态码，进而正确处理拒绝逻辑。

具体代码实现 #

from channels.handler import AsgiHandler

def websocket_connect(message):
    # 这里替换成你的连接验证逻辑，比如检查token、权限等
    is_valid = False  # 示例：验证不通过
    
    if not is_valid:
        # 构造自定义HTTP响应，状态码换成你需要的（比如401、400等）
        custom_response = AsgiHandler.http_response(
            status=401,
            headers=[
                (b'Content-Type', b'text/plain'),
            ],
            content=b'Unauthorized: Invalid token provided',
        )
        # 发送响应，完成拒绝
        message.reply_channel.send(custom_response)
        return
    
    # 验证通过时正常接受连接
    message.reply_channel.send({'accept': True})

关键细节说明 #

• 状态码选择：这里要用合法的HTTP状态码（比如401未授权、400错误请求、403禁止），不要用WebSocket的关闭码（那些是连接建立后断开时用的）。客户端会在握手阶段收到这个HTTP状态码，触发对应的错误处理逻辑。
• 时机要求：这个方法只能在websocket_connect阶段使用，也就是握手还没完成的时候。一旦你调用了{'accept': True}，就只能用close方法带WebSocket关闭码来断开连接了。
• 客户端适配：客户端需要监听WebSocket握手的HTTP状态码（比如在浏览器里，onerror事件或者通过前置请求检测握手响应），而不是等待WebSocket的关闭事件，因为这是在握手阶段就直接拒绝了，并没有建立起WebSocket连接。

内容的提问来源于stack exchange，提问作者MCBama