Send logout to ZabbixAPI before exit (#58393)

pull/58422/head
Mikhail Naletov 5 years ago committed by ansibot
parent a688d2ee39
commit f6490422d9

@ -157,6 +157,7 @@ class Proxy(object):
def add_proxy(self, data): def add_proxy(self, data):
try: try:
if self._module.check_mode: if self._module.check_mode:
self._zapi.logout()
self._module.exit_json(changed=True) self._module.exit_json(changed=True)
parameters = {} parameters = {}
@ -165,24 +166,29 @@ class Proxy(object):
parameters[item] = data[item] parameters[item] = data[item]
proxy_ids_list = self._zapi.proxy.create(parameters) proxy_ids_list = self._zapi.proxy.create(parameters)
self._zapi.logout()
self._module.exit_json(changed=True, self._module.exit_json(changed=True,
result="Successfully added proxy %s (%s)" % result="Successfully added proxy %s (%s)" %
(data['host'], data['status'])) (data['host'], data['status']))
if len(proxy_ids_list) >= 1: if len(proxy_ids_list) >= 1:
return proxy_ids_list['proxyids'][0] return proxy_ids_list['proxyids'][0]
except Exception as e: except Exception as e:
self._zapi.logout()
self._module.fail_json(msg="Failed to create proxy %s: %s" % self._module.fail_json(msg="Failed to create proxy %s: %s" %
(data['host'], e)) (data['host'], e))
def delete_proxy(self, proxy_id, proxy_name): def delete_proxy(self, proxy_id, proxy_name):
try: try:
if self._module.check_mode: if self._module.check_mode:
self._zapi.logout()
self._module.exit_json(changed=True) self._module.exit_json(changed=True)
self._zapi.proxy.delete([proxy_id]) self._zapi.proxy.delete([proxy_id])
self._zapi.logout()
self._module.exit_json(changed=True, self._module.exit_json(changed=True,
result="Successfully deleted" + result="Successfully deleted"
" proxy %s" % proxy_name) + " proxy %s" % proxy_name)
except Exception as e: except Exception as e:
self._zapi.logout()
self._module.fail_json(msg="Failed to delete proxy %s: %s" % self._module.fail_json(msg="Failed to delete proxy %s: %s" %
(proxy_name, str(e))) (proxy_name, str(e)))
@ -204,6 +210,7 @@ class Proxy(object):
def update_proxy(self, proxy_id, data): def update_proxy(self, proxy_id, data):
try: try:
if self._module.check_mode: if self._module.check_mode:
self._zapi.logout()
self._module.exit_json(changed=True) self._module.exit_json(changed=True)
parameters = {'proxyid': proxy_id} parameters = {'proxyid': proxy_id}
@ -222,14 +229,17 @@ class Proxy(object):
if len(parameters) > 1: if len(parameters) > 1:
self._zapi.proxy.update(parameters) self._zapi.proxy.update(parameters)
self._zapi.logout()
self._module.exit_json( self._module.exit_json(
changed=True, changed=True,
result="Successfully updated proxy %s (%s)" % result="Successfully updated proxy %s (%s)" %
(data['host'], proxy_id) (data['host'], proxy_id)
) )
else: else:
self._zapi.logout()
self._module.exit_json(changed=False) self._module.exit_json(changed=False)
except Exception as e: except Exception as e:
self._zapi.logout()
self._module.fail_json(msg="Failed to update proxy %s: %s" % self._module.fail_json(msg="Failed to update proxy %s: %s" %
(data['host'], e)) (data['host'], e))

Loading…
Cancel
Save