Python:目录树遍历和生成

dirlist.py

#coding：utf-8
import sys
reload(sys)
sys.setdefaultencoding('utf8')

from os.path import basename, isdir
from os import listdir

def traverse(path, depth=0):
    print depth* '| ' + '|_', basename(path)
    if(isdir(path)):
        for item in listdir(path):
            traverse(path+'/'+item, depth+1)

if __name__ == '__main__':
    traverse('./')