用Python生成目录树

http://www.iteye.com/topic/494257

 

 

# 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

产生文件的效果图：

• 
• 大小: 43.7 KB