Python Glob：如何输入目录和单个文件

Question

这是一个简单的问题。

我正在使用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做到这一点，对吗？

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

我在这里找到了globe示例，但它们似乎都适用于目录，而不是在您输入单个文件时

Answer 1

已经快午夜了，但至少两天后我找到了解决方案：

#!/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)