python中递归记录指定后缀名的文件示例分享

# coding=utf-8

import os

# 三个全局变量：

root_folder = './root'   # 将被递归的文件夹根目录

save_txt = './paths.txt' # 记录路径的文档

suffix_name = '.jpg'     # 后缀名

# 递归记录指定后缀名的文件的绝对路径

def record(folder, save_txt):

    save_file = open(save_txt, 'a')

    for name in os.listdir(folder):

        if os.path.isdir(os.path.join(folder, name)):

            record(os.path.join(folder, name), save_txt)

        elif name.endswith(suffix_name):

            save_file.write('{}\n'.format(os.path.join(folder, name)))

    save_file.close()

def main():

    # 如果save_txt已存在，则删除

    try:

        os.remove(save_txt)

    except OSError:

        pass

    # 开始递归记录

    record(os.path.abspath(root_folder), save_txt)

if __name__ == '__main__':

    main()