遍历目录中的所有文件和目录，并生成全路径

# -*- coding:UTF-8 -*-

"""
遍历目录中的所有文件和目录，并生成全路径
"""

import os

target_path = "D:/temp/"

'''
path: 遍历的路径
file_type: 文件类型列表，如果为空遍历所有文件，不为空遍历指定文件如[".c", ".h", ".py"]等
'''
def generate_file_list(path, file_type=[]):
    walks = os.walk(path)
    for walk in walks:
        for file in walk[2]:
            if not file_type:   # empty
                yield walk[0] + "/" + file
            else:
                root, ext = os.path.splitext(file)
                if ext in file_type:
                    yield walk[0] + "/" + file

def generate_dir_list(path):
    walks = os.walk(path)
    for walk in walks:
        for dir_name in walk[1]:
            yield walk[0] + "/" + dir_name

for file in generate_file_list(target_path, filetype=[".txt", ".c"]):
    print(file)

# for dir_name in generate_dir_list(target_path):
#     print(dir_name)

转载于:https://www.cnblogs.com/wangshide/p/4612068.html