AsyncWebsocketConsumer: KeyError "code".

[VladosikPaltosik]

channels/generic/websocket.py in websocket_disconnect at line 238

                await self.channel_layer.group_discard(group, self.channel_name)
        except AttributeError:
            raise InvalidChannelLayerError(
                "BACKEND is unconfigured or doesn't support groups"
            )
        await self.disconnect(message["code"])
        raise StopConsumer()

Got KeyError "code". How to solve a problem???

[carltongibson]

This would be addressed by #1905, which is lacking a regression test currently.

[VladosikPaltosik]

This would be addressed by #1905, which is lacking a regression test currently.

Perhaps I can handle this error? Because my Sentry are full of such errors.

class ConnectionConsumer(AsyncWebsocketConsumer):

    async def connect(self):
        """ Called on connection """
        try:
            await self.accept()
        except (ConnectionClosedError, ConnectionClosedOK, KeyError):
            await self.close(code=1005)

    async def receive(self, text_data=None, bytes_data=None):
        """ Receive a message and send it back """
        text_data_json = json.loads(text_data)
        message = text_data_json.get('message', None)
        try:
            await self.send(text_data=json.dumps({'message': message}))
        except (ConnectionClosedError, ConnectionClosedOK, KeyError):
            await self.close(code=1005)