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

import glob
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 favicon
        shutil.copytree(
            f"{source_dir}/.neuron-generator/templates/favicon",
            f"{target_dir}/static",
            dirs_exist_ok=True,
        )

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

        # Copy favicon
        [
            shutil.copy2(f, f"{target_dir}/static")
            for f in glob.glob(
                f"{source_dir}/.neuron-generator/templates/favicon/favicon*"
            )
        ]

        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"
            )
        )