pulumi-templates/pvm-ansible/__main__.py

"""ProxmoxVE provider, Ansible powered with Python"""

import pulumi
import pulumi_proxmoxve as proxmox
import pulumi_command as command
import yaml
from dotmap import DotMap

# Import configuration parameters from input.yaml
input_ = DotMap(yaml.safe_load(open("./input.yaml", "r").read()))

# Check pulumi configuration
config = pulumi.Config()

provider = proxmox.Provider("proxmoxve",
    endpoint=input_.pve.url,
    insecure=input_.pve.insecure,
    username=input_.pve.username,
    password=config.require_secret("password")
)

vm_name = input_.vm.name
vm_username = input_.vm.username
ansible_playbook = input_.ansible_playbook

vm = proxmox.vm.VirtualMachine("vm",
    name = vm_name,
    node_name = input_.pve.nodename,
    agent = proxmox.vm.VirtualMachineAgentArgs(
        enabled = True,
        trim = True,
        type = "virtio"
    ),
    bios = "seabios",
    cpu = proxmox.vm.VirtualMachineCpuArgs(
        cores = input_.vm.cores,
        sockets = input_.vm.sockets
    ),
    memory = proxmox.vm.VirtualMachineMemoryArgs(dedicated = input_.vm.ram),
    clone = proxmox.vm.VirtualMachineCloneArgs(
        node_name = input_.pve.nodename,
        vm_id = input_.vm.clone,
        full = True
    ),
    disks = [
        proxmox.vm.VirtualMachineDiskArgs(
            interface = "virtio0",
            datastore_id = input_.pve.storage,
            size = input_.vm.disksize,
            file_format = "raw"
        )
    ],
    network_devices = [
        proxmox.vm.VirtualMachineNetworkDeviceArgs(
            bridge = "vmbr0",
            model = "virtio"
        )
    ],
    on_boot = True,
    operating_system = proxmox.vm.VirtualMachineOperatingSystemArgs(
        type = "l26"
    ),
    initialization = proxmox.vm.VirtualMachineInitializationArgs(
        type = "nocloud",
        datastore_id = input_.pve.storage,
        user_account = proxmox.vm.VirtualMachineInitializationUserAccountArgs(
            username = vm_username,
            password = input_.vm.password,
            keys = input_.vm.sshkeys
        )
    ),
    opts = pulumi.ResourceOptions(
        provider = provider,
        custom_timeouts=pulumi.CustomTimeouts(create="3m")
    )
)

# First item of the ipv4_addresses is of the loopback interface (so the usual 127.0.0.1). Let's get the second and grab only the text.

# Creating the inventory file
inventory = command.local.Command("a-inventory",
    create = vm.ipv4_addresses[1][0].apply(
        lambda ipaddr: f"echo '{vm_name} ansible_host={ipaddr} ansible_user={vm_username}' >./inventory"
    ),
    delete = "rm -f ./inventory",
    opts = pulumi.ResourceOptions(depends_on = [ vm ])
)

# Try the deployment with ansible

# Applying the command
execute_ansible = command.local.Command("a-deploy",
    create = vm.ipv4_addresses[1][0].apply(
        lambda ipaddr: f"ansible-playbook {ansible_playbook}"
    ),
    delete = "rm -f ./ansible.log",
    opts = pulumi.ResourceOptions(depends_on = [ inventory ])
)

# Outputs
pulumi.export("ip", vm.ipv4_addresses[1][0])