欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  后端开发

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

程序员文章站 2022-05-25 15:36:28
...

仅需更改 三个全局变量 即可 直接调用:

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

以上就是python中递归记录指定后缀名的文件示例分享的详细内容,更多请关注其它相关文章!