用Python编写生成树状结构的文件目录的脚本的教程
有时候需要罗列下U盘等移动设备或一个程序下面的目录结构的需求。基于这样的需求个人整理了一个使用Python的小工具,期望对有这方面需求的朋友有所帮助。以下为具体代码:
如果你所有要求的文件目录不需要完整的文件路径的话,直接更换下面的注释代码即可~
#-*-coding:utf-8-*-
importos
deflist_files(startPath):
fileSave=open('list.txt','w')
forroot,dirs,filesinos.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)
forfinfiles:
#fileSave.write('{}{}'.format(subIndent,f)+'\n')
fileSave.write('{}{}{}'.format(subIndent,os.path.abspath(root),f)+'\n')
fileSave.close()
dir=raw_input('pleaseinputthepath:')
list_files(dir)