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)