github/krkn
1
0
Fork 0
mirror of https://github.com/krkn-chaos/krkn.git synced 2026-02-14 18:10:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding in pod exec using k8s python client

Browse Source
This commit is contained in:
prubenda
2020-11-16 13:04:18 -05:00
parent 37fbbb5254
commit 5d86077f63
3 changed files with 29 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
kraken/kubernetes/client.py
View File

@@ -1,4 +1,5 @@
from kubernetes import client, config
from kubernetes.stream import stream
from kubernetes.client.rest import ApiException
import logging
import kraken.invoke.command as runcommand
@@ -72,6 +73,19 @@ def get_all_pods(label_selector=None):
return pods
# Execute command in pod
def exec_cmd_in_pod(command, pod_name, namespace):
exec_command = ['bash', '-c', command]
try:
ret = stream(cli.connect_get_namespaced_pod_exec, pod_name, namespace,
command=exec_command, stderr=True, stdin=False,
stdout=True, tty=False)
except Exception:
return False
return ret
# Obtain node status
def get_node_status(node):
try:

19
kraken/time_actions/common_time_functions.py
View File

@@ -10,8 +10,12 @@ import sys
def pod_exec(pod_name, command, namespace):
i = 0
for i in range(5):
response = runcommand.invoke('kubectl exec %s -n %s -- %s' % (pod_name, namespace, command))
if "unauthorized" in response.lower() or "authorization" in response.lower():
response = kubecli.exec_cmd_in_pod(command, pod_name, namespace)
if not response:
time.sleep(2)
continue
elif "unauthorized" in response.lower() or "authorization" in response.lower():
time.sleep(2)
continue
else:
break
@@ -19,13 +23,12 @@ def pod_exec(pod_name, command, namespace):
def node_debug(node_name, command):
response = runcommand.invoke("oc debug node/" + node_name + ' -- chroot /host ' + command)
return response
def skew_time(scenario):
skew_command = "date -s "
skew_command = "date --set "
if scenario['action'] == "skew_date":
skewed_date = "00-01-01"
skew_command += skewed_date
@@ -101,8 +104,8 @@ def check_date_time(object_type, names):
node_datetime = string_to_date(node_datetime_string)
counter = 0
while not first_date_time < node_datetime < datetime.datetime.utcnow():
time.sleep(5)
logging.info("Date/time on node %s still not reset, waiting 5 seconds and retrying"
time.sleep(10)
logging.info("Date/time on node %s still not reset, waiting 10 seconds and retrying"
% node_name)
node_datetime_string = node_debug(node_name, skew_command)
node_datetime = string_to_date(node_datetime_string)
@@ -121,8 +124,8 @@ def check_date_time(object_type, names):
pod_datetime_string = pod_exec(pod_name[0], skew_command, pod_name[1])
pod_datetime = string_to_date(pod_datetime_string)
while not first_date_time < pod_datetime < datetime.datetime.utcnow():
time.sleep(5)
logging.info("Date/time on pod %s still not reset, waiting 5 seconds and retrying"
time.sleep(10)
logging.info("Date/time on pod %s still not reset, waiting 10 seconds and retrying"
% pod_name[0])
first_date_time = datetime.datetime.utcnow()
pod_datetime = pod_exec(pod_name[0], skew_command, pod_name[1])

8
run_kraken.py
View File

@@ -161,10 +161,10 @@ def pod_scenarios(scenarios_list, config, failed_post_scenarios):
pre_action_output = run_post_action(kubeconfig_path, pod_scenario[1])
else:
pre_action_output = ''
scenario_logs = runcommand.invoke("powerfulseal autonomous --use-pod-delete-instead-of-ssh-kill"
" --policy-file %s --kubeconfig %s --no-cloud"
" --inventory-kubernetes --headless"
% (pod_scenario[0], kubeconfig_path))
scenario_logs = runcommand.invoke("powerfulseal autonomous --use-pod-delete-instead-"
"of-ssh-kill --policy-file %s --kubeconfig %s "
"--no-cloud --inventory-kubernetes --headless"
% (pod_scenario[0], kubeconfig_path))
# Display pod scenario logs/actions
print(scenario_logs)