84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Copyright 2017 Red Hat, Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import yaml
|
|
|
|
from ansible.plugins.lookup import LookupBase
|
|
|
|
from tripleo_validations import utils
|
|
|
|
|
|
DOCUMENTATION = """
|
|
lookup: roles_info
|
|
description: Retrieve role information from Heat and Swift.
|
|
long_description:
|
|
- Load role information using the Heat API.
|
|
options:
|
|
_terms:
|
|
description: Optional filter attribute and filter value
|
|
author: Brad P. Crochet <brad@redhat.com>
|
|
"""
|
|
|
|
EXAMPLES = """
|
|
- name: Get all role info from Heat and Swift
|
|
debug:
|
|
msg: |
|
|
{{ lookup('roles_info', wantlist=True) }}
|
|
|
|
"""
|
|
|
|
RETURN = """
|
|
_raw:
|
|
description: A Python list with results from the API call.
|
|
"""
|
|
|
|
|
|
class LookupModule(LookupBase):
|
|
def _get_object_yaml(self, swiftclient, container, obj):
|
|
obj_ret = swiftclient.get_object(container=container, obj=obj)
|
|
return yaml.safe_load(obj_ret[1])
|
|
|
|
def run(self, terms, variables=None, **kwargs):
|
|
"""Returns server information from nova."""
|
|
swift = utils.get_swift_client(variables)
|
|
plan = variables.get('plan')
|
|
plan_env = self._get_object_yaml(swift, plan, 'plan-environment.yaml')
|
|
roles_data = self._get_object_yaml(swift, plan, 'roles_data.yaml')
|
|
|
|
def default_role_data(role):
|
|
return {
|
|
'name': role['name'],
|
|
'count': role.get('CountDefault', 0),
|
|
'flavor': role.get('FlavorDefault', 'baremetal')
|
|
}
|
|
|
|
roles = list(map(default_role_data, roles_data))
|
|
|
|
parameter_defaults = plan_env.get('parameter_defaults', {})
|
|
|
|
for role in roles:
|
|
new_count = parameter_defaults.get("%sCount" % role['name'])
|
|
if new_count:
|
|
role['count'] = new_count
|
|
|
|
new_flavor = parameter_defaults.get("Overcloud%sFlavor" %
|
|
role['name'])
|
|
if new_flavor:
|
|
role['flavor'] = new_flavor
|
|
|
|
return roles
|