replacing all yaml.load with yaml.safe_load

Author: paigerube14

openshift_performance/ci/scripts/increase_pods.py

@@ -19,7 +19,7 @@ def print_new_yaml(num, fileName):
     # append to file
     yaml = YAML()
     with open(fileName, "r") as f:
-        yaml_file = yaml.load(f)
+        yaml_file = yaml.safe_load(f)
         yaml_file['projects'][0]['num'] = num
     with open(fileName, "w+") as f:
         yaml.indent(mapping=2, sequence=4, offset=2)
@@ -30,7 +30,7 @@ def print_new_yaml_temp(num, fileName):
     # append to file
     yaml = YAML()
     with open(fileName, "r") as f:
-        yaml_file = yaml.load(f)
+        yaml_file = yaml.safe_load(f)
         yaml_file['projects'][0]['templates'][0]['num'] = num
     with open(fileName, "w+") as f:
         yaml.indent(mapping=2, sequence=4, offset=2)

openshift_scalability/ci/scripts/get_pod_total.py

@@ -5,7 +5,7 @@
 def get_pod_counts_python(file):
     pod_list = []
     with open(file, "r") as f:
-        yaml_file = yaml.load(f, Loader=yaml.FullLoader)
+        yaml_file = yaml.safe_load(f)
         for proj in yaml_file['projects']:
             for pod in proj['pods']:
                 if "total" in pod.keys():
@@ -17,7 +17,7 @@ def get_pod_counts_golang(file):
 
     pod_list=[]
     with open(file, "r") as f:
-        yaml_file = yaml.load(f, Loader=yaml.FullLoader)
+        yaml_file = yaml.safe_load(f)
         for proj in yaml_file['ClusterLoader']['projects']:
             for pod in proj['pods']:
                 if "num" in pod.keys():

openshift_scalability/cluster-loader.py

@@ -52,7 +52,7 @@
 
 testconfig = {}
 with open(options.cfgfile) as stream:
-    testconfig = yaml.load(stream)
+    testconfig = yaml.safe_load(stream)
 
 globalvars = {}
 globalvars["cleanoption"] = options.cleanall

openshift_scalability/scripts/drainNode/drain_helper.py

@@ -4,7 +4,7 @@
 def print_new_yaml_temp(num, fileName):
     # append to file
     with open(fileName, "r") as f:
-        yaml_file = yaml.load(f, Loader=yaml.FullLoader)
+        yaml_file = yaml.safe_load(f)
         yaml_file['projects'][0]['templates'][0]['num'] = num
     with open(fileName, "w+") as f:
         str_file = yaml.dump(yaml_file)

openshift_scalability/scripts/pod_density/increase_pods.py

@@ -17,7 +17,7 @@ def run(command):
 def print_new_yaml(num, fileName):
     # append to file
     with open(fileName, "r") as f:
-        yaml_file = yaml.load(f, Loader=yaml.FullLoader)
+        yaml_file = yaml.safe_load(f)
         print("yaml file " + str(yaml_file))
         yaml_file['projects'][0]['num'] = num
         print("new yaml file " + str(yaml_file))
@@ -30,7 +30,7 @@ def print_new_yaml(num, fileName):
 def print_new_yaml_temp(num, fileName):
     # append to file
     with open(fileName, "r") as f:
-        yaml_file = yaml.load(f, Loader=yaml.FullLoader)
+        yaml_file = yaml.safe_load(f)
         print("yaml file " + str(yaml_file))
         yaml_file['projects'][0]['templates'][0]['num'] = num
         print("new yaml file " + str(yaml_file))

openshift_scalability/utils.py

@@ -438,7 +438,7 @@ def single_project(testconfig, projname, globalvars):
         if globalvars["kubeopt"]:
             tmpfile=tempfile.NamedTemporaryFile(mode='w')
             with open("content/namespace-default.yaml") as infile:
-                nsconfig = yaml.load(infile)
+                nsconfig = yaml.safe_load(infile)
             nsconfig["metadata"]["name"] = projname
             with open(tmpfile.name, 'w+') as f:
                 yaml.dump(nsconfig, f, default_flow_style=False)
@@ -568,7 +568,7 @@ def quota_handler(inputquota, globalvars):
         quotafile = "content/quota-default.json"
 
     with open(quotafile,'r') as infile:
-        qconfig = json.load(infile)
+        qconfig = json.safe_load(infile)
     qconfig["metadata"]["namespace"] = globalvars["namespace"]
     qconfig["metadata"]["name"] = quota["name"]
     tmpfile=tempfile.NamedTemporaryFile(mode='w')
@@ -619,7 +619,7 @@ def service_handler(inputservs, globalvars):
 
         service_config = {}
         with open(servfile) as stream:
