在Python cmd模块

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

我发现了一些使用Ctrl-C实现所需行为的hacky方法。

设置use_rawinput=False并替换stdin

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

您可以将use_rawinput设置为false，并传递一个不同的类似文件的对象，以替换stdin中的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并切换到新行。

Monkey-patch 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()会调用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()

请注意，这会更改{strong>所有Cmd个对象的TestShell，而不只是Cmd.cmdloop()个对象。如果你不能接受，你可以......

复制cmdloop()来源并对其进行修改

由于您正在对其进行子类化，因此您可以Cmd.cmdloop()执行任何操作。 ＆＃34;你想要的任何东西＆＃34;可能包括复制input()的部分并重写其他部分。通过调用另一个函数替换对KeyboardInterrupt的调用，或者在重写的cmdloop()中捕获并处理cmd。

如果您害怕使用新版本的Python更改底层实现，您可以将整个ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"模块复制到一个新模块中，并更改您想要的内容。