python生成目录树(os模块用法附效果图)

#!/usr/bin/env python
#coding=utf-8
# www.jb200.com

import os
import sys
 
class dir(object): 
def __init__(self): 
self.SPACE = "" 
self.list = []
self.size = 0
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 
try:
self.size += os.path.getsize(myfile) 
except Exception, e:
pass
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) + " i--" + 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(sys.argv[1]) # input directory
#d.writeList("1.txt") # write to file
#print '%s' % (''.join(d.list))
print ''.join(d.list)
print '%s total size : %d bytes' % (sys.argv[1],  d.size)