summaryrefslogtreecommitdiffstats
path: root/qtip/ansible_library/modules/fuel.py
blob: 04154e345bf931d4e18498967573ee21607e7f0f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#!/usr/bin/python

###############################################################
# Copyright (c) 2017 ZTE Corporation
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

from collections import defaultdict
import json

from ansible.module_utils.basic import AnsibleModule


ANSIBLE_METADATA = {'metadata_version': '1.0',
                    'status': ['preview'],
                    'supported_by': 'community'}

DOCUMENTATION = '''
---
module: fuel
short_description: collecting facts from fuel environments
description:
    - Use this module to create a dynamic inventory from fuel master.
version_added: "2.2"
author: "Yujun Zhang (@yujunz)"
options:
notes:
requirements:
    - Host 'fuel-master' is in ~/.ssh/config
'''

RETURN = '''
ansible_facts:
  description: facts collected for ansible
  returned: success
  type: dictionary
  contains:
    hosts:
      description: host grouped by hostname, cluster, role and manufacture
      type: dict
    hosts_meta:
      description: hosts meta data indexed by hostname
      type: dict
'''

EXAMPLES = '''
---
- hosts: fuel-master
  tasks:
  - name: collect facts of fuel hosts
    fuel:
  - debug: var=hostvars
  - name: add compute node to ansible inventory
    add_host:
      name: "{{ hosts_meta[item]['ip'] }}"
      groups: fuel-compute
      ansible_user: root
      ansible_ssh_common_args: '-o StrictHostKeyChecking=No -o ProxyJump=fuel-master'
    with_items: "{{ hosts.compute }}"
- hosts: fuel-compute
  tasks:
  - name: check ssh connection
    ping:
'''


def generate_inventory(nodes):
    """Generate ansible inventory from node list in json format

    Modified from https://github.com/martineg/ansible-fuel-inventory/blob/master/fuel.py
    """
    hosts = defaultdict(list)
    hosts_meta = {}

    for node in nodes:
        # skip deleting, offline, deploying and discovering/unprovisioned nodes
        if node['pending_deletion'] or (not node['online']) \
                or node['status'] == 'deploying' or node['status'] == 'discover':
            continue

        hostname = node['hostname']
        cluster_id = node['cluster']
        hw_vendor = node['meta']['system']['manufacturer'].lower()

        [hosts[role.strip()].append(hostname) for role in node['roles'].split(",")]
        hosts["cluster-{0}".format(cluster_id)].append(hostname)
        hosts["hw-{0}-servers".format(hw_vendor)].append(hostname)

        node_meta = {
            'name': node['name'],
            'online': node['online'],
            'os_platform': node['os_platform'],
            'status': node['status'],
            'ip': node['ip'],
            'mac': node['mac'],
            'cluster': cluster_id,
            'ansible_ssh_host': node['ip'],
            'ansible_user': 'root'
        }
        hosts_meta[hostname] = node_meta

    return {'hosts': hosts, 'hosts_meta': hosts_meta}


def main():
    module = AnsibleModule(argument_spec=dict())

    cmd = [module.get_bin_path('fuel', True), 'node', '--json']
    (rc, out, err) = module.run_command(cmd)

    if rc is not None and rc != 0:
        return module.fail_json(msg=err)

    nodes = json.loads(out)

    module.exit_json(changed=False, ansible_facts=generate_inventory(nodes))


if __name__ == '__main__':
    main()