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
164
165
166
167
168
|
"""Library for openshift operations.
Various utility functions for interacting with OCP/OpenShift.
"""
import re
import types
import yaml
from glusto.core import Glusto as g
PODS_WIDE_RE = re.compile(
'(\S+)\s+(\S+)\s+(\w+)\s+(\d+)\s+(\S+)\s+(\S+)\s+(\S+)\n')
def oc_get_pods(ocp_node):
"""Gets the pods info with 'wide' option in the current project.
Args:
ocp_node (str): Node in which ocp command will be executed.
Returns:
dict : dict of pods info in the current project.
"""
cmd = "oc get -o wide --no-headers=true pods"
ret, out, err = g.run(ocp_node, cmd)
if ret != 0:
g.log.error("Failed to get ocp pods on node %s" % ocp_node)
raise AssertionError('failed to get pods: %r' % (err,))
return _parse_wide_pods_output(out)
def _parse_wide_pods_output(output):
"""Parse the output of `oc get -o wide pods`.
"""
# Interestingly, the output of get pods is "cooked" in such a way that
# the values in the ready, status, & restart fields are not accessible
# from YAML/JSON/templating forcing us to scrape the output for
# these values
# (at the time of this writing the logic is in
# printPodBase in kubernetes/pkg/printers/internalversion/printers.go )
# Possibly obvious, but if you don't need those values you can
# use the YAML output directly.
#
# TODO: Add unit tests for this parser
pods_info = {}
for each_pod_info in PODS_WIDE_RE.findall(output):
pods_info[each_pod_info[0]] = {
'ready': each_pod_info[1],
'status': each_pod_info[2],
'restarts': each_pod_info[3],
'age': each_pod_info[4],
'ip': each_pod_info[5],
'node': each_pod_info[6],
}
return pods_info
def oc_get_pods_full(ocp_node):
"""Gets all the pod info via YAML in the current project.
Args:
ocp_node (str): Node in which ocp command will be executed.
Returns:
dict: The YAML output converted to python objects
(a top-level dict)
"""
cmd = "oc get -o yaml pods"
ret, out, err = g.run(ocp_node, cmd)
if ret != 0:
g.log.error("Failed to get ocp pods on node %s" % ocp_node)
raise AssertionError('failed to get pods: %r' % (err,))
return yaml.load(out)
def get_ocp_gluster_pod_names(ocp_node):
"""Gets the gluster pod names in the current project.
Args:
ocp_node (str): Node in which ocp command will be executed.
Returns:
list : list of gluster pod names in the current project.
Empty list, if there are no gluster pods.
Example:
get_ocp_gluster_pod_names(ocp_node)
"""
pod_names = oc_get_pods(ocp_node).keys()
return [pod for pod in pod_names if pod.startswith('glusterfs-')]
def oc_login(ocp_node, username, password):
"""Login to ocp master node.
Args:
ocp_node (str): Node in which ocp command will be executed.
username (str): username of ocp master node to login.
password (str): password of ocp master node to login.
Returns:
bool : True on successful login to ocp master node.
False otherwise
Example:
oc_login(ocp_node, "test","test")
"""
cmd = "oc login --username=%s --password=%s" % (username, password)
ret, _, _ = g.run(ocp_node, cmd)
if ret != 0:
g.log.error("Failed to login to ocp master node %s" % ocp_node)
return False
return True
def switch_oc_project(ocp_node, project_name):
"""Switch to the given project.
Args:
ocp_node (str): Node in which ocp command will be executed.
project_name (str): Project name.
Returns:
bool : True on switching to given project.
False otherwise
Example:
switch_oc_project(ocp_node, "storage-project")
"""
cmd = "oc project %s" % project_name
ret, _, _ = g.run(ocp_node, cmd)
if ret != 0:
g.log.error("Failed to switch to project %s" % project_name)
return False
return True
def oc_rsh(ocp_node, pod_name, command, log_level=None):
"""Run a command in the ocp pod using `oc rsh`.
Args:
ocp_node (str): Node on which oc rsh command will be executed.
pod_name (str): Name of the pod on which the command will
be executed.
command (str|list): command to run.
log_level (str|None): log level to be passed to glusto's run
method.
Returns:
A tuple consisting of the command return code, stdout, and stderr.
"""
prefix = ['oc', 'rsh', pod_name]
if isinstance(command, types.StringTypes):
cmd = ' '.join(prefix + [command])
else:
cmd = prefix + command
# unpack the tuple to make sure our return value exactly matches
# our docstring
ret, stdout, stderr = g.run(ocp_node, cmd, log_level=log_level)
return (ret, stdout, stderr)
|
