kunlun/clear_output.py

import os
import shutil




def clear(dir_path:str,rm_dir:str):
    dir_list=os.listdir(dir_path)
    for item in dir_list:
        path=os.path.join(dir_path,item)
        if os.path.isdir(path):
            if item == rm_dir:
                print('remove: ',path)
                shutil.rmtree(path)
            else:
                clear(path,rm_dir)
        elif os.path.isfile(path):
            if item == rm_dir:
                print('remove: ',path)
                os.remove(path)





if __name__ == "__main__":
    clear('../','.output')
    clear('../','src_files.txt')