在Python cmd模块中处理CTRL-C

Question

我使用cmd模块编写了一个Python 3.5应用程序.我想要实现的最后一件事是正确处理CTRL-C(sigint)信号.我希望它的行为或多或少与Bash的行为方式相同:

• 在光标所在的位置打印^ C.
• 清除缓冲区以删除输入文本
• 跳到下一行,打印提示,然后等待输入

基本上:

/test $ bla bla bla|
# user types CTRL-C
/test $ bla bla bla^C
/test $ 

以下是作为可运行示例的简化代码:

import cmd
import signal


class TestShell(cmd.Cmd):
    def __init__(self):
        super().__init__()

        self.prompt = '$ '

        signal.signal(signal.SIGINT, handler=self._ctrl_c_handler)
        self._interrupted = False

    def _ctrl_c_handler(self, signal, frame):
        print('^C')
        self._interrupted = True

    def precmd(self, line):
        if self._interrupted:
            self._interrupted = False
            return ''

        if line == 'EOF':
            return 'exit'

        return line

    def emptyline(self):
        pass

    def do_exit(self, line):
        return True


TestShell().cmdloop()

这几乎可行.当我按下CTRL-C时,在光标处打印^ C,但我仍然需要按回车键.然后,该precmd方法通知self._interrupted处理程序设置的标志,并返回一个空行.这是我可以接受的,但我想以某种方式不必按此输入.

我想我不知何故需要强行input()归来,有人有想法吗？

Answer 1

我找到了一些 hacky 方法来使用 Ctrl-C 实现您想要的行为。

设置use_rawinput=False和替换stdin

这一个（或多或少……）坚持cmd.Cmd. 不幸的是，它禁用了 readline 支持。

您可以设置use_rawinput为 false 并传递一个不同的类文件对象来替换stdinin Cmd.__init__()。在实践中，仅readline()在此对象上调用。因此，您可以创建一个包装器来stdin捕获KeyboardInterrupt并执行您想要的行为：

class _Wrapper:

    def __init__(self, fd):
        self.fd = fd

    def readline(self, *args):
        try:
            return self.fd.readline(*args)
        except KeyboardInterrupt:
            print()
            return '\n'


class TestShell(cmd.Cmd):

    def __init__(self):
        super().__init__(stdin=_Wrapper(sys.stdin))
        self.use_rawinput = False
        self.prompt = '$ '

    def precmd(self, line):
        if line == 'EOF':
            return 'exit'
        return line

    def emptyline(self):
        pass

    def do_exit(self, line):
        return True


TestShell().cmdloop()

当我在终端上运行它时，Ctrl-C 显示^C并切换到新行。

猴子补丁 input()

如果你想要 的结果input()，除了你想要 Ctrl-C 的不同行为，一种方法是使用不同的函数而不是input()：

def my_input(*args):   # input() takes either no args or one non-keyword arg
    try:
        return input(*args)
    except KeyboardInterrupt:
        print('^C')   # on my system, input() doesn't show the ^C
        return '\n'

然而，如果你只是盲目地设置input = my_input，你会得到无限递归，因为my_input()will call input()，现在是它本身。但这是可以修复的，您可以__builtins__在cmd模块中修补字典以在以下input()期间使用修改后的方法Cmd.cmdloop()：

def input_swallowing_interrupt(_input):
    def _input_swallowing_interrupt(*args):
        try:
            return _input(*args)
        except KeyboardInterrupt:
            print('^C')
            return '\n'
    return _input_swallowing_interrupt


class TestShell(cmd.Cmd):

    def cmdloop(self, *args, **kwargs):
        old_input_fn = cmd.__builtins__['input']
        cmd.__builtins__['input'] = input_swallowing_interrupt(old_input_fn)
        try:
            super().cmdloop(*args, **kwargs)
        finally:
            cmd.__builtins__['input'] = old_input_fn

    # ...

注意这改变input()了所有Cmd的对象，而不仅仅是TestShell对象。如果这对你来说是不可接受的，你可以……

复制Cmd.cmdloop()源码并修改

因为你是子类化它，你可以cmdloop()做任何你想做的事情。“你想要的任何东西”可能包括复制Cmd.cmdloop()和重写其他部分。要么将调用替换为input()对另一个函数的调用，要么KeyboardInterrupt在重写的cmdloop().

如果您害怕底层实现会随着新版本的 Python 发生变化，您可以将整个cmd模块复制到一个新模块中，然后更改您想要的内容。