From: Alfredo Deza <alfredo@deza.pe>
Date: Tue, 26 Feb 2019 14:56:09 +0000 (-0500)
Subject: backends: allow passing --context to kubernetes/oc connections
X-Git-Tag: 1.1.0~5
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=b39a1ba1f6bd553712457dec61a0db48f7544312;p=remoto.git

backends: allow passing --context to kubernetes/oc connections

Signed-off-by: Alfredo Deza <alfredo@deza.pe>
---

diff --git a/remoto/backends/kubernetes.py b/remoto/backends/kubernetes.py
index e524012..610bbd8 100644
--- a/remoto/backends/kubernetes.py
+++ b/remoto/backends/kubernetes.py
@@ -6,23 +6,29 @@ class KubernetesConnection(BaseConnection):
     executable = 'kubectl'
     remote_import_system = 'json'
 
-    def __init__(self, pod_name, namespace=None, **kw):
+    def __init__(self, pod_name, namespace=None, context=None, **kw):
         self.namespace = namespace
+        self.context = context
         self.pod_name = pod_name
         super(KubernetesConnection, self).__init__(hostname='localhost', **kw)
 
     def command_template(self):
+        base_command = [self.executable]
+        if self.context:
+            base_command.extend(['--context', self.context])
+
+        base_command.extend(['exec', '-i'])
+
         if self.namespace:
-            prefix = [
-                self.executable, 'exec', '-i', '-n',
-                self.namespace, self.pod_name, '--', '/bin/sh', '-c'
-            ]
-        else:
-            prefix = [
-                self.executable, 'exec', '-i',
-                self.pod_name, '--', '/bin/sh', '-c'
-            ]
-        return prefix
+            base_command.extend(['-n', self.namespace])
+
+        base_command.extend([
+            self.pod_name,
+            '--',
+            '/bin/sh',
+            '-c'
+        ])
+        return base_command
 
     def cmd(self, cmd):
         tmpl = self.command_template()