Python glob:如何输入目录和单个文件

时间:2018-02-04 08:32:48

标签: python python-3.x input hierarchy glob

这是一个简单的问题。

我正在使用glob来使用以下代码打印文件夹的完整层次结构:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import argparse
import sys
import glob

parser = argparse.ArgumentParser()                              
parser.add_argument('-input', dest='input',help="input one or more files",metavar=None)                     
args = parser.parse_args()

def dirlist(path):

    for i in glob.glob(os.path.join(path, "*")):
        if os.path.isfile(i):
            print (i)

        elif os.path.isdir(i):
            dirname = os.path.basename(i)               
            dirlist(i)

path = os.path.normpath(args.input)
dirlist(path)

效果很好,你只需要运行python3 Test.py -input .

正如您在argparse帮助描述中所看到的,我想输入目录,但也输入单个文件。

我不认为可以使用glob来完成,是吗?

你能建议我一个可以帮助我打印目录和文件的完整层次结构的库吗?

我找到了here一长列globe示例,但它们似乎都适用于目录,而不是输入单个文件

1 个答案:

答案 0 :(得分:0)

快到午夜了,但至少我在2天后找到了解决方案:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import argparse
import sys
import glob

parser = argparse.ArgumentParser()                              
parser.add_argument('-input', dest='input',help="input one or more files",metavar=None)                     
args = parser.parse_args()

def dirlist(path):

    if os.path.isfile(path):
        print (path)
    elif os.path.isdir(path):
        for i in glob.glob(os.path.join(path, "*")):
            if os.path.isfile(i):
                print (i)

            elif os.path.isdir(i):
                dirname = os.path.basename(i)               
                dirlist(i)


path = os.path.normpath(args.input)
dirlist(path)