Replace PYyaml load function to safe_load

selfuryon · selfuryon · commit 66067a4c6081 · 2019-05-08T02:22:21.000+05:00
diff --git a/examples/arista_eos.py b/examples/arista_eos.py
@@ -23,8 +23,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'arista_eos']
     await asyncio.wait(tasks)
 
diff --git a/examples/aruba_aos_6.py b/examples/aruba_aos_6.py
@@ -26,8 +26,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'aruba_aos_6']
     await asyncio.wait(tasks)
 
diff --git a/examples/aruba_aos_8.py b/examples/aruba_aos_8.py
@@ -26,8 +26,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'aruba_aos_8']
     await asyncio.wait(tasks)
 
diff --git a/examples/cisco_ios.py b/examples/cisco_ios.py
@@ -31,8 +31,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'cisco_ios']
     await asyncio.wait(tasks)
 
diff --git a/examples/cisco_iosxr.py b/examples/cisco_iosxr.py
@@ -35,8 +35,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'cisco_ios_xr']
     await asyncio.gather(*tasks)
 
diff --git a/examples/cisco_nxos.py b/examples/cisco_nxos.py
@@ -19,8 +19,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'cisco_nxos']
     await asyncio.wait(tasks)
 
diff --git a/examples/fujitsu_switch.py b/examples/fujitsu_switch.py
@@ -21,8 +21,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'fujitsu_switch']
     await asyncio.wait(tasks)
 
diff --git a/examples/hp_comware.py b/examples/hp_comware.py
@@ -26,8 +26,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'hp_comware']
     await asyncio.wait(tasks)
 
diff --git a/examples/hp_comware_limited.py b/examples/hp_comware_limited.py
@@ -26,8 +26,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'hp_comware_limited']
     await asyncio.wait(tasks)
 
diff --git a/examples/juniper_junos.py b/examples/juniper_junos.py
@@ -34,8 +34,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'juniper_junos']
     await asyncio.wait(tasks)
 
diff --git a/examples/mikrotik_routeros.py b/examples/mikrotik_routeros.py
@@ -24,8 +24,8 @@ async def task(param):
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'mikrotik_routeros']
     await asyncio.wait(tasks)
 
diff --git a/examples/terminal.py b/examples/terminal.py
@@ -12,19 +12,19 @@
 
 
 async def task(param):
-    async with netdev.create(**param) as arista:
+    async with netdev.create(**param) as term:
         # Testing sending simple command
-        out = await arista.send_command('ls -al', strip_command=True)
+        out = await term.send_command('ls -al', strip_command=True)
         print(out)
-        out = await arista.send_command('pwd', strip_command=True)
+        out = await term.send_command('pwd', strip_command=True)
         print(out)
-        out = await arista.send_command('echo test', strip_command=True)
+        out = await term.send_command('echo test', strip_command=True)
         print(out)
 
 
 async def run():
-    config = yaml.load(open(config_path, 'r'))
-    devices = yaml.load(open(config['device_list'], 'r'))
+    config = yaml.safe_load(open(config_path, 'r'))
+    devices = yaml.safe_load(open(config['device_list'], 'r'))
     tasks = [task(dev) for dev in devices if dev['device_type'] == 'terminal']
     await asyncio.wait(tasks)
 
diff --git a/tests/test_arista_eos.py b/tests/test_arista_eos.py
@@ -14,9 +14,9 @@ class TestArista(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'arista_eos']
                 return params
 
diff --git a/tests/test_aruba_aos_6.py b/tests/test_aruba_aos_6.py
@@ -14,9 +14,9 @@ class TestAOS6(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'aruba_aos_6']
                 return params
 
diff --git a/tests/test_aruba_aos_8.py b/tests/test_aruba_aos_8.py
@@ -14,9 +14,9 @@ class TestAOS8(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'aruba_aos_8']
                 return params
 
diff --git a/tests/test_cisco_asa.py b/tests/test_cisco_asa.py
@@ -14,9 +14,9 @@ class TestASA(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'cisco_asa']
                 return params
 
diff --git a/tests/test_cisco_ios.py b/tests/test_cisco_ios.py
@@ -14,9 +14,9 @@ class TestIOS(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'cisco_ios']
                 return params
 
diff --git a/tests/test_cisco_iosxr.py b/tests/test_cisco_iosxr.py
@@ -14,9 +14,9 @@ class TestIOSXR(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'cisco_ios_xr']
                 return params
 
diff --git a/tests/test_cisco_nxos.py b/tests/test_cisco_nxos.py
@@ -15,9 +15,9 @@ class TestNXOS(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'cisco_nxos']
                 return params
 
diff --git a/tests/test_fujitsu_switch.py b/tests/test_fujitsu_switch.py
@@ -14,9 +14,9 @@ class TestFujitsu(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'fujitsu_switch']
                 return params
 
diff --git a/tests/test_hp_comware.py b/tests/test_hp_comware.py
@@ -14,9 +14,9 @@ class TestComware(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'hp_comware']
                 return params
 
diff --git a/tests/test_hp_comware_limited.py b/tests/test_hp_comware_limited.py
@@ -14,9 +14,9 @@ class TestComwareLimited(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'hp_comware_limited']
                 return params
 
diff --git a/tests/test_juniper_junos.py b/tests/test_juniper_junos.py
@@ -14,9 +14,9 @@ class TestJunOS(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'juniper_junos']
                 return params
 
diff --git a/tests/test_mikrotik_routeros.py b/tests/test_mikrotik_routeros.py
@@ -14,9 +14,9 @@ class TestRouterOS(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'mikrotik_routeros']
                 return params
 
diff --git a/tests/test_terminal.py b/tests/test_terminal.py
@@ -14,9 +14,9 @@ class TestTerminal(unittest.TestCase):
     @staticmethod
     def load_credits():
         with open(config_path, 'r') as conf:
-            config = yaml.load(conf)
+            config = yaml.safe_load(conf)
             with open(config['device_list'], 'r') as devs:
-                devices = yaml.load(devs)
+                devices = yaml.safe_load(devs)
                 params = [p for p in devices if p['device_type'] == 'terminal']
                 return params
 
