這是一個使用Python生成文件、目錄樹的程序,其中遇到一個問題是:如何確定某個目錄下的某一文件是最后一個遍歷的文件。因為最后一個遍歷的文件前應添加"└─",非最后一個文件前添加"├─"。看了Python的API文檔沒有找到相關的系統函數。現在做法是:先統計出某個目錄下的文件個數,在遍歷目錄時,當個數相等時,就可以確定該目錄遍歷結束。
# encoding: utf-8 import os class dir(object): def __init__(self): self.SPACE = "" self.list = [] def getCount(self, url): files = os.listdir(url) count = 0; for file in files: myfile = url + "//" + file if os.path.isfile(myfile): count = count + 1 return count def getDirList(self, url): files = os.listdir(url) fileNum = self.getCount(url) tmpNum = 0 for file in files: myfile = url + "//" + file size = os.path.getsize(myfile) if os.path.isfile(myfile): tmpNum = tmpNum +1 if (tmpNum != fileNum): self.list.append(str(self.SPACE) + "├─" + file + "/n") else: self.list.append(str(self.SPACE) + "└─" + file + "/n") if os.path.isdir(myfile): self.list.append(str(self.SPACE) + "├─" + file + "/n") # change into sub directory self.SPACE = self.SPACE + "│ " self.getDirList(myfile) # if iterator of sub directory is finished, reduce "│ " self.SPACE = self.SPACE[:-4] 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.getDirList("c:/windows") # input directory d.writeList("c:/1.txt") # write to file 以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持VEVB武林網。
新聞熱點
疑難解答