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

Python glob: how input directories and single files

这是一个简单的问题。

我使用 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 示例,但它们似乎都适用于目录,而不是当您输入单个文件时

快午夜了,但至少我在 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)