Use yaml.safe_load

ramnes · ramnes · commit 3240b45c46b1 · 2019-01-09T10:10:48.000+01:00
See kubernetes-client#718 for rationale.
diff --git a/examples/create_deployment.py b/examples/create_deployment.py
@@ -26,7 +26,7 @@ def main():
     config.load_kube_config()
 
     with open(path.join(path.dirname(__file__), "nginx-deployment.yaml")) as f:
-        dep = yaml.load(f)
+        dep = yaml.safe_load(f)
         k8s_beta = client.ExtensionsV1beta1Api()
         resp = k8s_beta.create_namespaced_deployment(
             body=dep, namespace="default")
diff --git a/kubernetes/e2e_test/test_extensions.py b/kubernetes/e2e_test/test_extensions.py
@@ -50,7 +50,7 @@ def test_create_deployment(self):
         - containerPort: 80
 '''
         resp = api.create_namespaced_deployment(
-            body=yaml.load(deployment % name),
+            body=yaml.safe_load(deployment % name),
             namespace="default")
         resp = api.read_namespaced_deployment(name, 'default')
         self.assertIsNotNone(resp)
