diff --git a/src/app.py b/src/app.py index eb0619e..2437e2d 100644 --- a/src/app.py +++ b/src/app.py @@ -1,11 +1,12 @@ from constants import SOURCE from constants import TARGET -from lib.create_source_dir import create_source_dir +from lib.create_target_dir import create_target_dir +from lib.transfer_files import transfer_files def main(): - output_dir = create_source_dir() - print(output_dir) + target_dir = create_target_dir(TARGET, SOURCE) + transfer_files(f"{TARGET}/{target_dir}", SOURCE) if __name__ == "__main__": diff --git a/src/lib/create_source_dir.py b/src/lib/create_source_dir.py deleted file mode 100644 index 4b51a01..0000000 --- a/src/lib/create_source_dir.py +++ /dev/null @@ -1,21 +0,0 @@ -import os -import shutil -import uuid -from termcolor import colored -from constants import SOURCE, TARGET -from pathlib import Path - - -def create_source_dir(): - unique_dir_name = uuid.uuid4() - if os.path.isdir(TARGET): - shutil.rmtree(TARGET) - - # Create route directory and /static directory - os.makedirs(f"{TARGET}/{str(unique_dir_name)}/static") - print( - colored( - f"Created new Neuron output directory: {SOURCE}/{unique_dir_name}", "green" - ) - ) - return unique_dir_name diff --git a/src/lib/create_target_dir.py b/src/lib/create_target_dir.py new file mode 100644 index 0000000..a114f7c --- /dev/null +++ b/src/lib/create_target_dir.py @@ -0,0 +1,28 @@ +import os +import shutil +import uuid +from termcolor import colored + + +def create_target_dir(target_dir, source_dir): + try: + unique_dir_name = uuid.uuid4() + + # Overwrite existing output directory + if os.path.isdir(target_dir): + shutil.rmtree(target_dir) + + # Create new output directory + os.makedirs(f"{target_dir}/{str(unique_dir_name)}") + print( + colored( + f" Created new Neuron output directory: {source_dir}/{unique_dir_name}", + "green", + ) + ) + return unique_dir_name + + except Exception as e: + print( + colored(f" Error occurred when creating target directory: {str(e)}", "red") + ) diff --git a/src/lib/transfer_files.py b/src/lib/transfer_files.py new file mode 100644 index 0000000..b4008eb --- /dev/null +++ b/src/lib/transfer_files.py @@ -0,0 +1,17 @@ +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"))