我使用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标志,并返回一个空行.这是我可以接受的,但我想以某种方式不必按此输入.
我想我不知何故需要强制输入()返回,是否有人有想法？
设置use_rawinput = False并替换stdin
这个(或多或少……)粘在cmd.Cmd的公共接口上.不幸的是,它禁用了readline支持.
您可以将use_rawinput设置为false并传递一个不同的类文件对象来替换Cmd .__ init __()中的stdin.实际上,只在此对象上调用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()的结果,除了你想要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(),现在它本身就是.但这是可以修复的,你可以在cmd模块中修补__builtins__字典,以便在Cmd.cmdloop()期间使用修改后的input()方法：
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
# ...
请注意,这会更改所有Cmd对象的input(),而不仅仅是TestShell对象.如果这是你不能接受的,你可以……
复制Cmd.cmdloop()源并进行修改
由于你是子类化它,你可以让cmdloop()做你想做的任何事情. “你想要的任何东西”可能包括复制Cmd.cmdloop()的部分并重写其他部分.通过调用另一个函数替换对input()的调用,或者在重写的cmdloop()中捕获并处理KeyboardInterrupt.
如果您害怕使用新版本的Python更改底层实现,您可以将整个cmd模块复制到一个新模块中,并更改您想要的内容.