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)