cmd/prune_dirs.py

45 lines
1.1 KiB
Python
Raw Normal View History

2019-12-14 22:08:19 +00:00
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:]))