1 #!/usr/bin/env python
  2 
  3 import os
  4 import os.path
  5 import re
  6 import sys
  7 
  8 re_sp = re.compile("/")
  9 max_level = 0x7FFFFFFF
 10 
 11 
 12 def get_dirname(path) :
 13     """
 14     get the dirname by a path
 15     """
 16     global re_sp
 17 
 18     path = os.path.realpath(path)
 19     lst_ret = re_sp.split(path)
 20     if 0 == len(lst_ret) :
 21         return ""
 22     return lst_ret[len(lst_ret)-1]
 23 
 24 
 25 def get_startmark(path, lvl) :
 26     """
 27     get the startmark which will be displayed first
 28     """
 29     global max_level
 30 
 31     new_str = ""
 32     for i in range(lvl) :
 33         new_str += '|'
 34 
 35     if True == os.path.isdir(path) :
 36         if len(os.listdir(path)) > 0 and lvl == max_level:
 37             new_str += "+"
 38         else :
 39             new_str += "-"
 40         for i in range(lvl) :
 41             new_str += ' '
 42     else :
 43         new_str += '|'
 44         for i in range(lvl) :
 45             new_str += ' '
 46     
 47     return new_str
 48 
 49 
 50 def dfs_dir(path, lvl) :
 51     """
 52     deep first search the path
 53     """
 54     global max_level
 55 
 56     new_str = get_startmark(path, lvl)
 57     print "%s%s" % (new_str, get_dirname(path)),
 58 
 59     #check file type
 60     if True == os.path.islink(path) :
 61         print "%s" % "[L]",
 62     if True == os.path.ismount(path) :
 63         print "%s" % "[M]",
 64     if False == os.path.isdir(path) :
 65         print "[%d]" % os.path.getsize(path)    
 66     else :
 67         print ""
 68 
 69     if False == os.path.isdir(path) :
 70         return
 71 
 72     if lvl == max_level :
 73         return
 74 
 75     files = os.listdir(path)
 76     for each_file in files :
 77         new_path = "%s/%s" % (path, each_file)
 78         dfs_dir(new_path, lvl+1)
 79 
 80 
 81 def pytree() :
 82     """
 83     the main function to run
 84     """
 85     global max_level
 86 
 87     argvs = sys.argv
 88     
 89     if 0 == len(argvs) % 2 :
 90         print "Argvs invalid!"
 91         sys.exit(-1)
 92     else :
 93         cur_path = ""
 94 
 95         for tmp_str in argvs :
 96             if tmp_str.startswith("-") :
 97                 if "-i" == tmp_str :
 98                     idx = argvs.index(tmp_str)
 99                     max_level = int(argvs[idx+1])
100                 if "-p" == tmp_str:
101                     idx = argvs.index(tmp_str)
102                     cur_path = os.path.realpath(argvs[idx+1])
103 
104         if "" == cur_path :
105             cur_path = os.getcwd()
106 
107         if False == os.path.isdir(cur_path) :
108             print "Argvs invalid : path not exits!"
109         else :
110             print "List Dir : %s" % cur_path
111             dfs_dir(cur_path, 0)
112 
113     
114 
115 
116 if __name__ == "__main__" :
117     pytree()
118 

豪 2009-03-29 19:56 发表评论