1 ##############################################################################
2 # Copyright (c) 2016 Dan Radez (dradez@redhat.com) and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
15 from apex.common import constants
16 from apex.common import utils
19 class Inventory(dict):
21 This class parses an APEX inventory yaml file into an object. It
22 generates or detects all missing fields for deployment.
24 It then collapses one level of identification from the object to
25 convert it to a structure that can be dumped into a json file formatted
26 such that Triple-O can read the resulting json as an instackenv.json file.
28 def __init__(self, source, ha=True, virtual=False):
30 self.root_device = constants.DEFAULT_ROOT_DEV
31 if isinstance(source, str):
32 with open(source, 'r') as inventory_file:
33 yaml_dict = yaml.safe_load(inventory_file)
34 # collapse node identifiers from the structure
35 init_dict['nodes'] = list(map(lambda n: n[1],
36 yaml_dict['nodes'].items()))
38 # assume input is a dict to build from
44 pairs = (('pm_addr', 'ipmi_ip'), ('pm_password', 'ipmi_pass'),
45 ('pm_user', 'ipmi_user'), ('mac', 'mac_address'),
46 ('cpu', 'cpus'), (None, 'disk_device'))
50 if y == 'disk_device':
51 self.root_device = node[y]
54 elif x is not None and y in node:
58 # aarch64 is always uefi
59 if 'arch' in node and node['arch'] == 'aarch64':
60 node['capabilities'] += ',boot_mode:uefi'
64 super().__init__({'nodes': list(map(munge_node, init_dict['nodes']))})
66 # verify number of nodes
67 if ha and len(self['nodes']) < 5:
68 raise ApexInventoryException('You must provide at least 5 '
69 'nodes for HA deployment')
70 elif len(self['nodes']) < 2:
71 raise ApexInventoryException('You must provide at least 2 nodes '
72 'for non-HA deployment')
75 self['host-ip'] = '192.168.122.1'
76 self['power_manager'] = \
77 'nova.virt.baremetal.virtual_power_driver.VirtualPowerManager'
79 self['ssh-key'] = 'INSERT_STACK_USER_PRIV_KEY'
80 self['ssh-user'] = 'root'
82 def dump_instackenv_json(self):
83 print(json.dumps(dict(self), sort_keys=True, indent=4))
85 def get_node_counts(self):
87 Return numbers of controller and compute nodes in inventory
88 :param inventory: node inventory data structure
89 :return: number of controller and compute nodes in inventory
95 if 'profile:control' in node['capabilities']:
97 elif 'profile:compute' in node['capabilities']:
100 raise ApexInventoryException("Node missing capabilities "
101 "key: {}".format(node))
102 return num_control, num_compute
105 class ApexInventoryException(Exception):