如何在IPython中向ThreadedKernelClient添加回调？

Question

目前，我通过以下方式使用ThreadedKernelClient ：

kernel_manager = KernelManager(client_class='IPython.kernel.threaded.ThreadedKernelClient')
kernel_manager.start_kernel()
kernel_client = kernel_manager.client()

但是，我不确定如何注册传入消息的回调（在运行execute ）。 谁能指出我正确的方向？

Answer 1

我所做的是ThreadedZMQSocketChannel和ThreadedKernelClient子类，并重写了call_handlers ，例如：

class TestChannel(ThreadedZMQSocketChannel):
    def call_handlers(self, msg):
        # do something


class TestClient(ThreadedKernelClient):
    iopub_channel_class = Type(TestChannel)
    shell_channel_class = Type(TestChannel)
    stdin_channel_class = Type(TestChannel)

然后，您的经理看起来像：

kernel_manager = KernelManager(kernel_name='python', client_class='__main__.TestClient')

一旦您start_channels ，您的call_handlers方法就会被诸如execute方法的结果execute 。