Merge "Fixing network configuration bug in dea.yaml"
[genesis.git] / compass / deploy / status_callback.py
1 # (C) 2012, Michael DeHaan, <michael.dehaan@gmail.com>
2
3 # This file is part of Ansible
4 #
5 # Ansible is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # Ansible is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with Ansible.  If not, see <http://www.gnu.org/licenses/>.
17
18 import httplib
19 import json
20 import sys
21 import logging
22
23 def task_error(host, data):
24     logging.info("task_error: host=%s,data=%s" % (host, data))
25
26     if type(data) == dict:
27         invocation = data.pop('invocation', {})
28
29     notify_host("localhost", host, "failed")
30
31 class CallbackModule(object):
32     """
33     logs playbook results, per host, in /var/log/ansible/hosts
34     """
35
36     def on_any(self, *args, **kwargs):
37         pass
38
39     def runner_on_failed(self, host, res, ignore_errors=False):
40         task_error(host, res)
41
42     def runner_on_ok(self, host, res):
43         pass
44
45     def runner_on_skipped(self, host, item=None):
46         pass
47
48     def runner_on_unreachable(self, host, res):
49         pass
50
51     def runner_on_no_hosts(self):
52         pass
53
54     def runner_on_async_poll(self, host, res, jid, clock):
55         pass
56
57     def runner_on_async_ok(self, host, res, jid):
58         pass
59
60     def runner_on_async_failed(self, host, res, jid):
61         task_error(host, res)
62
63     def playbook_on_start(self):
64         pass
65
66     def playbook_on_notify(self, host, handler):
67         pass
68
69     def playbook_on_no_hosts_matched(self):
70         pass
71
72     def playbook_on_no_hosts_remaining(self):
73         pass
74
75     def playbook_on_task_start(self, name, is_conditional):
76         pass
77
78     def playbook_on_vars_prompt(self, varname, private=True, prompt=None, encrypt=None, confirm=False, salt_size=None, salt=None, default=None):
79         pass
80
81     def playbook_on_setup(self):
82         pass
83
84     def playbook_on_import_for_host(self, host, imported_file):
85         pass
86
87     def playbook_on_not_import_for_host(self, host, missing_file):
88         pass
89
90     def playbook_on_play_start(self, name):
91         pass
92
93     def playbook_on_stats(self, stats):
94         logging.info("playbook_on_stats enter")
95         hosts = sorted(stats.processed.keys())
96         host_vars = self.playbook.inventory.get_variables(hosts[0])
97         cluster_name = host_vars['cluster_name']
98         failures = False
99         unreachable = False
100
101         for host in hosts:
102             summary = stats.summarize(host)
103
104             if summary['failures'] > 0:
105                 failures = True
106             if summary['unreachable'] > 0:
107                 unreachable = True
108
109         if failures or unreachable:
110             for host in hosts:
111                 notify_host("localhost", host, "error")
112             return
113
114         for host in hosts:
115             clusterhost_name = host + "." + cluster_name
116             notify_host("localhost", clusterhost_name, "succ")
117
118
119 def raise_for_status(resp):
120     if resp.status < 200 or resp.status > 300:
121         raise RuntimeError("%s, %s, %s" % (resp.status, resp.reason, resp.read()))
122
123 def auth(conn):
124     credential = {}
125     credential['email'] = "admin@huawei.com"
126     credential['password'] = "admin"
127     url = "/api/users/token"
128     headers = {"Content-type": "application/json",
129                "Accept": "*/*"}
130     conn.request("POST", url, json.dumps(credential), headers)
131     resp = conn.getresponse()
132
133     raise_for_status(resp)
134     return json.loads(resp.read())["token"]
135
136 def notify_host(compass_host, host, status):
137     if status == "succ":
138         body = {"ready": True}
139         url = "/api/clusterhosts/%s/state_internal" % host
140     elif status == "error":
141         body = {"state": "ERROR"}
142         host = host.strip("host")
143         url = "/api/clusterhosts/%s/state" % host
144     else:
145         logging.error("notify_host: host %s with status %s is not supported" \
146                 % (host, status))
147         return
148
149     headers = {"Content-type": "application/json",
150                "Accept": "*/*"}
151
152     conn = httplib.HTTPConnection(compass_host, 80)
153     token = auth(conn)
154     headers["X-Auth-Token"] = token
155     logging.info("host=%s,url=%s,body=%s,headers=%s" % (compass_host,url,json.dumps(body),headers))
156     conn.request("POST", url, json.dumps(body), headers)
157     resp = conn.getresponse()
158     try:
159         raise_for_status(resp)
160         logging.info("notify host status success!!! status=%s, body=%s" % (resp.status, resp.read()))
161     except Exception as e:
162         logging.error("http request failed %s" % str(e))
163         raise
164     finally:
165         conn.close()
166
167 if __name__ == "__main__":
168     if len(sys.argv) != 3:
169         logging.error("params: host, status is need")
170         sys.exit(1)
171
172     host = sys.argv[1]
173     status = sys.argv[2]
174     notify_host(host, status)