clean_pycache.py

'''
This script is used to clean the __pycache__ folders in the current directory.
'''

import os
import shutil


if __name__ == '__main__':
    # Define the directory to start searching from
    dir_list = []
    for subdir in os.listdir():
        if not '.' in subdir and not 'Data' in subdir:
            dir_list.append(subdir)


    # Walk through all directories and files in the repository
    for dir2check in dir_list:
        for dirpath, dirnames, filenames in os.walk(dir2check):
            # If a __pycache__ directory is found, remove it
            if '__pycache__' in dirnames:
                cache_path = os.path.join(dirpath, '__pycache__')
                shutil.rmtree(cache_path)
                print(f'Removed {cache_path}')


    # Clear the __pycache__ in the current directory
    if '__pycache__' in os.listdir():
        shutil.rmtree('__pycache__')
        print(f'Removed __pycache__ in the current directory')