Inventory

from pyinfra import Inventory

# Simple inventory with host list
inventory = Inventory(
    (["web1.example.com", "web2.example.com"], {})
)

# Inventory with host data
inventory = Inventory(
    ([
        ("web1.example.com", {"ssh_user": "ubuntu"}),
        ("web2.example.com", {"ssh_user": "admin"}),
    ], {})
)

# Inventory with groups
inventory = Inventory(
    (["web1", "web2", "db1"], {}),
    webservers=(["web1", "web2"], {"app_port": 80}),
    databases=(["db1"], {"db_port": 5432}),
)

def __init__(
    self,
    names_data: tuple[list, dict],
    override_data: dict = None,
    **groups: dict[str, tuple[list, dict]]
):
    """Create an inventory.
    
    Args:
        names_data: Tuple of (host_names, global_data)
        override_data: Data that overrides all other data
        **groups: Named groups with (host_names, group_data) tuples
    """

inventory = Inventory((["host1", "host2", "host3"], {}))

# Iterate
for host in inventory:
    print(host.name)

# Get count
print(f"Total hosts: {len(inventory)}")

# Get active hosts (not failed)
active_hosts = inventory.get_active_hosts()
for host in active_hosts:
    print(f"Active: {host.name}")

print(f"Active count: {inventory.len_active_hosts()}")

# Get all activated hosts (includes failed)
for host in inventory.iter_activated_hosts():
    print(f"Activated: {host.name}")

print(f"Activated count: {inventory.len_activated_hosts()}")

from pyinfra.api.exceptions import NoHostError

# Get host by name
host = inventory.get_host("web1")

# Get with default value
host = inventory.get_host("missing", default=None)

# Raises NoHostError if not found
try:
    host = inventory.get_host("missing")
except NoHostError:
    print("Host not found")

from pyinfra.api.exceptions import NoGroupError

# Get list of hosts in group
web_hosts = inventory.get_group("webservers")
for host in web_hosts:
    print(host.name)

# Get with default
api_hosts = inventory.get_group("api", default=[])

# Raises NoGroupError if not found
try:
    hosts = inventory.get_group("missing")
except NoGroupError:
    print("Group not found")

# Get all global inventory data
global_data = inventory.get_data()
print(global_data.get("environment"))

# Get override data
overrides = inventory.get_override_data()

# Get data for specific host
host_data = inventory.get_host_data("web1")
print(host_data.get("ssh_user"))

# Get data for specific group
group_data = inventory.get_group_data("webservers")
print(group_data.get("app_port"))

# Get combined data from multiple groups
host = inventory.get_host("web1")
groups_data = inventory.get_groups_data(host.groups)
# Returns combined data from all groups the host belongs to

from pyinfra import Inventory

# Mix SSH and Docker hosts
inventory = Inventory((
    [
        "server1.example.com",  # SSH by default
        "@docker/ubuntu:22.04",  # Docker connector
        "@local",  # Local connector
    ],
    {}
))

from pyinfra import Inventory

inventory = Inventory(
    # All hosts with global data
    ([
        "web1.example.com",
        "web2.example.com",
        "db1.example.com",
        "cache1.example.com",
    ], {
        "ssh_user": "deploy",
        "environment": "production",
    }),
    
    # Webserver group
    webservers=([
        "web1.example.com",
        "web2.example.com",
    ], {
        "app_port": 8000,
        "workers": 4,
    }),
    
    # Database group
    databases=([
        "db1.example.com",
    ], {
        "db_port": 5432,
        "max_connections": 100,
    }),
    
    # Cache group
    cache=([
        "cache1.example.com",
    ], {
        "cache_size": "1g",
    }),
)

# Access groups
web_hosts = inventory.get_group("webservers")
for host in web_hosts:
    print(f"{host.name}: port={host.data.app_port}")

from pyinfra import Inventory

# Empty inventory
inventory = Inventory.empty()

from pyinfra import Inventory, Config, State
from pyinfra.operations import apt, files

# Create inventory with groups and data
inventory = Inventory(
    # All hosts
    ([
        ("web1.example.com", {"ssh_user": "ubuntu"}),
        ("web2.example.com", {"ssh_user": "ubuntu"}),
        ("db1.example.com", {"ssh_user": "admin"}),
    ], {
        # Global data
        "environment": "production",
        "ssh_port": 22,
    }),
    
    # Groups
    webservers=([
        "web1.example.com",
        "web2.example.com",
    ], {
        "app_port": 8000,
        "app_name": "myapp",
    }),
    
    databases=([
        "db1.example.com",
    ], {
        "db_port": 5432,
    }),
)

# Initialize state
config = Config(SUDO=True)
state = State(inventory, config)
state.init(inventory, config)

# Connect to all hosts
for host in inventory:
    host.connect()
    print(f"Connected to {host.name}")

# Deploy to webservers group
web_hosts = inventory.get_group("webservers")
for host in web_hosts:
    print(f"Deploying to {host.name}")
    print(f"  App: {host.data.app_name} on port {host.data.app_port}")
    print(f"  Environment: {host.data.environment}")

# Check inventory stats
print(f"\nInventory stats:")
print(f"  Total hosts: {len(inventory)}")
print(f"  Active hosts: {inventory.len_active_hosts()}")
print(f"  Groups: {list(inventory.groups.keys())}")

# Get host by name
web1 = inventory.get_host("web1.example.com")
print(f"\nHost {web1.name} is in groups: {web1.groups}")