Python：使用Glob查找特定的子文件夹

Question

I am looking to recursively search through a folder containing many sub-folders. Some sub-folders contain a specific folder that I want to loop through.

I am familiar with the glob.glob method to find specific files:

import glob, os
from os import listdir
from os.path import isfile, join

os.chdir(pathname) #change directory to path of choice
files = [f for f in glob.glob("filename.filetype") if isfile(join(idir, f))]

Some sub folders in a directory have a time stamp (YYYYMMDD) as their names all containing identical file names. Some of those sub folders contain folders within them with a name, let's call it "A". I'm hoping to create a code that will recursively search for the folder within called "A" within these "specific sub folders". Is there a way to use glob.glob to find these specific sub-folders within a directory?

I am aware of a similar question: How can I search sub-folders using glob.glob module in Python?

but this person seems to be looking for specific files, whereas I am looking for pathnames.

Answer 1

You can use os.walk which will walk the tree. Each iteration shows you the directory and its immediate subdirectories, so the test is simple.

import os
import re

# regular expression to match YYYYMMDD timestamps (but not embedded in
# other numbers like 2201703011).
timestamp_check = re.compile(re.compile(r"[^\d]?[12]\d3[01]\d[0123]\d")).search

# Option 1: Stop searching a subtree if pattern is found
A_list = []
for root, dirs, files in os.walk(pathname):
    if timestamp_check(os.path.basename(root)) and 'A' in dirs:
        A_list.append(os.path.join(root, A))
        # inplace modification of `dirs` trims subtree search
        del dirs[:]

# Option 2: Search entire tree, even if matches found
A_list = [os.path.join(root, 'A') 
    for root, dirs, files in os.walk(pathname) 
    if timestamp_check(os.path.basename(root)) and 'A' in dirs]