在atexit中调用input()触发ValueError：关闭文件上的I/O操作

这并非常见的“如何保持Python脚本输出窗口打开”问题的重复。这类问题的标准解决方案是调用input()等待用户输入，我已经使用多年，但这里发现了一个未被提及的问题。

我有一个在Windows下运行、向终端输出内容的程序：从资源管理器启动时会打开终端窗口，但程序运行结束后窗口会自动关闭。为此我添加了命令行选项--pause，模拟批处理中pause命令的效果——程序运行结束后输出“Press enter to continue”并等待，方便用户查看输出的警告或结果信息。

实现方式 #

1. 在if __name__ == '__main__':代码块之前定义暂停函数：

def console_pause(args):
    if args.pause:
        input("Press enter to continue")

2. 在if __name__ == '__main__':作用域内注册退出回调：

atexit.register(console_pause, args)

问题现象 #

这个方案在资源管理器中运行完全正常，但在批处理脚本中运行时会出现错误：
程序正常输出内容后：

Round-trip yielded identical result
Wrote filename.pkl containing 194128 rows × 7 columns
Wrote filename2.pkl containing 194128 rows × 7 columns
Wrote filename3.pkl containing 194128 rows × 7 columns
Press enter to continue

按下回车后，终端抛出错误：

Press enter to continueException ignored in atexit callback: <function console_pause at 0x000001B7FF2A0FE0>
Traceback (most recent call last):
  File "scriptname.py", line 611, in console_pause
    input("Press enter to continue")
ValueError: I/O operation on closed file.

已知分析与排除的方案 #

我大致理解问题根源：程序结束过程中，标准输入(stdin)在input()调用返回前就被关闭了。之前没发现这个问题是因为按下回车后窗口立即关闭，根本看不到错误信息——这正是我想要避免的情况。

我需要明确的具体错误原因，以及直接解决问题的替代方案，而非转移问题的思路。目前已有几个思路但都不符合需求：

• 不认同将暂停代码放在__main__末尾：如果程序因异常终止，__main__末尾的代码不会执行，用户无法查看异常回溯信息，而atexit回调会在异常终止时触发
• 否决系统级方案（比如批处理中加pause）：一是需要保持执行中暂停和结束时暂停的一致性（执行中用input()），二是我需要暂停逻辑依赖命令行参数，批处理的pause无法实现这一点
• 尝试过用try...finally包裹主代码：异常发生时，finally中的input()会正常执行，但用户按下回车后，解释器才输出堆栈跟踪，窗口随即关闭，用户根本来不及查看

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