neuron-zk-generator/src/lib/transfer_files.py

import shutil

from termcolor import colored


def transfer_files(target_dir, source_dir):
    try:
        # Copy templates
        print(colored("INFO Copying HTML/MD templates...", "light_blue"))
        shutil.copytree(
            f"{source_dir}/.neuron-generator/templates", target_dir, dirs_exist_ok=True
        )
        neuron_template = open(f"{target_dir}/neuron.dhall", "x")
        neuron_template.close()
        print(colored("SUCCESS Templates transferred", "light_green"))

        # Copy images to /static
        print(colored("INFO Copying static files...", "light_blue"))
        shutil.copytree(
            f"{source_dir}/img",
            f"{target_dir}/static",
        )
        print(colored("SUCCESS Static files transferred", "light_green"))

        print(colored("INFO Copying zettels...", "light_blue"))

        # Copy notes
        shutil.copytree(f"{source_dir}/zk", f"{target_dir}", dirs_exist_ok=True)
        print(colored("SUCCESS Zettels transferred", "light_green"))
    except Exception as e:
        print(
            colored(
                f"ERROR Error occurred when transferring files: {str(e)}", "light_red"
            )
        )