diff python/tree.py @ 374:6e0853b16457

initial tree prog
author Jeff Hammel <jhammel@mozilla.com>
date Wed, 24 Jul 2013 02:38:27 -0700
parents
children 9314c1008189
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/python/tree.py	Wed Jul 24 02:38:27 2013 -0700
@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+tree in python
+"""
+
+import optparse
+import os
+import sys
+
+here = os.path.dirname(os.path.realpath(__file__))
+
+def depth(directory):
+    directory = os.path.abspath(directory)
+    level = 0
+    while True:
+        directory, remainder = os.path.split(directory)
+        level += 1
+        if not remainder:
+            break
+    return level
+
+def tree(directory):
+    retval = []
+    level = depth(directory)
+    directories = {}
+    for dirpath, dirnames, filenames in os.walk(directory, topdown=True):
+        indent = depth(dirpath) - level
+        dirnames[:] = sorted(dirnames, key=lambda x: x.lower())
+        directories[dirpath] = dirnames
+        retval.append('%s%s%s' % ('│' * (indent-1),
+                                  '├' if indent else '',
+                                  os.path.basename(dirpath)))
+        retval.extend(['%s%s%s' % ('│' * (indent),
+                                   '├' if index < len(filenames) -1 else '└',
+                                    name)
+                       for index, name in
+                       enumerate(sorted(filenames, key=lambda x: x.lower()))
+                       ])
+    return '\n'.join(retval)
+
+def main(args=sys.argv[1:]):
+
+    usage = '%prog [options]'
+    parser = optparse.OptionParser(usage=usage, description=__doc__)
+    options, args = parser.parse_args(args)
+    if not args:
+        args = ['.']
+
+    not_directory = [arg for arg in args
+                     if not os.path.isdir(arg)]
+    if not_directory:
+        parser.error("Not a directory: %s" % (', '.join(not_directory)))
+
+    for arg in args:
+        print (tree(arg))
+
+if __name__ == '__main__':
+    main()