Mercurial > hg > config

comparison python/tree2.py @ 382:9ac9561b6574

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
copy a thing
author Jeff Hammel <jhammel@mozilla.com>
date Fri, 26 Jul 2013 03:58:21 -0700
parents
children 8d1ad56761b0
comparison
equal deleted inserted replaced
381:397d0ac832b6 382:9ac9561b6574
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()