113 lines
2.9 KiB
Python
113 lines
2.9 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
from ansible.module_utils.basic import *
|
||
|
|
from ansible.module_utils.elasticsearch_api import *
|
||
|
|
|
||
|
|
class ElasticsearchTemplate:
|
||
|
|
def __init__(self, api, name):
|
||
|
|
self.api = api
|
||
|
|
self.name = name
|
||
|
|
self.exist = False
|
||
|
|
self.data = {}
|
||
|
|
|
||
|
|
def get_data(self):
|
||
|
|
status_code, data = self.api.get('_template/{}'.format(self.name))
|
||
|
|
if status_code == 200:
|
||
|
|
self.exist = True
|
||
|
|
self.data = data[self.name]
|
||
|
|
|
||
|
|
def dict_has_changed(self, new_data, data):
|
||
|
|
for option, value in new_data.items():
|
||
|
|
if not option in data:
|
||
|
|
if value:
|
||
|
|
return True
|
||
|
|
|
||
|
|
elif type(value) is dict:
|
||
|
|
if not type(data[option]) is dict:
|
||
|
|
return True
|
||
|
|
for k, v in value.items():
|
||
|
|
if not k in data[option]:
|
||
|
|
return True
|
||
|
|
if str(data[option][k]) != str(v):
|
||
|
|
return True
|
||
|
|
|
||
|
|
elif type(value) is list:
|
||
|
|
if not type(data[option]) is list:
|
||
|
|
return True
|
||
|
|
if data[option].sort() != value.sort():
|
||
|
|
return True
|
||
|
|
|
||
|
|
elif data[option] != value:
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
def has_changed(self, options):
|
||
|
|
if options['index_patterns'].sort() != self.data['index_patterns'].sort():
|
||
|
|
return True
|
||
|
|
|
||
|
|
if options['settings']:
|
||
|
|
if self.dict_has_changed(options['settings'], self.data['settings']):
|
||
|
|
return True
|
||
|
|
elif self.data['settings']:
|
||
|
|
return True
|
||
|
|
|
||
|
|
if options['mappings']['properties']:
|
||
|
|
if not self.data['mappings']:
|
||
|
|
return True
|
||
|
|
elif self.dict_has_changed(options['mappings']['properties'], self.data['mappings']['properties']):
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
if self.data['mappings']:
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
def create(self, options):
|
||
|
|
self.api.put(
|
||
|
|
'_template/{}'.format(self.name),
|
||
|
|
options
|
||
|
|
)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
fields = {
|
||
|
|
'name': { 'type': 'str', 'required': True },
|
||
|
|
'index_patterns': { 'type': 'list', 'default': [] },
|
||
|
|
'settings': { 'type': 'dict', 'default': {} },
|
||
|
|
'mappings': { 'type': 'dict', 'default': {} },
|
||
|
|
'api_url': { 'type': 'str', 'default': 'http://127.0.0.1:9200' },
|
||
|
|
'api_user': { 'type': 'str', 'default': None },
|
||
|
|
'api_password': { 'type': 'str', 'default': None },
|
||
|
|
}
|
||
|
|
module = AnsibleModule(argument_spec=fields)
|
||
|
|
changed = False
|
||
|
|
|
||
|
|
options = {
|
||
|
|
'index_patterns': module.params['index_patterns'],
|
||
|
|
'settings': module.params['settings'],
|
||
|
|
'mappings': {
|
||
|
|
'properties': module.params['mappings'],
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
api = ElasticsearchApi(
|
||
|
|
module.params['api_url'],
|
||
|
|
module.params['api_user'],
|
||
|
|
module.params['api_password']
|
||
|
|
)
|
||
|
|
|
||
|
|
template = ElasticsearchTemplate(
|
||
|
|
api,
|
||
|
|
module.params['name'],
|
||
|
|
)
|
||
|
|
template.get_data()
|
||
|
|
|
||
|
|
if not template.exist or template.has_changed(options):
|
||
|
|
template.create(options)
|
||
|
|
changed = True
|
||
|
|
|
||
|
|
module.exit_json(changed=changed)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|