1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
##############################################################################
# Copyright (c) 2016 NEC Corporation and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import argparse
import json
import os
from keystoneclient.v2_0 import client
import requests
def force_down(hostname, force_down=True):
keystone = client.Client(username=os.environ['OS_USERNAME'],
password=os.environ['OS_PASSWORD'],
tenant_name=os.environ['OS_TENANT_NAME'],
auth_url=os.environ['OS_AUTH_URL'])
for service in keystone.auth_ref['serviceCatalog']:
if service['type'] == 'compute':
base_url = service['endpoints'][0]['internalURL']
break
url = '%s/os-services/force-down' % base_url.replace('/v2/', '/v2.1/')
data = {
'forced_down': force_down,
'binary': 'nova-compute',
'host': hostname,
}
headers = {
'Content-Type': 'application/json',
'Accept': 'application/json',
'X-Auth-Token': keystone.auth_ref['token']['id'],
'X-OpenStack-Nova-API-Version': 2.11,
}
print requests.put(url, data=json.dumps(data), headers=headers)
def get_args():
parser = argparse.ArgumentParser(description='Doctor Test Cleaner')
parser.add_argument('hostname', metavar='HOSTNAME', type=str, nargs='?',
help='a nova-compute hostname to force down')
parser.add_argument('--unset', action='store_true', default=False,
help='unset force_down flag')
return parser.parse_args()
def main():
args = get_args()
force_down(args.hostname, not(args.unset))
if __name__ == '__main__':
main()
|
