目录中的快速文件夹搜索 [Python]

Question

情况：主目录中有很多文件和文件夹，非常旧的工作计算机，旧的 python 2.something（软件绑定，所以现在无需执行任何操作），所有数据都在网络驱动器上。

我有一个变量，其中包含它应该找到的文件夹名称，每个项目中有大约 2-4 个这样的文件夹名称。对于 ShotName = "DVX_010_400_BG_V001"

预期输出：

C:\Users\Dame\Desktop\NUKESTUFF\TPN\FilmLance\DVX\3dcomp\DVX_010_400_BG_V001

C:\Users\Dame\Desktop\NUKESTUFF\TPN\FilmLance\DVX\input\DVX_010_400_BG_V001

C:\Users\Dame\Desktop\NUKESTUFF\TPN\FilmLance\DVX\render\DVX_010_400_BG_V001

目标：找到与 ShotName 变量同名的文件夹并返回其路径

问题：当前代码太慢并且挂起。

在我的桌面环境中，主文件夹位于 dekstop（快速 NVME ssd）上，最快的处理器，现代 python，所以速度很快，大约 2 秒。我最好的代码：

import os

ShotName = "DVX_010_400_BG_V001"
head_dir= "C:\\Users\\Dame\\Desktop\\NUKESTUFF\\TPN\\"


def dir_list_folder(head_dir, ShotName):
    """Return a list of the full paths of the subdirectories
    under directory 'head_dir' named 'ShotName'"""
    dirList = []
    for fn in os.listdir(head_dir):
        dirfile = os.path.join(head_dir, fn)
        if os.path.isdir(dirfile):
            print (dirfile)
            if fn.upper() == ShotName.upper():
                dirList.append(dirfile)
            else:
                dirList += dir_list_folder(dirfile, ShotName)


    return dirList

if __name__ == '__main__':
    for item in dir_list_folder(head_dir, ShotName):
        # print (item)

        if "render" in item :
            ProjectPath = item.split("\\")[:-2]
            ProjectPath = "/".join(ProjectPath)
            print (f"ProjectPath = {ProjectPath}")

Answer 1

你尝试过 os.walk 吗？您可以这样使用 os.walk：

def FindFolder(Path, WantedFolder):
    corresponding = []
    for root, _, _ in os.walk(Path):
        
        if root.split("\\")[-1] == WantedFolder:
            corresponding.append(root)

    return corresponding

您可能必须根据系统的架构来进行分割。该解决方案对于 odl 系统来说应该非常有效。