python递归获取目录下指定文件

Posted 2020-11-30 Mars.wang

获取一个目录下所有指定格式的文件是实际生产中常见需求.

import os
#递归获取一个目录下所有的指定格式的文件

def get_jsonfile(path,file_list):
    dir_list=os.listdir(path)
    for x in dir_list:
        new_x=os.path.join(path,x)
        if os.path.isdir(new_x):
            get_jsonfile(new_x,file_list)
        else:
            file_tuple=os.path.splitext(new_x)
            if file_tuple[1]==‘.json‘:
                file_list.append(new_x)
    return file_list


if __name__==‘__main__‘:
    file_list=[]
    path=‘/Users/binwang/Documents‘
    get_jsonfile(path,file_list)
    for json_file in file_list:
        print(json_file)