如何更改Windows控制台窗口上X按钮的功能？

Question

我在Python中有一个常规的控制台窗口（Windows 8.1），我想更改close / X按钮的功能。 游戏结束时，需要进行一些清理工作，例如，我该如何覆盖它呢？

Answer 1

您可以通过使用SetConsoleCtrlHandler指定指定用于Ctrl-C，Ctrl-Break，控制台退出和系统关闭事件的函数来拦截控制台关闭事件。 对于控制台退出事件，您不能阻止控制台被破坏，但是您可以拦截该事件并运行一些代码。 例如：

import sys
import time
import win32api

forever = True
handled_ctrl_type = None

def handler(ctrl_type):
    global forever, handled_ctrl_type
    forever = False
    handled_ctrl_type = ctrl_type
    print("HANDLER")
    return True

def main(args = None):
    win32api.SetConsoleCtrlHandler(handler, True)
    n = 0
    while forever:
        time.sleep(1)
        print(n)
        n = n + 1
    print("exiting...", handled_ctrl_type)
    return 0

if __name__ == '__main__':
    sys.exit(main())

如果这是在控制台中运行（使用Python 3），并且您按了Ctrl-C，则应该得到以下内容：

c:\src>python python\setconsolectrl.py
0
HANDLER
1
exiting... 0

其中0是CTRL_C_EVENT的值。 对于控制台出口，我们应该在此处设置2，以便我们可以判断生成了哪种事件。

如果您运行此程序并按红色大X来关闭控制台，则可以看到控制台被销毁时正在打印HANDLER。 您应该在文件中添加一些日志记录，以确保真正调用了所有清理操作，但这看起来应该可以解决您的问题。