用Python生成目錄樹

堆碼狂人

這是一個(gè)使用Python生成文件、目錄樹的程序，其中遇到一個(gè)問題是：如何確定某個(gè)目錄下的某一文件是最后一個(gè)遍歷的文件。因?yàn)樽詈笠粋�(gè)遍歷的文件前應(yīng)添加"└─"，非最后一個(gè)文件前添加"├─"�？戳薖ython的API文檔沒有找到相關(guān)的系統(tǒng)函數(shù)。現(xiàn)在做法是：先統(tǒng)計(jì)出某個(gè)目錄下的文件個(gè)數(shù)，在遍歷目錄時(shí)，當(dāng)個(gè)數(shù)相等時(shí)，就可以確定該目錄遍歷結(jié)束。

1. # encoding: utf-8
   
2. 
   
3. import os
   
4. class dir(object):
   
5.         def __init__(self):
   
6.                 self.SPACE = ""
   
7.                 self.list = []
   
8.        
   
9.         def getCount(self, url):
   
10.                 files = os.listdir(url)
    
11.                 count = 0;
    
12.                 for file in files:
    
13.                         myfile = url + "\\" + file
    
14.                         if os.path.isfile(myfile):
    
15.                                 count = count + 1
    
16.                 return count
    
17.         def getDirList(self, url):
    
18.                 files = os.listdir(url)
    
19.                 fileNum = self.getCount(url)
    
20.                 tmpNum = 0
    
21.                 for file in files:
    
22.                         myfile = url + "\\" + file
    
23.                         size = os.path.getsize(myfile)
    
24.                         if os.path.isfile(myfile):
    
25.                                 tmpNum = tmpNum +1
    
26.                                 if (tmpNum != fileNum):
    
27.                                         self.list.append(str(self.SPACE) + "├─" + file + "\n")
    
28.                                 else:
    
29.                                         self.list.append(str(self.SPACE) + "└─" + file + "\n")
    
30.                         if os.path.isdir(myfile):
    
31.                                 self.list.append(str(self.SPACE) + "├─" + file + "\n")
    
32.                                 # change into sub directory
    
33.                                 self.SPACE = self.SPACE + "│  "
    
34.                                 self.getDirList(myfile)
    
35.                             # if iterator of sub directory is finished, reduce "│  "
    
36.                                 self.SPACE = self.SPACE[:-4]
    
37.                 return self.list
    
38.         def writeList(self, url):
    
39.                 f = open(url, 'w')
    
40.                 f.writelines(self.list)
    
41.                 print "ok"
    
42.                 f.close()
    
43. if __name__ == '__main__':
    
44.         d = dir()
    
45.         d.getDirList("c:/windows") # input directory
    
46.         d.writeList("c:/1.txt") # write to file

復(fù)制代碼

產(chǎn)生文件的效果圖：

anonymous0502

你取到的文件列表都放到變量里保存了，取列表變量的最后一個(gè)元素不就得到你要遍歷的列表的最后一個(gè)文件名了？（同一目錄下沒有文件會(huì)重名�。�
是這樣吧

只愛睡覺

敢不敢tree?