c26f8827854295228a774c928b307224ad128e2d,lib/streamlit/proxy/ClientWebSocket.py,ClientWebSocket,on_message,#ClientWebSocket#Any#,88

Before Change

    def on_message(self, msg):
        LOGGER.debug(f"Received message of length {len(msg)}.")
        backend_msg = protobuf.BackMsg()
        try:
            backend_msg.ParseFromString(payload)
            command  = backend_msg.command
            if command == protobuf.BackMsg.Command.Value("HELP"):
                os.system("python -m streamlit help &")
            elif command == protobuf.BackMsg.Command.Value("CLOUD_UPLOAD"):
                LOGGER.debug("CLOUD_UPLOAD!!!")
                yield self._save_cloud(connection, ws)
            else:
                LOGGER.warning("no handler for %s",
                    protobuf.BackMsg.Command.Name(backend_msg.command))
        except Exception as e:
            LOGGER.error("Cannot parse binary message: %s", e)

    @Proxy.stop_proxy_on_exception()
    def on_close(self):
        LOGGER.debug("Received close message.")
        self._is_open = False

After Change

    def on_message(self, msg):
        Run callback for websocket messages.
        LOGGER.debug(repr(msg))
        yield self._handle_backend_msg(msg, self._connection, self)

    @gen.coroutine
    def _handle_backend_msg(self, payload, connection, ws):
        backend_msg = protobuf.BackMsg()

Instances