Python 8行代码实现查找目录下所有子文件, 子目录

代码如下, findall_files 表示获得目标文件夹下所有子文件 (包括任意深度的子目录中的), findall_subdirs 表示获得目标文件夹下所有子目录:

# file_finder.py
import os


def findall_files(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend([os.path.join(root, f) for f in files])
    
    return collector


def findall_subdirs(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend(os.path.join(root, d) for d in dirs)
    
    return collector

注意: os.path.join 在 windows 上是通过反斜杠连接的, 比如我的 root 是 “…/A”, 我的 file 是 “B.txt”, 则连接的结果是 “…/A\B.txt”.

如果想统一用正斜杠的话, 可以用下面改良后的代码:

# file_finder.py
import os


def findall_files(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend([os.path.join(root, f) for f in files])
    
    return [x.replace('\\', '/') for x in collector]


def findall_subdirs(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend(os.path.join(root, d) for d in dirs)
    
    return [x.replace('\\', '/') for x in collector]