在单线程 Python 脚本中拥有控制台

Question

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

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

Answer 1

您可以将 InteractiveConsole 子类化（来自内置“代码”模块）并
使用重定向 stdout/stderr 的包装器覆盖 push() 方法
在转发到基础之前到 StringIO 实例
InteractiveConsole 的push() 方法。你的包装器可以返回一个 2 元组
(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

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

Answer 2

单线程或多线程都可以，但如果您选择不使用线程，则需要使用轮询（例如，在 C 中可以使用 poll(2) 来完成）并检查控制台和/或者 TCP 连接已准备好输入。