Add method to dynamically set namespace in yaml files

Author: johscheuer

kubernetes/utils/create_from_yaml.py

@@ -23,13 +23,14 @@
 from kubernetes import client
 
 
-def create_from_yaml(k8s_client, yaml_file, verbose=False, **kwargs):
+def create_namespaced_from_yaml(k8s_client, yaml_file, verbose=False, namespace=None, **kwargs):
     """
     Perform an action from a yaml file. Pass True for verbose to
     print confirmation information.
     Input:
     yaml_file: string. Contains the path to yaml file.
     k8s_cline: an ApiClient object, initialized with the client args.
+    namespace: string. Defining the namespace where the new objects will be created
 
     Available parameters for performing the subsequent action:
     :param async_req bool
@@ -55,12 +56,18 @@ def create_from_yaml(k8s_client, yaml_file, verbose=False, **kwargs):
         kind = yml_object["kind"]
         kind = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', kind)
         kind = re.sub('([a-z0-9])([A-Z])', r'\1_\2', kind).lower()
-        # Decide which namespace we are going to put the object in,
-        # if any
-        if "namespace" in yml_object["metadata"]:
-            namespace = yml_object["metadata"]["namespace"]
+
+        # If a namespace if defined overwrite it
+        if namespace is not None:
+            yml_object["metadata"]["namespace"] = namespace
         else:
-            namespace = "default"
+            # Decide which namespace we are going to put the object in,
+            # if any
+            if "namespace" in yml_object["metadata"]:
+                namespace = yml_object["metadata"]["namespace"]
+            else:
+                namespace = "default"
+
         # Expect the user to create namespaced objects more often
         if hasattr(k8s_api, "create_namespaced_{0}".format(kind)):
             resp = getattr(k8s_api, "create_namespaced_{0}".format(kind))(
@@ -71,3 +78,20 @@ def create_from_yaml(k8s_client, yaml_file, verbose=False, **kwargs):
         if verbose:
             print("{0} created. status='{1}'".format(kind, str(resp.status)))
         return k8s_api
+
+
+def create_from_yaml(k8s_client, yaml_file, verbose=False, **kwargs):
+    """
+    Perform an action from a yaml file. Pass True for verbose to
+    print confirmation information.
+    Input:
+    yaml_file: string. Contains the path to yaml file.
+    k8s_cline: an ApiClient object, initialized with the client args.
+
+    Available parameters for performing the subsequent action:
+    :param async_req bool
+    :param bool include_uninitialized: If true, partially initialized resources are included in the response.
+    :param str pretty: If 'true', then the output is pretty printed.
+    :param str dry_run: When present, indicates that modifications should not be persisted. An invalid or unrecognized dryRun directive will result in an error response and no further processing of the request. Valid values are: - All: all dry run stages will be processed
+    """
+    return create_namespaced_from_yaml(k8s_client, yaml_file, verbose, None, **kwargs)