glob.glob

时间:2015-05-01 18:04:59

标签: python glob

我试图为其中一个CLI脚本提供通配符支持,我正在使用pythons glob模块。对于测试,我尝试了这个:

>>> import glob
>>> for f in glob.glob('/Users/odin/Desktop/test_folder/*.log'):
...     print f
...
/Users/odin/Desktop/test_folder/test1.log
/Users/odin/Desktop/test_folder/test2.log
/Users/odin/Desktop/test_folder/test3.log

这完美地工作,并提供正确的输出,因为我有上面给出的3个文件。 但是当我在CLI的论证中使用相同的代码时,它失败了。我这样做

#code...
parser.add_argument( "-f", "--file", type=str, help="Full path to the file to upload." )
#code...
if args.file:
    for f in glob.glob(args.file):
        _upload_part(f)

我将其作为

运行
python cli.py -f /Users/odin/Desktop/test_folder/*.log

这给了我错误:

cli.py: error: unrecognized arguments: /Users/odin/Desktop/test_folder/test2.log /Users/odin/Desktop/test_folder/test3.log

当我逐一浏览列表时,我不明白为什么所有文件都被立即添加到参数中。

编辑 -

nargs朝着正确的方向迈出了一步,但现在出现了这个错误:

 `Traceback (most recent call last):
      File "cli.py", line 492, in <module>
        for f in glob.glob(args.file):
      File "/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/glob.py", line 27, in glob
        return list(iglob(pathname))
      File "/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/glob.py", line 38, in iglob
        dirname, basename = os.path.split(pathname)
      File "/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/posixpath.py", line 92, in split
        i = p.rfind('/') + 1
    AttributeError: 'list' object has no attribute 'rfind'`

1 个答案:

答案 0 :(得分:3)

这是您的脚本收到的内容:

python cli.py -f file1 file2 file3 fileN

其中 N 是该模式匹配的文件总数(Shell会自动扩展通配符)。另一方面,您的文件参数配置为只接收一个文件/模式,因此最简单(也是最好的,在我看来)解决方案是添加nargs='+' argument

parser.add_argument('-f', '--file', type=str, help='...', nargs='+')
# code
for f in args.file:
    print(f)

这样您就可以删除所有glob次来电和if args.file次来电。

另一个选择是引用命令行参数:

python cli.py -f '/Users/odin/Desktop/test_folder/*.log'