| 104 | raise WebSocketDisconnected('Cannot call "send" once a close message has been sent.') |
| 105 | |
| 106 | async def accept( |
| 107 | self, |
| 108 | subprotocol: str | None = None, |
| 109 | headers: Iterable[tuple[bytes, bytes]] | None = None, |
| 110 | ) -> None: |
| 111 | headers = headers or [] |
| 112 | |
| 113 | if self.client_state == WebSocketState.CONNECTING: # pragma: no branch |
| 114 | # If we haven't yet seen the 'connect' message, then wait for it first. |
| 115 | await self.receive() |
| 116 | await self.send({"type": "websocket.accept", "subprotocol": subprotocol, "headers": headers}) |
| 117 | |
| 118 | def _raise_on_disconnect(self, message: Message) -> None: |
| 119 | if message["type"] == "websocket.disconnect": |