Python生成目录树代码,Python代码库,德仔网

代码语言

代码分类

【Python】 Python生成目录树

作者:Dezai.CN / 发布于2013/7/30/ 1828

	# 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

试试其它关键字

　生成目录树　

同语言下

可能有用的

Dezai.CN贡献的其它代码(4037)