仅需更改 三个全局变量 即可 直接调用:
# 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()