如何获取可变数量的文件作为python脚本的输入？

Question

例如，我希望能够运行以下假设命令：

$ python script.py *.txt option1 option2

并在与* .txt匹配的每个文件上执行

目前，我仅发现一次操作一个文件的信息

from sys import argv

self, file, option1, option2 = argv

perform_operation(file, option1, option2)

#function definition

Answer 1

您要使用argparse-module：

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("--option1")
parser.add_argument("--option2")
parser.add_argument("files", nargs="+")

opts = parser.parse_args()

print opts.option1
print opts.option2
print opts.files

像这样使用：

 beer:~ deets$ python2.7 /tmp/argparse-test.py  text foo bar baz
 None
 None
 ['text', 'foo', 'bar', 'baz']

Answer 2

argv是一个列表。 假设您仅要传递文件名参数。 如果更复杂，则请按Deets的回答进行。

self = sys.argv[0]
arguments = sys.argv[1:]

现在， arguments是程序参数的列表。 假设我们要一次处理它们：

for argument in arguments:
    work(argument)

或者我们想将它们全部传递给一个函数：

work(arguments)

至于传递*.txt作为参数。 您的外壳程序（甚至在程序运行之前）将为您完成大部分工作。

如果我运行python program.py *.txt ，其中*.txt指向3个文本文件，那么我的shell将会扩展为使我的程序看到python program.py a.txt b.txt c.txt 。

Answer 3

multifile.py

"""
Usage:
    multifile.py <file>...
    multifile.py -h

Prints something about all the <file>... files.
"""

def main(files):
    for fname in files:
        print fname

if __name__ == "__main__":
    from docopt import docopt
    args = docopt(__doc__)
    files = args["<file>"]
    main(files)

用它

首先安装docopt

$ pip install docopt

调用不带参数的命令：

$ python multifile.py
Usage:
    multifile.py <file>...
    multifile.py -h

尝试帮助

$ python multifile.py -h
Usage:
    multifile.py <file>...
    multifile.py -h

Prints something about all the <file>... files.

将其用于一个文件：

$ python multifile.py alfa.py 
alfa.py

使用通配符将它用于多个文件：

$ python multifile.py ../*.py

    ../camera2xml.py
    ../cgi.py
    ../classs.py

结论

• docopt允许更多选项（请参阅docopt ）
• 命令行解析在Python中很容易
  • 自2.7版以来， argparse似乎是Python的标准部分
  • argparse可以做很多事情，但是需要在许多行上进行相当复杂的调用
  • plac是不错的选择，在大多数情况下可以快速进行服务
  • 在我看来docopt在所需的代码行中最灵活，同时最短

Answer 4

使用stdlib中的inputfile

stdlib中有一个通常被忽略的库，称为inputfile 。

默认情况下，它会将命令行或标准输入中的所有输入作为文件名处理，并且不仅允许遍历这些文件，还允许遍历其中的所有行，对其进行修改，解压缩以及许多其他实际操作。

filenames.py列出所有文件名

import fileinput

for line in fileinput.input():
    print "File name is: ", fileinput.filename()
    fileinput.nextfile()

称它为：

$ python filenames.py *.txt
File name is: films.txt
File name is: highscores.txt
File name is: Logging.txt
File name is: outtext.txt
File name is: text.txt

upperlines.py用大写字母打印多个文件中的所有行

import fileinput

for line in fileinput.input():
    print line.upper(),

并称之为：

$ python upperlines.py *.txt
THE SHAWSHANK REDEMPTION (1994)
THE GODFATHER (1972)
THE GODFATHER: PART II (1974)
THE DARK KNIGHT (2008)
PULP FICTION (1994)
JAN HAS SCORE OF 101
PIETER HAS SCORE OF 900
CYRIL HAS SCORE OF 2
2014 APR 11  07:14:03.155  SECTORBLAH
   INTERESTINGCONTENT
   INTERESTING1 = 843
1. LUV_DEV <- HE'S A DEVELOPER
2. AMIT_DEV <- HE'S A DEVELOPER
....

upperlinesinplace.py将文件中的所有行都转换为大写

import fileinput

for line in fileinput.input(inplace=True):
    print line.upper(),

结论

• fileinput作为默认参数sys.argv[:1]并遍历所有文件和行
• 您可以传递自己的文件名列表进行处理
• fileinput允许就地更改，过滤，读取文件名，行号...
• fileinput甚至允许处理压缩文件