遍历获取一个路径下所有子文件夹及文件

遍历获取一个路径下所有子文件夹及文件

# -*- coding:utf-8 -*-
import os


def get_dir_file(rootdir):
    objects = os.listdir(rootdir)
    dir_list = [] #存放目录列表
    file_list = [] #存放文件列表
    for obj in objects:
        judge_dir = os.path.isdir(os.path.join(rootdir, obj))
        if judge_dir:#判断是否是目录os.path.join()用来将路径拼接
            dir_list.append(os.path.join(rootdir, obj))#保存时保存完整路径才能对其进行后续操作
            # print ("dir:",obj)
        else:
            file_list.append(os.path.join(rootdir, obj))
            # print ("file:",obj)
    # print ("目录列表：",dir_list)
    # #如果项判断是否是文件时用os.isfile()
    # print ("文件列表：",file_list)
    if len(dir_list)>0:
        for dir_list_i in dir_list:
            file_list_i = get_dir_file(dir_list_i)[1]
            for file_i in file_list_i:
                file_list.append(file_i)
    return (dir_list,file_list)

def main():
	path = '/zy_DATA/zy/test'
	result = get_dir_file(path)
	print(result[0])   # 包含的子目录文件夹
	print(result[1])   # 所有文件

if __name__ == "__main__":
	main()