Function main

scripts_python/create_sandbox.py:311–387

()

def main():
    parser = argparse.ArgumentParser(description="Create a sandbox environment from a task")
    parser.add_argument("--task", help="Name of the task to create sandbox for")
    parser.add_argument(
        "--db", required=True, help="Database type (duckdb, sqlite, postgres, snowflake)"
    )
    parser.add_argument("--project-type", required=True, help="Project type (dbt, other)")

    args = parser.parse_args()

    task_name = args.task
    db_type = args.db
    project_type = args.project_type

    print(f"Creating sandbox for task: {task_name}")
    print(f"Database type: {db_type}")
    print(f"Project type: {project_type}")
    print("-" * 50)

    # Step 1: Check if task exists
    if not check_task_exists(task_name):
        sys.exit(1)

    print(f"✓ Task '{task_name}' found")

    # Step 2: Find matching variant
    variant = find_matching_variant(task_name, db_type, project_type)
    if not variant:
        sys.exit(1)

    print(
        f"✓ Found matching variant: db_name='{variant['db_name']}', project_name='{variant['project_name']}'"
    )
    if "migration_directory" in variant:
        print(f"✓ Migration directory: {variant['migration_directory']}")

    # Step 3: Wipe sandbox directory
    if not wipe_sandbox_directory():
        sys.exit(1)

    # Step 4: Copy shared project contents
    if not copy_project_contents(task_name, variant):
        sys.exit(1)

    # Step 5: Copy database file
    if not copy_database_file(task_name, variant):
        sys.exit(1)

    # Step 6: Copy migration files (if specified)
    if not copy_migration_files(variant):
        sys.exit(1)

    # Step 7: Copy task files (scripts, seeds, tests)
    if not copy_task_files(task_name):
        sys.exit(1)

    # Step 8: Copy shared scripts
    if not copy_shared_scripts():