Add prune_dirs.py.
This commit is contained in:
parent
514af31c18
commit
214a244e9d
1 changed files with 44 additions and 0 deletions
44
prune_dirs.py
Normal file
44
prune_dirs.py
Normal file
|
@ -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:]))
|
Loading…
Reference in a new issue