How to Mock an Ansible Inventory in Python

import json
from pathlib import Path


class InventoryMock:
    def __init__(self, inventory_file: str):
        self.inventory_file = Path(inventory_file)
        self.hosts = {}

    def load(self):
        if not self.inventory_file.exists():
            raise FileNotFoundError(f"Inventory file {self.inventory_file} not found")
        with open(self.inventory_file) as f:
            data = json.load(f)
        for group, details in data.get("all", {}).get("children", {}).items():
            for host in details.get("hosts", {}):
                self.hosts[host] = {
                    "group": group,
                    **details["hosts"][host].get("vars", {})
                }
        return self

    def run_playbook(self, playbook: str):
        print(f"Running playbook: {playbook}")
        for host, vars_data in self.hosts.items():
            print(f"  -> {host}: {vars_data}")


if __name__ == "__main__":
    inventory = {
        "all": {
            "children": {
                "web": {
                    "hosts": {
                        "web1": {"vars": {"port": 8080, "role": "nginx"}},
                        "web2": {"vars": {"port": 8081, "role": "nginx"}}
                    }
                },
                "db": {
                    "hosts": {
                        "db1": {"vars": {"port": 5432, "role": "postgres"}}
                    }
                }
            }
        }
    }
    with open("inventory.json", "w") as f:
        json.dump(inventory, f)

    mock = InventoryMock("inventory.json")
    mock.load()
    mock.run_playbook("setup.yml")