Skip to content
Snippets Groups Projects
Commit 49219be9 authored by jurgenhaas's avatar jurgenhaas
Browse files

Adjust plugin for Ansible version 2

parent d10c6313
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
action_plugins/serverdensity.py
+ 67
− 79
View file @ 49219be9
......@@ -4,21 +4,30 @@ import requests
import tempfile
import yaml
from ansible.callbacks import vv
from ansible.errors import AnsibleError as ae
from ansible.runner.return_data import ReturnData
from ansible.utils import parse_kv
class ActionModule(object):
from ansible.plugins.action import ActionBase
from ansible.utils.boolean import boolean
from ansible.vars import VariableManager
try:
from __main__ import display
except ImportError:
from ansible.utils.display import Display
display = Display()
class ActionModule(ActionBase):
''' Create new host or sync all of your inventory over at ServerDensity'''
### We need to be able to modify the inventory
BYPASS_HOST_LOOP = True
TRANSFERS_FILES = False
def __init__(self, runner):
self.runner = runner
def __init__(self, task, connection, play_context, loader, templar,
shared_loader_obj):
super(ActionModule, self).__init__(task, connection, play_context,
loader, templar, shared_loader_obj)
self.api_token = None
self.args = {}
self.vars = {}
self.devices = []
self.services = []
......@@ -29,30 +38,28 @@ class ActionModule(object):
self.force_update = False
self.cache_file_name = None
def run(self, conn, tmp, module_name, module_args, inject, complex_args=None, **kwargs):
def run(self, tmp=None, task_vars=None):
''' handler for file transfer operations '''
if task_vars is None:
task_vars = dict()
self.vars = task_vars
if self.runner.noop_on_check(inject):
return ReturnData(conn=conn, comm_ok=True, result=dict(skipped=True, msg='check mode not supported for this module'))
result = super(ActionModule, self).run(tmp, task_vars)
args = {}
if complex_args:
args.update(complex_args)
args.update(parse_kv(module_args))
if 'api_token' in args:
self.api_token = args.get('api_token')
else:
allgroup = self.runner.inventory.get_group('all')
allvariables = allgroup.get_variables()
if 'sd_api_token' in allvariables:
self.api_token = allvariables.get('sd_api_token')
else:
raise ae("'api_token' is a required argument.")
if self._play_context.check_mode:
result['skipped'] = True
result['msg'] = 'check mode not supported for this module'
return result
self.api_token = self._task.args.get('api_token', task_vars.get('sd_api_token', False))
if not self.api_token:
raise ae("'api_token' is a required argument or you define the variable 'sd_api_token' in your inventory.")
self.force_update = args.get('force', False)
self.cache_file_name = args.get('cache', False)
cleanup = args.get('cleanup', False)
just_download = args.get('readonly', False)
output = args.get('output', False)
self.force_update = boolean(self._task.args.get('force'))
self.cache_file_name = self._task.args.get('cache', False)
cleanup = boolean(self._task.args.get('cleanup'))
just_download = boolean(self._task.args.get('readonly'))
output = self._task.args.get('output', False)
if just_download:
self.force_update = False
......@@ -62,30 +69,29 @@ class ActionModule(object):
if output and self.cache_file_name != 'False':
self.cache_file_name = tempfile.mktemp(prefix='sd_', suffix='.json')
result = {}
self.list_all()
if output:
if os.path.exists(self.cache_file_name):
vv('Writing upstream settings to %s' % os.path.abspath(output))
display.vv('Writing upstream settings to %s' % os.path.abspath(output))
with open(self.cache_file_name, 'r') as content_file:
content = json.load(content_file)
with open(output, 'w') as output_file:
yaml.safe_dump(content, output_file, encoding='utf-8', allow_unicode=True)
if just_download:
vv('Downloaded settings to %s' % self.cache_file_name)
return ReturnData(conn=conn, comm_ok=True, result=result)
result['msg'] = 'Downloaded settings to %s' % self.cache_file_name
result['changed'] = False
return result
services = {}
devicegroup_alerts = {}
servicegroup_alerts = {}
vv('Ensure hosts...')
for host in self.runner.host_set:
vv('- ' + host)
host_vars = inject['hostvars'][host]
display.vv('Ensure hosts...')
for host in self.vars.get('ansible_play_hosts'):
display.vv('- ' + host)
host_vars = self.vars['hostvars'][host]
facts = host_vars.get('ansible_facts', {})
location = host_vars.get('location')
if not location:
......@@ -145,50 +151,49 @@ class ActionModule(object):
alerts = host_vars.get('sd_alerts')
if alerts:
vv('- - Ensure device alerts...')
display.vv('- - Ensure device alerts...')
for alertname in alerts:
vv('- - - ' + alertname)
display.vv('- - - ' + alertname)
alert = alerts.get(alertname)
alert.__setitem__('host', host)
self.ensure_alert(alert, 'device')
vv('Ensure device group alerts...')
display.vv('Ensure device group alerts...')
for groupname in devicegroup_alerts:
vv('- ' + groupname)
display.vv('- ' + groupname)
group_alerts = devicegroup_alerts.get(groupname)
for alertname in group_alerts:
vv('- - ' + alertname)
display.vv('- - ' + alertname)
alert = group_alerts.get(alertname)
self.ensure_alert(alert, 'deviceGroup', groupname)
vv('Ensure services...')
display.vv('Ensure services...')
for servicename in services:
vv('- ' + servicename)
display.vv('- ' + servicename)
service = services.get(servicename)
self.ensure_service(servicename, service)
alerts = service.get('alerts')
if alerts:
vv('- - Ensure service alerts...')
display.vv('- - Ensure service alerts...')
for alertname in alerts:
vv('- - - ' + alertname)
display.vv('- - - ' + alertname)
alert = alerts.get(alertname)
alert.__setitem__('service', service.get('name'))
self.ensure_alert(alert, 'service')
vv('Ensure service group alerts...')
display.vv('Ensure service group alerts...')
for servicegroupname in servicegroup_alerts:
vv('- ' + servicegroupname)
display.vv('- ' + servicegroupname)
alert = servicegroup_alerts.get(servicegroupname)
groupname = alert.get('group')
self.ensure_alert(alert, 'serviceGroup', groupname)
if cleanup:
vv('Cleanup unused alerts...')
display.vv('Cleanup unused alerts...')
self.cleanup_alerts()
vv('Completed successfully!')
return ReturnData(conn=conn, comm_ok=True, result=result)
result['msg'] = 'Completed successfully!'
return result
def _request(self, path, data = None, method = 'GET'):
encoder = json.JSONEncoder()
......@@ -300,27 +305,18 @@ class ActionModule(object):
return False
def _list_objects(self, type, path):
vv("Reading %s from SD" % type)
display.vv("Reading %s from SD" % type)
changed = False
allgroup = self.runner.inventory.get_group('all')
allvariables = allgroup.get_variables()
if '_serverdensity_' + type not in allvariables:
if not self.vars.has_key('_serverdensity_' + type):
changed = True
objects = self._request(path)
allgroup.set_variable('_serverdensity_' + type, objects)
else:
objects = allvariables.get('_serverdensity_' + type)
return changed, objects
self.vars.__setitem__('_serverdensity_' + type, self._request(path))
return changed, self.vars.get('_serverdensity_' + type)
def _list_devices_agent_key(self):
for device in self.devices:
hostname = device.get('name')
if hostname in self.runner.inventory._hosts_cache:
host = self.runner.inventory._hosts_cache[hostname]
host.set_variable('sd_agent_key', device.get('agentKey'))
if hostname in self.runner.inventory._vars_per_host:
hostvars = self.runner.inventory._vars_per_host.get(hostname)
hostvars.__setitem__('sd_agent_key', device.get('agentKey'))
if hostname in self.vars['ansible_play_hosts']:
self.vars['hostvars'][hostname]['sd_agent_key'] = device.get('agentKey')
def list_devices(self):
if len(self.devices) == 0:
......@@ -342,12 +338,10 @@ class ActionModule(object):
def list_notifications(self):
if len(self.notifications) == 0:
allgroup = self.runner.inventory.get_group('all')
allvariables = allgroup.get_variables()
if 'sd_notifications' in allvariables:
self.notifications = allvariables.get('sd_notifications')
if self.vars.has_key('sd_notifications'):
self.notifications = self.vars.get('sd_notifications')
else:
filename = self.runner.inventory.basedir() + '/sd_notifications.json'
filename = self.vars.get('inventory_dir') + '/sd_notifications.json'
if os.path.exists(filename):
with open(filename, 'r') as content_file:
self.notifications = json.load(content_file)
......@@ -428,13 +422,7 @@ class ActionModule(object):
self.devices.remove(old_device)
self.devices.append(device)
self.cache_update(False)
if hostname in self.runner.inventory._hosts_cache:
host = self.runner.inventory._hosts_cache[hostname]
host.set_variable('sd_agent_key', device.get('agentKey'))
if hostname in self.runner.inventory._vars_per_host:
hostvars = self.runner.inventory._vars_per_host.get(hostname)
hostvars.__setitem__('sd_agent_key', device.get('agentKey'))
self.vars['hostvars'][hostname]['sd_agent_key'] = device.get('agentKey')
def ensure_service(self, servicename, service):
serviceId = self._get_service_id(servicename)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment