Linux用树形结构显示目录结构

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

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

运行效果如下：

$ ./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