delete_libraries.py - delete node_moduleses and venvs

delete_libraries.py

import os
import sys
import argparse
import shutil
import traceback


def process(path, interactive=False):
    if not interactive:
        print(path)
        return
    if input('Delete {}? [y/N] '.format(path)) == 'y':
        def handle_error(func, pth, excinfo):
            print('{}: error removing {}: {}'.format(
                path,
                pth,
                excinfo,
            ))
        shutil.rmtree(path, ignore_errors=False, onerror=handle_error)


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument('roots', nargs='+', metavar='root')
    ap.add_argument('-i', '--interactive', action='store_true')
    args = ap.parse_args()
    
    for root in args.roots:
        for dirpath, dirnames, filenames in os.walk(root):
            new_dirnames = []
            for dirname in dirnames:
                if dirname in ('node_modules', 'venv'):
                    nm_path = os.path.join(dirpath, dirname)
                    process(nm_path, interactive=args.interactive)
                    continue
                new_dirnames.append(dirname)
            dirnames[:] = new_dirnames


if __name__ == '__main__':
    main()