学步园

在有些Linux上自带了tree命令来一树状结构显示一个目录，但是在有些linux上没有自带这个程序，所以这里用python写了一个小程序来实现这个功能，代码如下：

#!/usr/bin/python<br /> import os, sys, string<br /> class XXTree:<br /> def __init__(self):<br /> pass</p> <p> def printHelp(self, cmd):<br /> print 'Please use the following cmd:'<br /> print ' ' + cmd + ' dir'<br /> print 'e.g.'<br /> print ' ' + cmd + ' /home/fkong/tmp'<br /> def getTree(self, dir):<br /> list = self.getList(dir, 0)<br /> treelist = []</p> <p> for i in range(0, len(list)):<br /> fullpath = list[i]<br /> parpath = os.path.dirname(list[i])<br /> filename = os.path.basename(list[i])<br /> if(fullpath == dir):<br /> treelist.append(fullpath)<br /> continue</p> <p> path = fullpath.replace(dir, "")<br /> names = path.split("/")<br /> name = "`---" + names[len(names) - 1]<br /> for j in range(1, len(names) - 1):<br /> name = " " + name<br /> treelist.append(name)</p> <p> pos = name.index("`")<br /> j = i - 1<br /> while j > 0:<br /> name = treelist[j]<br /> if(name[pos] == '`' or name[pos] == ' '):<br /> name = name[0: pos] + "|" + name[pos + 1: len(name)]<br /> treelist[j] = name<br /> else:<br /> break<br /> j = j - 1</p> <p> for i in range(0, len(treelist)):<br /> print treelist[i]</p> <p> def getList(self, dir, layer):<br /> list = []<br /> if layer == 0: list.append(dir)<br /> files = os.listdir(dir)<br /> for file in files:<br /> file = os.path.join(dir, file)<br /> if os.path.isdir(file):<br /> list.append(file)<br /> list += self.getList(file, layer + 1)<br /> else :<br /> list.append(file)<br /> return list<br /> if len(sys.argv) != 2:<br /> t = XXTree()<br /> t.printHelp(sys.argv[0])<br /> else:<br /> t = XXTree()<br /> dir = sys.argv[1]<br /> t.getTree(dir)<br />  

运行效果如下：

 

$ ./xxtree.py  /home/fkong/workspace/jutility/.svn

/home/fkong/workspace/jutility/.svn

|---format

|---props

|---entries

|---prop-base

|---text-base

|---tmp

|   |---prop-base

|   |---props

|   `---text-base

`---all-wcprops