File size: 806 Bytes
54c22e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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)
|