import os
import shutil


def remove_chache_folders(current_repo: str = ""):
    """
    when executing a python program, most os will create several __pycache__
    folders. These folders are not needed after each run and can take some
    memory. This function removes these folders from the disk.
    """
    if current_repo == "":
        new_refs = [elem for elem in os.listdir()]
    else:
        new_refs = [current_repo + "/" + elem for elem in os.listdir(current_repo)]
    for elem in new_refs:
        if os.path.isdir(elem):
            if "__pycache__" in elem:
                shutil.rmtree(elem)
            else:
                remove_chache_folders(current_repo=elem)