changeset 382:9ac9561b6574

copy a thing
author Jeff Hammel <jhammel@mozilla.com>
date Fri, 26 Jul 2013 03:58:21 -0700
parents 397d0ac832b6
children 8d1ad56761b0
files python/tree2.py
diffstat 1 files changed, 73 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/python/tree2.py	Fri Jul 26 03:58:21 2013 -0700
@@ -0,0 +1,73 @@
+#!/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)
+    pre = []
+    directories = {}
+    lvlndctr = []
+    last = {}
+    passed_last = {}
+    columns = []
+    lastdepth = depth
+    indent = 0
+    for dirpath, dirnames, filenames in os.walk(directory, topdown=True):
+        basename = os.path.basename(dirpath)
+        parent = os.path.abspath(os.path.dirname(dirpath))
+        indent = depth(dirpath) - level
+        import pdb; pdb.set_trace()
+        dirnames[:] = sorted(dirnames, key=lambda x: x.lower())
+        last[os.path.abspath(dirpath)] = dirnames and dirnames[-1] or None
+        directories[dirpath] = dirnames
+
+        retval.append('%s%s%s %s' % ('│' * (indent-1),
+                                     ('├' if basename == basename else '└') if indent else '',
+                                     basename))
+        filenames = sorted(filenames, key=lambda x: x.lower())
+        retval.extend(['%s%s%s' % ('│' * (indent),
+                                   '├' if (((index < len(filenames) -1)) or dirnames) else '└',
+                                    name)
+                       for index, name in
+                       enumerate(filenames)
+                       ])
+    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()