Propagate secret to kaniko (#423)

* propagate secret to kaniko spec * update name
kubeflow · Nov 30, 2018 · 537357a · 537357a
1 parent 052de19
commit 537357a
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/sdk/python/kfp/compiler/_k8s_helper.py b/sdk/python/kfp/compiler/_k8s_helper.py
@@ -36,11 +36,25 @@ def _create_k8s_job(self, yaml_spec):
     """ _create_k8s_job creates a kubernetes job based on the yaml spec """
     pod = k8s_client.V1Pod(metadata=k8s_client.V1ObjectMeta(generate_name=yaml_spec['metadata']['generateName']))
     container = k8s_client.V1Container(name = yaml_spec['spec']['containers'][0]['name'],
-                                       image= yaml_spec['spec']['containers'][0]['image'],
-                                       args = yaml_spec['spec']['containers'][0]['args'])
+                                       image = yaml_spec['spec']['containers'][0]['image'],
+                                       args = yaml_spec['spec']['containers'][0]['args'],
+                                       volume_mounts = [k8s_client.V1VolumeMount(
+                                           name=yaml_spec['spec']['containers'][0]['volumeMounts'][0]['name'],
+                                           mount_path=yaml_spec['spec']['containers'][0]['volumeMounts'][0]['mountPath'],
+                                       )],
+                                       env = [k8s_client.V1EnvVar(
+                                           name=yaml_spec['spec']['containers'][0]['env'][0]['name'],
+                                           value=yaml_spec['spec']['containers'][0]['env'][0]['value'],
+                                       )])
     pod.spec = k8s_client.V1PodSpec(restart_policy=yaml_spec['spec']['restartPolicy'],
                                     containers = [container],
-                                    service_account_name=yaml_spec['spec']['serviceAccountName'])
+                                    service_account_name=yaml_spec['spec']['serviceAccountName'],
+                                    volumes=[k8s_client.V1Volume(
+                                        name=yaml_spec['spec']['volumes'][0]['name'],
+                                        secret=k8s_client.V1SecretVolumeSource(
+                                           secret_name=yaml_spec['spec']['volumes'][0]['secret']['secretName'],
+                                        )
+                                    )])
     try:
       api_response = self._corev1.create_namespaced_pod(yaml_spec['metadata']['namespace'], pod)
       return api_response.metadata.name, True