在未从现有终端调用时阻止控制台应用关闭？

Question

这类问题有很多变种.但是我特别想办法防止Python中的控制台应用程序在没有从终端(或其他控制台调用,因为它可能在Windows上调用)时调用.可能发生这种情况的一个示例是双击.pyWindows资源管理器中的文件.

通常我使用类似下面的代码片段,但即使从现有终端调用应用程序,它也会产生令人遗憾的副作用:

def press_any_key():
    if os.name == "nt":
        os.system("pause")
atexit.register(press_any_key)

它还假设所有Windows用户都从Windows"shell"调用该应用程序,并且只有Windows用户才能从现有终端以外的位置执行该程序.

是否有(最好是跨平台)方式来检测我的应用程序是否已从终端调用,和/或是否有必要为当前运行的实例提供"按任意键..."功能？请注意,采用批处理,bash或任何其他"包装器进程"解决方法是非常不受欢迎的.

Update0

使用下面的Alex Martelli的答案,我已经产生了这个功能:

def register_pause_before_closing_console():
    import atexit, os
    if os.name == 'nt':
        from win32api import GetConsoleTitle
        if not GetConsoleTitle().startswith(os.environ["COMSPEC"]):
            atexit.register(lambda: os.system("pause"))

if __name__ == '__main__':
    register_pause_before_closing_console()

如果出现其他合适的答案,我会为其他平台和桌面环境添加更多代码.

UPDATE1

在使用pywin32的过程中,我已经使用接受的答案生成了这个函数,该函数改进了上面的函数.注释掉的代码是源自Update0的替代实现.如果不能使用pywin32,请按照接受的答案中的链接进行操作.暂停或getch()品尝.

def _current_process_owns_console():
    #import os, win32api
    #return not win32api.GetConsoleTitle().startswith(os.environ["COMSPEC"])

    import win32console, win32process
    conswnd = win32console.GetConsoleWindow()
    wndpid = win32process.GetWindowThreadProcessId(conswnd)[1]
    curpid = win32process.GetCurrentProcessId()
    return curpid == wndpid

def register_pause_before_closing_console():
    import atexit, os, pdb
    if os.name == 'nt':
        if _current_process_owns_console():
            atexit.register(lambda: os.system("pause"))

if __name__ == '__main__':
    register_pause_before_closing_console()

Answer 1

首先,试图阻止你从聪明的黑客.设置一个单独的快捷方式是非常合适的,该快捷方式设计为从资源管理器运行,从命令行使用的脚本中执行稍微不同的操作(例如保持控制台打开).正如亚历克斯已经指出的那样,这不是nix上的一个问题,而且正确的做法总是彻底退出,否则你的用户会抱怨.

如果您仍然需要一种解决方法,这里的代码可以检测何时需要阻止控制台关闭时相当干净.需要Windows 2000或更高版本,逻辑包含在此函数中:

def owns_console():
    wnd = GetConsoleWindow()
    if wnd is None:
        return False
    return GetCurrentProcessId() == GetWindowThreadProcessId(wnd)

基本上,它获取拥有Python正在使用的控制台的进程的PID,以及我们的进程.如果它们是相同的,那么当我们退出控制台时会消失,所以它需要保持打开状态.如果它们不同,或者没有连接控制台,Python应该正常退出.