ansible_linux_desktop_setup/inventory.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Create a dynamic inventory for this ansible playbook
"""
import socket
import sys
import json

# create a dict to match hostnames to enviroments
env_dict = {
  'work':
    ['workstation.local'],
  'private':
    ['derpy.local', 'applejack.local']
}

def fqdn():
    """
    return fully qualified domain name
    """
    return socket.getfqdn()

def env(domain):
    """
    map a hostname to a space
    or print empty list if no one matched and exit
    """
    for key, values in env_dict.items():
        if domain in values:
            return key
    print(json.dumps(empty_host_list(domain), sort_keys=True, indent=2))
    sys.exit()

def empty_host_list(domain):
    """
    return empty host list
    """
    comment = f"No valid host found. Found '{domain}'. Return empty host list!"
    return json.loads('{"_meta": {"comment": "' + comment +
        '", "hostvars": {}}, "instances": {"hosts": []}}')

def formated_host_group_list(host, group):
    """
    build inventory and return it
    """
    # pylint: disable=line-too-long
    return json.loads('{"_meta": {"hostvars": {}},"' + str(group) + '": {"hosts": ["' + str(host) + '"]},"instances": {"children": ["' + str(group) + '"]}}')

def main():
    """
    main funktion
      will analyse on which host this script is started
      and will print the dynamic inventory to tell ansible
      which host_vars and group_vars should be used
    """
    host = fqdn()
    group = env(host)
    print(json.dumps(formated_host_group_list(host, group), sort_keys=True, indent=2))



#{
#    "_meta": {
#        "hostvars": { }
#    },
#
#    "instances": {
#        "hosts": ["10.66.70.33"]
#    }
# }

main()