用Python编写生成树状结构的文件目录的脚本的教程_Python

有时候需要罗列下U盘等移动设备或一个程序下面的目录结构的需求。基于这样的需求个人整理了一个使用Python的小工具，期望对有这方面需求的朋友有所帮助。以下为具体代码：

如果你所有要求的文件目录不需要完整的文件路径的话，直接更换下面的注释代码即可~

				?

									# -*- coding:utf-8 -*-

									import os

									def list_files(startPath):

									  fileSave = open(&#039;list.txt&#039;,&#039;w&#039;)

									  for root, dirs, files in os.walk(startPath):

									    level = root.replace(startPath, &#039;&#039;).count(os.sep)

									    indent = &#039; &#039; * 1 * level

									    #fileSave.write(&#039;{}{}/&#039;.format(indent, os.path.basename(root)) + &#039;\n&#039;)

									    fileSave.write(&#039;{}{}\\&#039;.format(indent, os.path.abspath(root)) + &#039;\n&#039;)

									    subIndent = &#039; &#039; * 1 * (level + 1)

									    for f in files:

									      #fileSave.write(&#039;{}{}&#039;.format(subIndent, f) + &#039;\n&#039;)

									      fileSave.write(&#039;{}{}{}&#039;.format(subIndent, os.path.abspath(root), f) + &#039;\n&#039;)

									  fileSave.close()

									dir = raw_input(&#039;please input the path:&#039;)

									list_files(dir)

用Python编写生成树状结构的文件目录的脚本的教程

相关文章

热门资讯