382
|
1 #!/usr/bin/env python
|
|
2 # -*- coding: utf-8 -*-
|
|
3
|
|
4 """
|
|
5 tree in python
|
|
6 """
|
|
7
|
|
8 import optparse
|
|
9 import os
|
|
10 import sys
|
|
11
|
|
12 here = os.path.dirname(os.path.realpath(__file__))
|
|
13
|
|
14 def depth(directory):
|
|
15 directory = os.path.abspath(directory)
|
|
16 level = 0
|
|
17 while True:
|
|
18 directory, remainder = os.path.split(directory)
|
|
19 level += 1
|
|
20 if not remainder:
|
|
21 break
|
|
22 return level
|
|
23
|
|
24 def tree(directory):
|
|
25 retval = []
|
|
26 level = depth(directory)
|
|
27 pre = []
|
|
28 directories = {}
|
|
29 lvlndctr = []
|
|
30 last = {}
|
|
31 passed_last = {}
|
|
32 columns = []
|
|
33 lastdepth = depth
|
|
34 indent = 0
|
|
35 for dirpath, dirnames, filenames in os.walk(directory, topdown=True):
|
|
36 basename = os.path.basename(dirpath)
|
|
37 parent = os.path.abspath(os.path.dirname(dirpath))
|
|
38 indent = depth(dirpath) - level
|
|
39 import pdb; pdb.set_trace()
|
|
40 dirnames[:] = sorted(dirnames, key=lambda x: x.lower())
|
|
41 last[os.path.abspath(dirpath)] = dirnames and dirnames[-1] or None
|
|
42 directories[dirpath] = dirnames
|
|
43
|
|
44 retval.append('%s%s%s %s' % ('│' * (indent-1),
|
|
45 ('├' if basename == basename else '└') if indent else '',
|
|
46 basename))
|
|
47 filenames = sorted(filenames, key=lambda x: x.lower())
|
|
48 retval.extend(['%s%s%s' % ('│' * (indent),
|
|
49 '├' if (((index < len(filenames) -1)) or dirnames) else '└',
|
|
50 name)
|
|
51 for index, name in
|
|
52 enumerate(filenames)
|
|
53 ])
|
|
54 return '\n'.join(retval)
|
|
55
|
|
56 def main(args=sys.argv[1:]):
|
|
57
|
|
58 usage = '%prog [options]'
|
|
59 parser = optparse.OptionParser(usage=usage, description=__doc__)
|
|
60 options, args = parser.parse_args(args)
|
|
61 if not args:
|
|
62 args = ['.']
|
|
63
|
|
64 not_directory = [arg for arg in args
|
|
65 if not os.path.isdir(arg)]
|
|
66 if not_directory:
|
|
67 parser.error("Not a directory: %s" % (', '.join(not_directory)))
|
|
68
|
|
69 for arg in args:
|
|
70 print (tree(arg))
|
|
71
|
|
72 if __name__ == '__main__':
|
|
73 main()
|