python生成目录树并显示文件大小

发布时间:2020-12-16编辑:脚本学堂
本文介绍了python生成目录树,并在目录树中显示文件大小的代码,python目录树生成的实现方法,感兴趣的朋友参考下。

本例生成目录树的方法,使用递归得到子文件夹以及文件,打印出树形,层次关系展示出来,完成目录树生成。

例如:
1--1
     2--1
          2
          3--1
               2
               3
     3--1
          2
          3
 
交错的层级关系,刚开始感觉很乱没有想明白,后来终于抓住了关键。只要算出每个层次的深度,就好办了。

定义了一个rank,进入一个子文件夹时,让rank+1,遍历完子文件夹rank就-1。
如图充分说明了递归、遍历的顺序以及rank值变化:(丑了点。。。)
<a href=http://www.jb200.com/jb/python/ target=_blank class=infotextkey>python</a>生成目录树1

代码:
 

复制代码 代码示例:

#!/usr/bin/python
# www.jb200.com
'''
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")

python 目录树生成的效果如下:
python生成目录树2