在Python中将argparse与sys.argv结合使用

时间:2011-12-07 23:03:07

标签: python command-line-arguments glob argparse

我目前有一个脚本,它通过sys.argv变量使用文件globbing,如下所示:

if len(sys.argv) > 1:
        for filename in sys.argv[1:]:

这非常适合处理一堆文件;但是,我也想在argparse模块中使用它。所以,我希望我的程序能够处理如下内容:

foo@bar:~$ myScript.py --filter=xyz *.avi

有没有人试图这样做,或者有一些关于如何继续的指示?

2 个答案:

答案 0 :(得分:16)

如果我说得对,那么你的问题就是将一个文件列表与一些标志或可选参数一起传递给命令。如果我找对你,那么你只需要利用argparse中的参数设置:

档案 p.py

import argparse

parser = argparse.ArgumentParser(description='SO test.')
parser.add_argument('--doh', action='store_true')
parser.add_argument('files', nargs='*')  # This is it!!
args = parser.parse_args()
print(args.doh)
print(args.files)

上面的注释行告知解析器期望未定义的数字> = 0(nargs ='*')位置参数。

从命令行运行脚本会产生以下输出:

$ ./p.py --doh *.py
True
['p2.py', 'p.py']
$ ./p.py *.py
False
['p2.py', 'p.py']
$ ./p.py p.py
False
['p.py']
$ ./p.py 
False
[]

观察文件在列表中的位置,无论它们是多少还是只有一个。

HTH!

答案 1 :(得分:1)

或者,您可以通过以下方式使用两者:

import sys
import argparse
if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("-v", "--verbose", help="increase verbosity", action="store_true")
    args, unknown = parser.parse_known_args()
    for file in sys.argv:
        if not file.startswith("-"):
            print(file)

然而,这仅适用于独立参数,否则参数值将被视为文件参数(除非您不将它们与空格分开,否则您将进一步改进代码)。