Initial code drop from Cisco

[nfvbench.git] / nfvbench / config.py

diff --git a/nfvbench/config.py b/nfvbench/config.py
new file mode 100644 (file)
index 0000000..b2972dd
--- /dev/null
+++ b/nfvbench/config.py
@@ -0,0 +1,56 @@
+# Copyright 2016 Cisco Systems, Inc.  All rights reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+#
+
+from attrdict import AttrDict
+import yaml
+
+
+def config_load(file_name, from_cfg=None):
+    """Load a yaml file into a config dict, merge with from_cfg if not None
+    The config file content taking precedence in case of duplicate
+    """
+    try:
+        with open(file_name) as fileobj:
+            cfg = AttrDict(yaml.safe_load(fileobj))
+    except IOError:
+        raise Exception("Configuration file at '{}' was not found. Please use correct path "
+                        "and verify it is visible to container if you run nfvbench in container."
+                        .format(file_name))
+
+    if from_cfg:
+        cfg = from_cfg + cfg
+
+    return cfg
+
+
+def config_loads(cfg_text, from_cfg=None):
+    """Same as config_load but load from a string
+    """
+    try:
+        cfg = AttrDict(yaml.load(cfg_text))
+    except TypeError:
+        # empty string
+        cfg = AttrDict()
+    if from_cfg:
+        return from_cfg + cfg
+    return cfg
+
+
+def test_config():
+    cfg = config_load('a1.yaml')
+    cfg = config_load('a2.yaml', cfg)
+    cfg = config_loads('color: 500', cfg)
+    config_loads('')
+    config_loads('#')