-            service_config = json.load(stream)
+            service_config = json.safe_load(stream)
         service_config["metadata"]["namespace"] = namespace
         service_config["metadata"]["name"] = basename
 
@@ -634,7 +634,7 @@ def ebs_create(globalvars):
     global ebsvolumeid
     ebsvolumeid = ec2_volume(ebsvolumesize,ebsvtype,ebstagprefix,ebsregion)
     with open("content/pv-default.json", "r") as pvstream:
-        pvjson = json.load(pvstream)
+        pvjson = json.safe_load(pvstream)
     pvjson["metadata"]["name"] = ebsvolumeid
     pvjson["spec"]["capacity"]["storage"] = str(ebsvolumesize) + "Gi"  # this has to be like this till k8s 23357 is fixed
     pvjson["spec"]["accessModes"] = [pvpermissions]
@@ -653,7 +653,7 @@ def ebs_create(globalvars):
     pvtmpfile.close()
 
     with open("content/pvc-default.json", "r") as pvcstream:
-        pvcjson = json.load(pvcstream)
+        pvcjson = json.safe_load(pvcstream)
     pvcjson["metadata"]["name"] = ebsvolumeid
     pvcjson["metadata"]["namespace"] = namespace
     pvcjson["spec"]["resources"]["requests"]["storage"] = str(ebsvolumesize) + "Gi"
@@ -673,7 +673,7 @@ def ebs_create(globalvars):
 def ceph_secret_create(cephsecret,globalvars):
     namespace = globalvars["namespace"]
     with open("content/ceph-secret.json") as cephsec:
-        cephsecjson = json.load(cephsec)
+        cephsecjson = json.safe_load(cephsec)
 
     cephsecjson["metadata"]["name"] = cephsecretname
     cephsecjson["metadata"]["namespace"] = namespace
@@ -704,7 +704,7 @@ def ceph_image_create(i,globalvars):
     # ceph_volume function will create ceph images at ceph storage cluster side
     ceph_volume(cephpool,cephimagename,imagesize)
     with open("content/pv-ceph.json") as pvstream:
-        pvjson = json.load(pvstream)
+        pvjson = json.safe_load(pvstream)
 
     pvjson["metadata"]["name"] =  "cephvol" + str(i)
     pvjson["metadata"]["namespace"] = namespace
@@ -730,7 +730,7 @@ def ceph_image_create(i,globalvars):
     pvtmpfile.close()
 
     with open("content/pvc-default.json", "r") as pvcstream:
-        pvcjson = json.load(pvcstream)
+        pvcjson = json.safe_load(pvcstream)
     pvcjson["metadata"]["name"] = "cephclaim" + str(i)
     pvcjson["metadata"]["namespace"] = namespace
     pvcjson["spec"]["resources"]["requests"]["storage"] = str(cephimagesize) + "Gi"
@@ -839,7 +839,7 @@ def pod_handler(inputpods, globalvars):
 
         pod_config = {}
         with open(podfile) as stream:
-            pod_config = json.load(stream)
+            pod_config = json.safe_load(stream)
         pod_config["metadata"]["namespace"] = namespace
         pod_config["metadata"]["name"] = basename
 
@@ -872,7 +872,7 @@ def rc_handler(inputrcs, globalvars):
 
         rc_config = {}
         with open(rcfile) as stream:
-            rc_config = json.load(stream)
+            rc_config = json.safe_load(stream)
         rc_config["metadata"]["namespace"] = namespace
         rc_config["metadata"]["name"] = basename
         rc_config["spec"]["replicas"] = replicas

reliability/test.py

@@ -2,5 +2,5 @@
 
 print("starting...")
 stream = open("./config/simple_reliability.yaml","r")
-y = yaml.load(stream)
+y = yaml.safe_load(stream)
 print(y['reliability']['appTemplates'][3]['template'])

shutdown/start_clouds/az_node_scenarios.py

@@ -12,7 +12,7 @@ def __init__(self, az_account):
         credentials = DefaultAzureCredential()
         logging.info("credential " + str(credentials))
 
-        az_account_yaml = yaml.load(az_account, Loader=yaml.FullLoader)
+        az_account_yaml = yaml.safe_load(az_account)
         subscription_id = az_account_yaml[0]['id']
         self.compute_client = ComputeManagementClient(credentials, subscription_id)
 

storage/mongodb/rewrite_external_vars.py

@@ -4,7 +4,7 @@
 def replace_value_in_file(fileName, key,value):
     # append to file
     with open(fileName, "r") as f:
-        yaml_file = yaml.load(f, Loader=yaml.FullLoader)
+        yaml_file = yaml.safe_load(f)
         yaml_file[key] = value
     with open(fileName, "w+") as f:
         f.write('---\n')