| import os | |
| import shutil | |
| def file_check(original_dir, new_dir): | |
| if os.path.isdir(original_dir): | |
| for file in os.listdir(original_dir): | |
| shutil.move(os.path.join(original_dir, file), os.path.join(new_dir, file)) | |
| if len(os.listdir(original_dir)) == 0: | |
| shutil.rmtree(original_dir) | |
| def remove_unneeded_yamls(demucs_dir): | |
| for file in os.listdir(demucs_dir): | |
| if file.endswith('.yaml'): | |
| if os.path.isfile(os.path.join(demucs_dir, file)): | |
| os.remove(os.path.join(demucs_dir, file)) | |
| def remove_temps(remove_dir): | |
| if os.path.isdir(remove_dir): | |
| try: | |
| shutil.rmtree(remove_dir) | |
| except Exception as e: | |
| print(e) | |