用Python实现Linux tree命令

#!/usr/bin/python
import os, sys, string

class XXXTree:
    def __init__(self):
        pass

    def printHelp(self, cmd):
        print "Please use the follow cmd"
        print " " + cmd + 'dir'
        print "e.g"
        print " " + cmd + "/root/Kedacom_src"

    def getTree(self, dir, op):
        list = self.getList(dir, 0, op)
        treeList = []
        #遍历文件列表，进行格式化输出处理，将处理后的元素添加到treeList中
        for i in range(1, len(list)):
            fullpath = list[i]
            filename = os.path.basename(fullpath)

            if(fullpath == dir):
                treeList.append(fullpath)
                continue

            path = fullpath.replace(dir, "")
            names = path.split('/')
            name = "|-- " + names[len(names)-1]
            for tmp in range(1, len(names)-1):
                name = " " + name 

            treeList.append(name)

            pos = name.find('&')
            if (pos >= 0):
                j = i-1
                if j >= 0:
                    name = treeList[j]
                    if (name[pos] == '&'):
                        tmp = name.index('|')
                        name = name[0:tmp] + "`" + name[tmp+1:pos]
                        treeList[j] = name

        for i in range(0, len(treeList)):
            print treeList[i]

    #递归获取目录下所有的文件，除去隐形文件
    def getList(self, dir, layer, op):
        list = [ ]
        if (layer == 0):list.append(dir)
        #对目录下的文件列表进行排序
        files = sorted(os.listdir(dir))
        for file in files:
            if (file[0] == '.'):
                continue
            file = os.path.join(dir, file)
            if os.path.isdir(file):
                list.append(file)
                list += self.getList(file, layer+1, op)
            elif (op == '-d'):
                pass
            else:
                list.append(file)
        newfile = file + '&'
        pos = list.index(file)
        list[pos] = newfile
        return list

if (len(sys.argv) < 2):
    t = XXXTree()
    t.printHelp(sys.argv[0])

else:
    t = XXXTree()

    dir = None
    if (len(sys.argv) == 2):dir = sys.argv[1]

    op = None
    if (len(sys.argv) == 3):
        op = sys.argv[1]
        dir = sys.argv[2]

    t.getTree(dir, op)