move to ruamel.yaml

nornir/core/configuration.py

@@ -2,7 +2,7 @@
 import os
 
 
-import yaml
+import ruamel.yaml
 
 
 CONF = {
@@ -93,7 +93,7 @@ class Config(object):
     def __init__(self, config_file=None, **kwargs):
         if config_file:
             with open(config_file, "r") as f:
-                data = yaml.load(f.read()) or {}
+                data = ruamel.yaml.safe_load(f.read()) or {}
         else:
             data = {}
 

nornir/plugins/inventory/simple.py

@@ -4,7 +4,7 @@
 
 from nornir.core.inventory import Inventory
 
-import yaml
+import ruamel.yaml
 
 
 class SimpleInventory(Inventory):
@@ -118,12 +118,12 @@ class SimpleInventory(Inventory):
 
     def __init__(self, host_file="hosts.yaml", group_file="groups.yaml", **kwargs):
         with open(host_file, "r") as f:
-            hosts = yaml.load(f.read())
+            hosts = ruamel.yaml.safe_load(f.read())
 
         if group_file:
             if os.path.exists(group_file):
                 with open(group_file, "r") as f:
-                    groups = yaml.load(f.read())
+                    groups = ruamel.yaml.safe_load(f.read())
             else:
                 logging.warning("{}: doesn't exist".format(group_file))
                 groups = {}

requirements.txt

@@ -1,5 +1,4 @@
 colorama
-pyyaml
 jinja2
 napalm>=2.3.0
 netmiko>=2.1.1

tests/core/test_configuration/config.yaml

@@ -1,6 +1,6 @@
 ---
 num_workers: 10
-raise_on_error: no
+raise_on_error: false
 user_defined: "asdasd"
 my_root:
     user_defined: "i am nested"