From 214a244e9d1b54515c48fb72dca56e46b81b5024 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <git@voussoir.net>
Date: Sat, 14 Dec 2019 14:08:19 -0800
Subject: [PATCH] Add prune_dirs.py.

---
 prune_dirs.py | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 prune_dirs.py

diff --git a/prune_dirs.py b/prune_dirs.py
new file mode 100644
index 0000000..4d19cec
--- /dev/null
+++ b/prune_dirs.py
@@ -0,0 +1,44 @@
+import argparse
+import os
+import sys
+
+from voussoirkit import spinal
+from voussoirkit import pathclass
+
+
+def prune_dirs(starting):
+    starting = pathclass.Path(starting)
+    walker = spinal.walk_generator(starting, yield_directories=True, yield_files=False)
+
+    double_check = set()
+
+    def pruneme(directory):
+        if directory == starting or directory not in starting:
+            return
+        if len(directory.listdir()) == 0:
+            print(directory.absolute_path)
+            os.rmdir(directory.absolute_path)
+            double_check.add(directory.parent)
+
+    for directory in walker:
+        pruneme(directory)
+
+    while double_check:
+        directory = double_check.pop()
+        pruneme(directory)
+
+
+def prune_dirs_argparse(args):
+    return prune_dirs(args.starting)
+
+def main(argv):
+    parser = argparse.ArgumentParser(description=__doc__)
+
+    parser.add_argument('starting')
+    parser.set_defaults(func=prune_dirs_argparse)
+
+    args = parser.parse_args(argv)
+    return args.func(args)
+
+if __name__ == '__main__':
+    raise SystemExit(main(sys.argv[1:]))