Python foo > bar(输入文件，输出文件)

Question

这可能是非常基本的问题，但我找不到任何答案。现在我有类似的东西:

import sys
inFile = sys.argv[1]
outFile = sys.argv[2]
with open(inFile, 'r+') as input,open(outFile,'w+') as out:
            #dosomething

我可以运行它./modulname foo bar(工作)。我如何更改它以便它可以与 /.modulname foo > bar 一起使用？ (现在它给我以下错误)。

./pagereport.py today.log > sample.txt
Traceback (most recent call last):
  File "./pagereport.py", line 7, in <module>
    outFile = sys.argv[2]
IndexError: list index out of range

Answer 1

您可以跳过第二次打开 (out)，而是使用 sys.stdout 进行写入。

如果您希望能够使用两种调用方式，argparse 可以通过组合 type= 与 add_argument 轻松实现此目的code> 到一个打开用于写入的文件，并将 sys.stdout 设为默认值。