import httplib import json import sys # noqa:F401 from ansible.plugins.callback import CallbackBase def task_error(display, host, data): display.display("task_error: host=%s,data=%s" % (host, data)) # if isinstance(data, dict): # invocation = data.pop('invocation', {}) notify_host(display, "localhost", host, "failed") class CallbackModule(CallbackBase): """ logs playbook results, per host, in /var/log/ansible/hosts """ CALLBACK_VERSION = 2.0 CALLBACK_TYPE = 'notification' CALLBACK_NAME = 'status_callback' CALLBACK_NEEDS_WHITELIST = True def __init__(self): super(CallbackModule, self).__init__() def v2_on_any(self, *args, **kwargs): pass def v2_runner_on_failed(self, host, res, ignore_errors=False): task_error(self._display, host, res) def v2_runner_on_ok(self, host, res): pass def v2_runner_on_skipped(self, host, item=None): pass def v2_runner_on_unreachable(self, host, res): pass def v2_runner_on_no_hosts(self): pass def v2_runner_on_async_poll(self, host, res, jid, clock): pass def v2_runner_on_async_ok(self, host, res, jid): pass def v2_runner_on_async_failed(self, host, res, jid): task_error(self._display, host, res) def v2_playbook_on_start(self): pass def v2_playbook_on_notify(self, host, handler): pass def v2_playbook_on_no_hosts_matched(self): pass def v2_playbook_on_no_hosts_remaining(self): pass def v2_playbook_on_task_start(self, name, is_conditional): pass def v2_playbook_on_vars_prompt(self, varname, private=True, prompt=None, encrypt=None, confirm=False, salt_size=None, salt=None, default=None): # noqa pass def v2_playbook_on_setup(self): pass def v2_playbook_on_import_for_host(self, host, imported_file): pass def v2_playbook_on_not_import_for_host(self, host, missing_file): pass def v2_playbook_on_play_start(self, play): self.play = play self.loader = self.play.get_loader() return def v2_playbook_on_stats(self, stats): self._display.display("playbook_on_stats enter") all_vars = self.play.get_variable_manager().get_vars(self.loader) host_vars = all_vars["hostvars"] hosts = sorted(stats.processed.keys()) cluster_name = host_vars[hosts[0]]['cluster_name'] failures = False unreachable = False for host in hosts: summary = stats.summarize(host) if summary['failures'] > 0: failures = True if summary['unreachable'] > 0: unreachable = True if failures or unreachable: for host in hosts: notify_host(self._display, "localhost", host, "error") return for host in hosts: clusterhost_name = host + "." + cluster_name notify_host(self._display, "localhost", clusterhost_name, "succ") def raise_for_status(resp): if resp.status < 200 or resp.status > 300: raise RuntimeError( "%s, %s, %s" % (resp.status, resp.reason, resp.read())) def auth(conn): credential = {} credential['email'] = "admin@huawei.com" credential['password'] = "admin" url = "/api/users/token" headers = {"Content-type": "application/json", "Accept": "*/*"} conn.request("POST", url, json.dumps(credential), headers) resp = conn.getresponse() raise_for_status(resp) return json.loads(resp.read())["token"] def notify_host(display, compass_host, host, status): if status == "succ": body = {"ready": True} url = "/api/clusterhosts/%s/state_internal" % host elif status == "error": body = {"state": "ERROR"} host = host.strip("host") url = "/api/clusterhosts/%s/state" % host else: display.error("notify_host: host %s with status %s is not supported" % (host, status)) return headers = {"Content-type": "application/json", "Accept": "*/*"} conn = httplib.HTTPConnection(compass_host, 80) token = auth(conn) headers["X-Auth-Token"] = token display.display("host=%s,url=%s,body=%s,headers=%s" % (compass_host, url, json.dumps(body), headers)) conn.request("POST", url, json.dumps(body), headers) resp = conn.getresponse() try: raise_for_status(resp) display.display( "notify host status success!!! status=%s, body=%s" % (resp.status, resp.read())) except Exception as e: display.error("http request failed %s" % str(e)) raise finally: conn.close()