Blame SOURCES/bz1977588-5-fence_kubevirt-get-namespace-from-context.patch

021bc5
From 647841dea9d93922779a4aa7d0b5f52f5bc2b4e9 Mon Sep 17 00:00:00 2001
021bc5
From: Dan Kenigsberg <danken@redhat.com>
021bc5
Date: Thu, 13 Jan 2022 14:57:26 +0200
021bc5
Subject: [PATCH] fence_kubevirt: take default namespace from context
021bc5
021bc5
If --namespace is not provided to kubectl, a default one is taken from
021bc5
kubeconfig context. Let fence_kubevirt behave similarly.
021bc5
021bc5
Signed-off-by: Dan Kenigsberg <danken@redhat.com>
021bc5
---
021bc5
 agents/kubevirt/fence_kubevirt.py      | 24 +++++++++++++-----------
021bc5
 2 files changed, 14 insertions(+), 12 deletions(-)
021bc5
021bc5
diff --git a/agents/kubevirt/fence_kubevirt.py b/agents/kubevirt/fence_kubevirt.py
021bc5
index 8392b75a0..8c27a0334 100755
021bc5
--- a/agents/kubevirt/fence_kubevirt.py
021bc5
+++ b/agents/kubevirt/fence_kubevirt.py
021bc5
@@ -12,12 +12,21 @@
021bc5
 except ImportError:
021bc5
     logging.error("Couldn\'t import kubernetes.client.exceptions.ApiException - not found or not accessible")
021bc5
 
021bc5
+def _get_namespace(options):
021bc5
+    from kubernetes import config
021bc5
+
021bc5
+    ns = options.get("--namespace")
021bc5
+    if ns is None:
021bc5
+        ns = config.kube_config.list_kube_config_contexts()[1]['context']['namespace']
021bc5
+
021bc5
+    return ns
021bc5
+
021bc5
 def get_nodes_list(conn, options):
021bc5
     logging.debug("Starting list/monitor operation")
021bc5
     result = {}
021bc5
     try:
021bc5
         apiversion = options.get("--apiversion")
021bc5
-        namespace = options.get("--namespace")
021bc5
+        namespace = _get_namespace(options)
021bc5
         include_uninitialized = True
021bc5
         vm_api = conn.resources.get(api_version=apiversion, kind='VirtualMachine')
021bc5
         vm_list = vm_api.get(namespace=namespace)
021bc5
@@ -31,7 +40,7 @@ def get_power_status(conn, options):
021bc5
     logging.debug("Starting get status operation")
021bc5
     try:
021bc5
         apiversion = options.get("--apiversion")
021bc5
-        namespace = options.get("--namespace")
021bc5
+        namespace = _get_namespace(options)
021bc5
         name = options.get("--plug")
021bc5
         vmi_api = conn.resources.get(api_version=apiversion,
021bc5
                                               kind='VirtualMachineInstance')
021bc5
@@ -61,7 +70,7 @@ def set_power_status(conn, options):
021bc5
     logging.debug("Starting set status operation")
021bc5
     try:
021bc5
         apiversion= options.get("--apiversion")
021bc5
-        namespace = options.get("--namespace")
021bc5
+        namespace = _get_namespace(options)
021bc5
         name = options.get("--plug")
021bc5
         action = 'start' if options["--action"] == "on" else 'stop'
021bc5
         virtctl_vm_action(conn, action, namespace, name, apiversion)
021bc5
@@ -75,7 +84,7 @@ def define_new_opts():
021bc5
         "longopt" : "namespace",
021bc5
         "help" : "--namespace=[namespace]        Namespace of the KubeVirt machine",
021bc5
         "shortdesc" : "Namespace of the KubeVirt machine.",
021bc5
-        "required" : "1",
021bc5
+        "required" : "0",
021bc5
         "order" : 2
021bc5
     }
021bc5
     all_opt["kubeconfig"] = {
021bc5
@@ -101,11 +110,6 @@ def virtctl_vm_action(conn, action, namespace, name, apiversion):
021bc5
     path = path.format(api_version=apiversion, namespace=namespace, name=name, action=action)
021bc5
     return conn.request('put', path, header_params={'accept': '*/*'})
021bc5
 
021bc5
-def validate_options(required_options_list, options):
021bc5
-    for required_option in required_options_list:
021bc5
-        if required_option not in options:
021bc5
-            fail_usage("Failed: %s option must be provided" % required_option)
021bc5
-
021bc5
 # Main agent method
021bc5
 def main():
021bc5
     conn = None
021bc5
@@ -127,8 +131,6 @@ def main():
021bc5
 
021bc5
     run_delay(options)
021bc5
 
021bc5
-    validate_options(['--namespace'], options)
021bc5
-
021bc5
     # Disable insecure-certificate-warning message
021bc5
     if "--ssl-insecure" in options:
021bc5
         import urllib3