How to Generate a cloud-init User Data Mock in Python

import json
from dataclasses import dataclass, asdict

@dataclass
class VMConfig:
    hostname: str
    cpus: int
    memory_mb: int
    ssh_key: str

def generate_cloud_init_mock(config: VMConfig) -> str:
    """Build a cloud-init user-data mock for a VM."""
    user_data = {
        "hostname": config.hostname,
        "users": [
            {
                "name": "admin",
                "sudo": "ALL=(ALL) NOPASSWD:ALL",
                "ssh_authorized_keys": [config.ssh_key],
            }
        ],
        "package_update": True,
        "packages": ["curl", "vim"],
    }
    user_data_str = f"#cloud-config\n{json.dumps(asdict(config))}\n" + json.dumps(user_data, indent=2)
    # Simulate provisioning by printing the mock payload
    print(user_data_str)
    return user_data_str

if __name__ == "__main__":
    demo = VMConfig(
        hostname="web-01",
        cpus=2,
        memory_mb=2048,
        ssh_key="ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQ... mock-key"
    )
    generate_cloud_init_mock(demo)