ansible-role-elasticsearch/filter_plugins/custom.py

__author__ = 'dale mcdiarmid'

import re

def modify_list(values=[], pattern='', replacement='', ignorecase=False):
    ''' Perform a `re.sub` on every item in the list'''
    if ignorecase:
        flags = re.I
    else:
        flags = 0
    _re = re.compile(pattern, flags=flags)
    return [_re.sub(replacement, value) for value in values]

def append_to_list(values=[], suffix=''):
    if isinstance(values, basestring):
        values = values.split(',')
    return [str(value+suffix) for value in values]

def array_to_str(values=[],separator=','):
    return separator.join(values)

def extract_role_users(users={}):
    role_users=[]
    for user,details in users.iteritems():
        if "roles" in details:
            for role in details["roles"]:
                role_users.append(role+":"+user)
    return role_users


class FilterModule(object):
    def filters(self):
        return {'modify_list': modify_list,
        'append_to_list':append_to_list,
        'array_to_str':array_to_str,
        'extract_role_users':extract_role_users}