Python中cmd.Cmd.cmdloop()的KeyboardInterrupt无法即时触发问题求助

按下Ctrl+C后需额外输入按键才会触发异常，无法立即清除控制台，现有代码如下：

import cmd

class MyCommandPrompt(cmd.Cmd):
    def __init__(self):
        cmd.Cmd.__init__(self)
        self.stop = False

    def emptyline(self) -> None:
        """Do nothing on empty input line"""
        pass

    def cmdloop(self, intro=None):
        """Boucle de commandes"""
        while not self.stop:
            try:
                super().cmdloop(intro="")
                self.postloop()
            except KeyboardInterrupt:
                os.system("cls")

if __name__ == "__main__":
    MyCommandPrompt().cmdloop()

问题原因 #

cmd.Cmd默认的cmdloop内部输入逻辑依赖终端缓冲，接收到Ctrl+C后不会立即抛出异常，需等待下一次输入操作才会触发信号处理。

解决方案 #

方案一：自定义输入循环，直接捕获异常 #

重写cmdloop，自行处理输入读取，确保KeyboardInterrupt被立即捕获：

import cmd
import os
import sys

class MyCommandPrompt(cmd.Cmd):
    def __init__(self):
        super().__init__()
        self.stop = False
        self.prompt = ">>> "

    def emptyline(self) -> None:
        """空输入行不执行任何操作"""
        pass

    def cmdloop(self, intro=None):
        """自定义命令循环"""
        if intro:
            self.preloop()
            print(intro)
        
        while not self.stop:
            try:
                line = input(self.prompt)
                if line:
                    self.onecmd(line)
            except KeyboardInterrupt:
                # 跨平台清屏
                os.system("cls" if os.name == "nt" else "clear")
                print("\n控制台已清除")
            except EOFError:
                self.stop = True
                print("\n退出程序")
        
        self.postloop()

if __name__ == "__main__":
    MyCommandPrompt().cmdloop()

方案二：注册SIGINT信号处理函数 #

通过信号捕获机制，在收到Ctrl+C时立即执行清屏操作，无需修改核心循环逻辑：

import cmd
import os
import signal

class MyCommandPrompt(cmd.Cmd):
    def __init__(self):
        super().__init__()
        self.stop = False
        self.prompt = ">>> "
        # 注册SIGINT信号处理器
        signal.signal(signal.SIGINT, self.handle_ctrl_c)

    def handle_ctrl_c(self, signum, frame):
        # 清屏
        os.system("cls" if os.name == "nt" else "clear")
        print("\n控制台已清除")
        # 重置提示符，避免输入位置混乱
        self.stdout.write(self.prompt)
        self.stdout.flush()

    def emptyline(self) -> None:
        pass

if __name__ == "__main__":
    MyCommandPrompt().cmdloop()

方案说明 #

• 方案一完全接管输入流程，能确保异常被立即捕获，逻辑清晰可控，适合需要自定义输入行为的场景。
• 方案二利用信号处理机制，对原有代码侵入性低，仅需添加信号处理函数即可实现需求。

内容的提问来源于stack exchange，提问作者Remi_Avec_Un_I