API for K8S namespace policies + fix push images API

Author: cstenac

dataikuapi/dss/future.py

@@ -11,6 +11,11 @@ def __init__(self, client, job_id, state=None, result_wrapper=lambda result: res
        self.state_is_peek = True
        self.result_wrapper = result_wrapper
 
+    @staticmethod
+    def from_resp(client, resp,result_wrapper=lambda result: result):
+        """Creates a DSSFuture from a parsed JSON response"""
+        return DSSFuture(client, resp.get('jobId', None), state=resp, result_wrapper=result_wrapper)
+
     @classmethod
     def get_result_wait_if_needed(cls, client, ret):
         if 'jobId' in ret:

dataikuapi/dssclient.py

@@ -917,6 +917,24 @@ def create_personal_api_key(self, label):
         return self._perform_json("POST", "/auth/personal-api-keys",
                 params={"label": label})
 
+    ########################################################
+    # Container execution
+    ########################################################
+
+    def push_base_images(self):
+        """
+        Push base images for Kubernetes container-execution and Spark-on-Kubernetes
+        """
+        resp = self._perform_json("POST", "/admin/container-exec/actions/push-base-images")
+        return DSSFuture.from_resp(self, resp)
+
+    def apply_kubernetes_namespaces_policies(self):
+        """
+        Apply Kubernetes namespaces policies defined in the general settings
+        """
+        resp = self._perform_json("POST", "/admin/container-exec/actions/apply-kubernetes-policies")
+        return DSSFuture.from_resp(self, resp)
+
     ########################################################
     # Licensing
     ########################################################