import shutil
from termcolor import colored


def transfer_files(target_dir, source_dir):
    try:
        # Copy images to /static
        print(colored(" Copying static files...", "blue"))
        shutil.copytree(f"{source_dir}/img", f"{target_dir}/static")
        print(colored(" Static files transferred!", "green"))

        print(colored(" Copying zettels...", "blue"))
        # Copy notes
        shutil.copytree(f"{source_dir}/zk", f"{target_dir}", dirs_exist_ok=True)
        print(colored(" Zettels transferred!", "green"))
    except Exception as e:
        print(colored(f"Error occurred when transferring files: {str(e)}", "red"))