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
|
import logging
import re
import subprocess
import time
import crmsh.userdir
import crmsh.utils
from crmsh.sh import ShellUtils
def get_online_nodes():
_, out, _ = ShellUtils().get_stdout_stderr('sudo crm_node -l')
if out:
return re.findall(r'[0-9]+ (.*) member', out)
else:
return None
def resource_cleanup():
subprocess.run(
['sudo', 'crm', 'resource', 'cleanup'],
stdin=subprocess.DEVNULL,
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL,
)
def before_step(context, step):
context.logger = logging.getLogger("Step:{}".format(step.name))
def before_tag(context, tag):
# tag @clean means need to stop cluster service
if tag == "clean":
time.sleep(3)
online_nodes = get_online_nodes()
if online_nodes:
resource_cleanup()
while True:
time.sleep(1)
rc, stdout, _ = ShellUtils().get_stdout_stderr('sudo crmadmin -D -t 1')
if rc == 0 and stdout.startswith('Designated'):
break
subprocess.call(
['sudo', 'crm', 'cluster', 'stop', '--all'],
stdin=subprocess.DEVNULL,
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL,
)
if tag == "skip_non_root":
sudoer = crmsh.userdir.get_sudoer()
if sudoer or crmsh.userdir.getuser() != 'root':
context.scenario.skip()
|
