代码如下, 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]