Python中的子文件夹如何通过文件夹迭代

Question

我有一个装有数千个.AI文件的文件夹。该文件夹的安排方式最初是带有客户名称的子文件夹，并且在每个子文件夹中，每个子文件夹是一个包含.AI的某些或多个子文件夹的唯一目录，或没有子文件夹只是.ai文件。

我需要一个程序，可以通过将每个.AI文件名在客户子文件夹内部（不管有多少个子文件夹或subs等人等）中迭代，并将其附加到列表中。然后，我将列出该列表，然后稍后再做一些OCR，但是一旦完成，我将清除列表并继续进入下一个子文件夹。

这是我试图尝试的代码，但失败了。它有时会返回一个空列表，或其中只有一个文件名的列表，当时它应该每次在其中带有一个或多个.AI文件名返回列表。

def folder_loop(folder):
    temp_list = []
    for root, dirs, files in os.walk(folder):
        for dir in dirs:
            for file in dir:
                if file.endswith("ai"):
                    temp_list.append(os.path.join(root, file))
        print(temp_list)
        temp_list.clear()

我是一个野生动物，我几乎不明白代码在做什么，所以我并不感到惊讶，它无效。有什么想法吗？

Answer 1

您可以尝试以下操作：

如果您想将 base 文件夹给出所有客户文件夹的位置，然后为每个客户文件夹中的所有列表都需要。

from pathlib import Path

def folder_loop(folder):
    for path in Path(folder).iterdir():
        if path.is_dir():
            yield list(path.rglob("*.ai"))

​path.rglob（“*。ai”） 是递归 globbing 给定的路径及其所有子文件夹的.ai -files。

要使用它：

the_folder = "..."
for file_list in folder_loop(the_folder):
    print(file_list)
    # do whatever you want to do with the files

如果您想给它一个文件夹，并且想要一个列表，其中包含所有.ai文件：

def folder_loop(folder):
    return list(Path(folder).rglob("*.ai"))

此处的屈服/返回列表中包含 path> path -Objects（非常方便）。如果您想要字符串，那么您可以这样做

       ....
            yield list(map(str, path.rglob("*.ai")))

等。

Answer 2

有一个社区帖子在这里答案。

话虽如此，我在个人实用程序工具箱中有以下方法。

def get_files_from_path(path: str=".", ext=None) -> list:
    """Find files in path and return them as a list.
    Gets all files in folders and subfolders

    See the answer on the link below for a ridiculously
    complete answer for this.
    https://stackoverflow.com/a/41447012/9267296
    Args:
        path (str, optional): Which path to start on.
                              Defaults to '.'.
        ext (str/list, optional): Optional file extention.
                                  Defaults to None.

    Returns:
        list: list of full file paths
    """
    result = []
    for subdir, dirs, files in os.walk(path):
        for fname in files:
            filepath = f"{subdir}{os.sep}{fname}"
            if ext == None:
                result.append(filepath)
            elif type(ext) == str and fname.lower().endswith(ext.lower()):
                result.append(filepath)
            elif type(ext) == list:
                for item in ext:
                    if fname.lower().endswith(item.lower()):
                        result.append(filepath)
    return result