python 生成目录树及显示文件大小的代码

电玩女神 2022-03-25 07:06 189阅读 0赞

# python 生成目录树及显示文件大小的代码 #

发布：dxy 字体：\[ 增加  减小\] 类型：转载

没啥技术含量，大家都说没用，只不过算法还有点意思。自己憋出来的，不知道是不是跟别人的一样。做递归得到子文件夹以及文件并不难，但是能够打印出树形，层次关系展示出来，有些难度。

比如

1--1

2--1

3--1

3  
  
交错的层级关系，刚开始感觉很乱没有想明白，后来终于抓住了关键。只要算出每个层次的深度，就好办了。   
  
我定义了一个rank，进入一个子文件夹时，让rank+1，遍历完子文件夹rank就-1。   
  
如图充分说明了递归、遍历的顺序以及rank值变化：（丑了点。。。）   
![在新窗口打开图片][20090723233213730.JPG]  
下面放代码：

复制代码代码如下:

'''   
Created on Jul 22, 2009   
  
@author: dirful   
'''   
import os   
class dir(object):   
  
def \_\_init\_\_(self):   
self.CONST =0   
self.SPACE =""   
self.list =\[\]   
def p(self,url):   
files = os.listdir(r''+url)   
for file in files:   
myfile = url + "\\\\"+file   
size = os.path.getsize(myfile)   
if os.path.isfile(myfile):   
self.list.append(str(self.SPACE)+"|\_\_\_\_"+file +" "+ str(size)+"\\n")   
\# print str(self.SPACE)+"|\_\_\_\_"+file +" "+ str(size)   
  
if os.path.isdir(myfile) :   
self.list.append(str(self.SPACE)+"|\_\_\_\_"+file + "\\n")   
\#get into the sub-directory,add "| "   
self.SPACE = self.SPACE+"| "   
self.p(myfile)   
\#when sub-directory of iteration is finished,reduce "| "   
self.SPACE = self.SPACE\[:-5\]   
return self.list   
  
def writeList(self,url):   
f = open(url,'w')   
f.writelines(self.list)   
print "ok"   
f.close()   
  
  
  
if \_\_name\_\_ == '\_\_main\_\_':   
d=dir()   
d.p("E:/eclipse")   
d.writeList("c:3.txt")

生成树如下。没有微软tree生成的好。。。。。。。

![20090723233213994.jpg][]

[20090723233213730.JPG]: http://www.jb51.net/upload/20090723233213730.JPG
[20090723233213994.jpg]: http://www.jb51.net/upload/20090723233213994.jpg