如何在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启动脚本之后，这也允许从终端读取。

无论如何，除了非常特殊的用例外，我强烈建议使用命令行参数，以允许程序从指定文件中读取，然后通常从stdin读取。