用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)

您可能感兴趣的文章:

• Python使用scrapy抓取网站sitemap信息的方法
• 用Python解析XML的几种常见方法的介绍
• python操作xml文件详细介绍
• 使用PYTHON创建XML文档
• Python 解析XML文件
• python使用xmlrpc实例讲解
• 使用Python脚本生成随机IP的简单方法
• Python实现的生成自我描述脚本分享(很有意思的程序)
• 用Python脚本生成Android SALT扰码的方法
• 利用Python脚本生成sitemap.xml的实现方法