okd/filter_plugins/oo_resourceconfig.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
# vim: expandtab:tabstop=4:shiftwidth=4
'''
Filters for configuring resources in openshift-ansible
'''

from ansible import errors

def get_attr(data, attribute=None):
    ''' This looks up dictionary attributes of the form a.b.c and returns
        the value.
        Ex: data = {'a': {'b': {'c': 5}}}
            attribute = "a.b.c"
            returns 5
    '''
    if not attribute:
        raise errors.AnsibleFilterError("|failed expects attribute to be set")

    ptr = data
    for attr in attribute.split('.'):
        ptr = ptr[attr]

    return ptr

def set_attr(item, key, value, attr_key=None, attr_value=None):
    if attr_key and attr_value:
        actual_attr_value = get_attr(item, attr_key)

        if str(attr_value) != str(actual_attr_value):
            continue # We only want to set the values on hosts with defined attributes

        kvp = item
        for attr in key.split('.'):
            if attr == key.split('.')[len(key.split('.'))-1]:
                kvp[attr] = value
                continue
            if attr not in kvp:
                kvp[attr] = {}

            kvp = kvp[attr]
    return item


def set_attrs(items, key, value, attr_key=None, attr_value=None):
    for item in items:
        create_update_key(item, key, value, attr_key, attr_value)

    return items


def oo_set_node_label(arg, key, value, attr_key=None, attr_value=None):
    ''' This cycles through openshift node definitions
        (from "osc get nodes -o json"), and adds a label.

        If attr_key and attr_value are set, this will only set the label on
        nodes where the attribute matches the specified value.

        Ex:
        - shell: osc get nodes -o json
          register: output

        - set_fact:
          node_facts: "{{ output.stdout
                             | from_json
                             | oo_set_node_label('region', 'infra',
                                            'metadata.name', '172.16.17.43') }}"
    '''
    arg['items'] = set_attrs(arg['items'], key, value, attr_key, attr_value)

    return arg


def oo_set_resource_node(arg, value):
    arg = set_attr(arg, 'template.podTemplate.nodeSelector.region', value)

    return arg


class FilterModule(object):
    ''' FilterModule '''
    def filters(self):
        ''' returns a mapping of filters to methods '''
        return {
            "oo_set_node_label": oo_set_node_label,
            "oo_set_resource_node": oo_set_resource_node
        }