【Python】遍历文件夹中所有文件

---

一个小递归函数：遍历文件夹中所有的文件，返回值（路径）包含扩展名

---

import os
import sys

'''
    扫描文件夹dir中所有文件，返回文件的相对路径列表
'''
def GetFileList(dir, fileList):
    newDir = dir
    if os.path.isfile(dir):         # 如果是文件则添加进 fileList
        fileList.append(dir)
    elif os.path.isdir(dir):
        for s in os.listdir(dir):   # 如果是文件夹
            newDir = os.path.join(dir, s)
            GetFileList(newDir, fileList)
    return fileList

# 主函数

# 重定向输出位置
output = sys.stdout
outputfile = open('test.txt', 'w')
sys.stdout = outputfile

list = GetFileList('myFolder', []) # 获取所有myFolder文件夹下所有文件名称（包含拓展名）

# 输出所有文件夹中的路径（相对于当前运行的.py文件的相对路径）
for route in list:
    # route 为路径
    print(route)

# 关闭输出重定向
outputfile.close()
sys.stdout = output