This repository has been archived by the owner on Jan 8, 2024. It is now read-only.
/
fence_evacuate.py
163 lines (139 loc) · 4.57 KB
/
fence_evacuate.py
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
#!/usr/bin/python -tt
import sys
import logging
import atexit
from mistralclient.api import client
sys.path.append("/usr/share/fence")
from fencing import run_delay, all_opt, atexit_handler, check_input, \
process_input, show_docs
def define_new_opts():
all_opt["domain"] = {
"getopt": "d:",
"longopt": "domain",
"help": "-d, --domain=[string] DNS domain in which hosts"
" live, useful when the cluster uses short names and"
" nova uses FQDN",
"required": "0",
"shortdesc": "DNS domain in which hosts live",
"default": "",
"order": 5,
}
all_opt["user"] = {
"getopt": "u:",
"longopt": "user",
"help": "-u, --user=string User to connect to Keystone",
"required": "1",
"shortdesc": "User to connect to Keystone",
"default": "",
"order": 5,
}
all_opt["password"] = {
"getopt": "p:",
"longopt": "password",
"help": "-p, --password=string Password to connect to Keystone",
"required": "1",
"shortdesc": "Password to connect to Keystone",
"default": "",
"order": 5,
}
all_opt["tenant"] = {
"getopt": "t:",
"longopt": "tenant",
"help": "-t, --tenant=string Tenant name to connect to Keystone",
"required": "1",
"shortdesc": "Tenant name to connect to Keystone",
"default": "",
"order": 5,
}
all_opt["auth_url"] = {
"getopt": "a:",
"longopt": "auth_url",
"help": "-a, --auth_url=string auth_url to connect to Keystone",
"required": "1",
"shortdesc": "auth_url to connect to Keystone",
"default": "",
"order": 5,
}
all_opt["wf_name"] = {
"getopt": "w:",
"longopt": "wf_name",
"help": "-w, --wf_name=string Workflow name to run by Mistral",
"required": "1",
"shortdesc": "Workflow name to run by Mistral",
"default": "",
"order": 5,
}
all_opt["mistral_url"] = {
"getopt": "m:",
"longopt": "mistral_url",
"help": "-m, --mistral_url=string URL where Mistral is listening",
"required": "1",
"shortdesc": "URL where Mistral is listening",
"default": "",
"order": 5,
}
all_opt["on_shared_storage"] = {
"getopt": "o:",
"longopt": "on_shared_storage",
"help": "-o, --on_shared_storage=string set to true if OpenStack uses"
"shared storage",
"required": "1",
"shortdesc": "set to true if OpenStack uses shared storage",
"default": "",
"order": 5,
}
def str2bool(v):
return v.lower() in ("yes", "true", "t", "1")
def format_message(host, on_shared_storage):
result = {
"search_opts": {
"host": host
},
"on_shared_storage": on_shared_storage
}
return result
def evacuate(host, options):
mistral_url = options["--mistral_url"]
user = options["--user"]
password = options["--password"]
tenant = options["--tenant"]
auth_url = options["--auth_url"]
wf_name = options["--wf_name"]
on_shared_storage = str2bool(options["--on_shared_storage"])
input_json = format_message(host, on_shared_storage)
mistral = client.client(
mistral_url=mistral_url,
username=user,
api_key=password,
project_name=tenant,
auth_url=auth_url
)
mistral.executions.create(wf_name,
input_json)
def main():
atexit.register(atexit_handler)
device_opt = ["user", "domain", "password", "tenant", "auth_url",
"wf_name", "port", "mistral_url", "on_shared_storage"]
define_new_opts()
all_opt["shell_timeout"]["default"] = "180"
options = check_input(device_opt, process_input(device_opt))
docs = {}
docs["shortdesc"] = "Fence agent for nova compute nodes"
docs["longdesc"] = "fence_evacuate is a Nova fencing notification agent"
docs["vendorurl"] = ""
show_docs(options, docs)
run_delay(options)
host = None
# Potentially we should make this a pacemaker feature
if options["--domain"] != "" and "--plug" in options:
options["--plug"] = options["--plug"] + "." + options["--domain"]
if "--plug" in options:
host = options["--plug"]
if options['--action'] in ['reboot', 'off']:
if host is None:
logging.error('No host specified')
sys.exit(1)
evacuate(host, options)
sys.exit(0)
if __name__ == "__main__":
main()