build.py

import os
import shutil
import zipfile
import argparse

def merge_folders(folder1, folder2, output_folder):
    """Merge two folders into the output folder."""
    # Create the output folder if it doesn't exist
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    
    # Copy all contents of folder1 to the output folder
    for item in os.listdir(folder1):
        item_path = os.path.join(folder1, item)
        if os.path.isdir(item_path):
            shutil.copytree(item_path, os.path.join(output_folder, item))
        else:
            shutil.copy2(item_path, output_folder)
    
    # Copy all contents of folder2 to the output folder
    for item in os.listdir(folder2):
        item_path = os.path.join(folder2, item)
        if os.path.isdir(item_path):
            shutil.copytree(item_path, os.path.join(output_folder, item), dirs_exist_ok=True)
        else:
            shutil.copy2(item_path, output_folder)

def zip_folder(folder_path, zip_name):
    """Zip the content of a folder into a zip file."""
    with zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                zipf.write(os.path.join(root, file), os.path.relpath(os.path.join(root, file), folder_path))

def main():
    # Set up argument parsing
    parser = argparse.ArgumentParser(description="Merge two folders and create a zip file.")
    parser.add_argument('folder', help="Name of the non-common folder to merge.")
    parser.add_argument('version', help="Version number to append to the zip file name.")
    args = parser.parse_args()

    # Get the directory of the current script
    script_dir = os.path.dirname(os.path.realpath(__file__))

    # Define paths for the "common" folder and the folder provided as the non-common folder (inside the 'packs' folder)
    folder1 = os.path.join(script_dir, 'common')  # Assuming 'common' is in the same directory as the script
    folder2 = os.path.join(script_dir, 'packs', args.folder)  # Now looking for the folder inside the 'packs' folder

    # Ensure the 'common' folder exists
    if not os.path.exists(folder1):
        print(f"Error: 'common' folder not found in {script_dir}")
        return

    # Ensure the specified folder inside 'packs' exists
    if not os.path.exists(folder2):
        print(f"Error: '{args.folder}' folder not found in {os.path.join(script_dir, 'packs')}")
        return

    # Define the working directory and output folder
    working_dir = os.path.join(script_dir, 'working')
    output_dir = os.path.join(script_dir, 'output')

    # Ensure the working directory exists
    if not os.path.exists(working_dir):
        os.makedirs(working_dir)

    # Ensure the output directory exists
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    # Temporary folder for merging the contents, named after the non-common folder
    temp_folder = os.path.join(working_dir, args.folder)

    # Ensure the temporary folder exists
    if not os.path.exists(temp_folder):
        os.makedirs(temp_folder)

    try:
        # Merge the folders into the temporary directory
        merge_folders(folder1, folder2, temp_folder)

        # Define the zip file path, named after the non-common folder and version
        zip_name = os.path.join(output_dir, f'{args.folder}-{args.version}.zip')

        # Zip the merged content
        zip_folder(temp_folder, zip_name)
        print(f"Merged and zipped into {zip_name}")
    
    finally:
        # Clean up the temporary folder after zipping
        if os.path.exists(temp_folder):
            shutil.rmtree(temp_folder)

if __name__ == '__main__':
    main()