python 組建目錄樹及顯示檔案大小的代碼

比如

1--1

2--1

2

3--1

2

3

3--1

2

3

交錯的層級關係，剛開始感覺很亂沒有想明白，後來終於抓住了關鍵。只要算出每個層次的深度，就好辦了。

我定義了一個rank，進入一個子檔案夾時，讓rank+1，遍曆完子檔案夾rank就-1。

充分說明了遞迴、遍曆的順序以及rank值變化：（醜了點。。。）

下面放代碼：

代碼如下:

'''
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產生的好。。。。。。。

