Add SB Event API specification
[doctor.git] / tests / inspector.py
1 ##############################################################################
2 # Copyright (c) 2016 NEC Corporation and others.
3 #
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
9
10 import argparse
11 from flask import Flask
12 from flask import request
13 import json
14 import os
15
16 import novaclient.client as novaclient
17
18 import nova_force_down
19
20
21 class DoctorInspectorSample(object):
22
23     nova_api_version = 2.11
24
25     def __init__(self):
26         self.nova = novaclient.Client(self.nova_api_version,
27                                       os.environ['OS_USERNAME'],
28                                       os.environ['OS_PASSWORD'],
29                                       os.environ['OS_TENANT_NAME'],
30                                       os.environ['OS_AUTH_URL'],
31                                       connection_pool=True)
32         # check nova is available
33         self.nova.servers.list(detailed=False)
34
35     def disable_compute_host(self, hostname):
36         opts = {'all_tenants': True, 'host': hostname}
37         for server in self.nova.servers.list(detailed=False, search_opts=opts):
38             self.nova.servers.reset_state(server, 'error')
39
40         # NOTE: We use our own client here instead of this novaclient for a
41         #       workaround.  Once keystone provides v2.1 nova api endpoint
42         #       in the service catalog which is configured by OpenStack
43         #       installer, we can use this:
44         #
45         # self.nova.services.force_down(hostname, 'nova-compute', True)
46         #
47         nova_force_down.force_down(hostname)
48
49
50 app = Flask(__name__)
51 inspector = DoctorInspectorSample()
52
53
54 @app.route('/events', methods=['POST'])
55 def event_posted():
56     app.logger.debug('event posted')
57     app.logger.debug('inspector = %s' % inspector)
58     app.logger.debug('received data = %s' % request.data)
59     d = json.loads(request.data)
60     hostname = d['hostname']
61     event_type = d['type']
62     if event_type == 'compute.host.down':
63         inspector.disable_compute_host(hostname)
64     return "OK"
65
66
67 def get_args():
68     parser = argparse.ArgumentParser(description='Doctor Sample Inspector')
69     parser.add_argument('port', metavar='PORT', type=int, nargs='?',
70                         help='a port for inspector')
71     return parser.parse_args()
72
73
74 def main():
75     args = get_args()
76     app.run(port=args.port, debug=True)
77
78
79 if __name__ == '__main__':
80     main()