diff --git a/proxy/plugin/modify_websocket_response.py b/proxy/plugin/modify_websocket_response.py index 7d5e9dd3..1ed4d88d 100644 --- a/proxy/plugin/modify_websocket_response.py +++ b/proxy/plugin/modify_websocket_response.py @@ -22,7 +22,9 @@ class ModifyWebsocketResponsePlugin(HttpProxyBasePlugin): while len(remaining) > 0: response = WebsocketFrame() remaining = response.parse(remaining) - self.client.queue(memoryview( - WebsocketFrame.text(b"modified websocket response") - )) + self.client.queue( + memoryview( + WebsocketFrame.text(b'modified websocket response'), + ), + ) return memoryview(b'') diff --git a/proxy/plugin/web_server_route.py b/proxy/plugin/web_server_route.py index 42e341ec..398b6686 100644 --- a/proxy/plugin/web_server_route.py +++ b/proxy/plugin/web_server_route.py @@ -39,6 +39,8 @@ class WebServerPlugin(HttpWebServerBasePlugin): self.client.queue(HTTPS_RESPONSE) def on_websocket_message(self, frame: WebsocketFrame) -> None: - self.client.queue(memoryview( - WebsocketFrame.text(b"Websocket route response"), - )) + self.client.queue( + memoryview( + WebsocketFrame.text(b'Websocket route response'), + ), + )