python - 在单线程 Python 脚本中有一个控制台

Question

我想在一个打开了多个 TCP 连接的单线程脚本中有一个交互式控制台。这意味着我不能只让标准输入阻塞线程。

有没有简单的方法来做到这一点？还是我应该将控制台放在它自己的线程中并完成它？

Answer 1

您可以子类化 InteractiveConsole(来自内置的“代码”模块)和 使用重定向 stdout/stderr 的包装器覆盖 push() 方法 在转发到基础之前到 StringIO 实例 InteractiveConsole 的 push() 方法。你的包装器可以返回一个二元组 (more, result) 其中 'more' 表示 InteractiveConsole 是否期望 更多输入，“结果”是 InteractiveConsole.push() 写入的内容 你的 StringIO 实例。

听起来比实际更难。这是基本前提:

import sys
from cStringIO import StringIO
from code import InteractiveConsole
from contextlib import contextmanager

__all__ = ['Interpreter']


@contextmanager
def std_redirector(stdin=sys.stdin, stdout=sys.stdin, stderr=sys.stderr):
    """Temporarily redirect stdin/stdout/stderr"""

    tmp_fds = stdin, stdout, stderr
    orig_fds = sys.stdin, sys.stdout, sys.stderr
    sys.stdin, sys.stdout, sys.stderr = tmp_fds
    yield
    sys.stdin, sys.stdout, sys.stderr = orig_fds


class Interpreter(InteractiveConsole):
    """Remote-friendly InteractiveConsole subclass

    This class behaves just like InteractiveConsole, except that it
    returns all output as a string rather than emitting to stdout/stderr

    """
    banner = ("Python %s\n%s\n" % (sys.version, sys.platform) +
              'Type "help", "copyright", "credits" or "license" '
              'for more information.\n')

    ps1 = getattr(sys, "ps1", ">>> ")
    ps2 = getattr(sys, "ps2", "... ")


    def __init__(self, locals=None):
        InteractiveConsole.__init__(self, locals=locals)
        self.output = StringIO()
        self.output = StringIO()

    def push(self, command):
        """Return the result of executing `command`

        This function temporarily redirects stdout/stderr and then simply
        forwards to the base class's push() method.  It returns a 2-tuple
        (more, result) where `more` is a boolean indicating whether the
        interpreter expects more input [similar to the base class push()], and
        `result` is the captured output (if any) from running `command`.

        """
        self.output.reset()
        self.output.truncate()
        with std_redirector(stdout=self.output, stderr=self.output):
            try:
                more = InteractiveConsole.push(self, command)
                result = self.output.getvalue()
            except (SyntaxError, OverflowError):
                pass
            return more, result

查看这个完整的示例，它接受来自 UDP 套接字的输入:

• http://files.evadeflow.com/pyrrepl.zip

启动两个控制台，其中一个运行server.py，另一个运行client.py。 你在 client.py 中看到的应该与 python 没有区别 常规交互式解释器，即使所有命令都在 往返于 server.py 进行评估。

当然，像这样使用套接字是非常不安全的，但它 说明如何异步评估外部输入。你 应该能够适应你的情况，只要你相信 输入源。当有人输入时，事情会变得“有趣”:

while True: continue

但这完全是另一个问题... :-)