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

# -*- 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)