用Python编写生成树状结构的文件目录的脚本的教程
程序员文章站
2022-05-20 23:29:39
有时候需要罗列下U盘等移动设备或一个程序下面的目录结构的需求。基于这样的需求个人整理了一个使用Python的小工具,期望对有这方面需求的朋友有所帮助。以下为具体代码:
如...
有时候需要罗列下U盘等移动设备或一个程序下面的目录结构的需求。基于这样的需求个人整理了一个使用Python的小工具,期望对有这方面需求的朋友有所帮助。以下为具体代码:
如果你所有要求的文件目录不需要完整的文件路径的话,直接更换下面的注释代码即可~
# -*- coding:utf-8 -*- import os def list_files(startPath): fileSave = open('list.txt','w') for root, dirs, files in os.walk(startPath): level = root.replace(startPath, '').count(os.sep) indent = ' ' * 1 * level #fileSave.write('{}{}/'.format(indent, os.path.basename(root)) + '\n') fileSave.write('{}{}\\'.format(indent, os.path.abspath(root)) + '\n') subIndent = ' ' * 1 * (level + 1) for f in files: #fileSave.write('{}{}'.format(subIndent, f) + '\n') fileSave.write('{}{}{}'.format(subIndent, os.path.abspath(root), f) + '\n') fileSave.close() dir = raw_input('please input the path:') list_files(dir)