如何从 Python 脚本中的文件和终端读取输入？

Question

我有一个 Python 脚本，它使用 shell 重定向从文件中读取一些数据，例如

python script.py < input.txt

处理完文件后，我想允许用户从终端输入一些参数。我怎样才能做到这一点？

Answer 1

重定向持续到进程终止。所以你不能做你想做的事。相反，使用命令行参数指定您的输入文件，如下所示：python script.py --input input.txt用于argparse从命令行中选择名称并像读取普通文件一样读取它。这样就可以sys.stdin自由地做你想做的交互事情了input()。

Answer 2

正如您已将问题标记为bash，我想我可以说包含明确打开的技巧/dev/tty。这通常仅用于询问不应存在于文件中的机密的程序。但是，即使在使用重定向的标准输入启动脚本之后，这也允许从终端读取。无论如何，除了非常特殊的用例，我强烈建议改用命令行参数，以允许程序从指定的文件中读取，然后正常从 stdin 中读取。