diff options
author | ahothan <ahothan@cisco.com> | 2017-07-28 17:08:46 -0700 |
---|---|---|
committer | ahothan <ahothan@cisco.com> | 2017-07-31 12:34:00 -0700 |
commit | 04a7de082bd221eae3c7004f4e0b99dfa4f8be91 (patch) | |
tree | c9fb7beaedc80479772ba24c3b47c85d49c22f76 /nfvbench/config.py | |
parent | b8f02ed4e72399840a93aceb02b8c53831bbe68a (diff) |
Initial code drop from Cisco1.0.0
Change-Id: Ie2993886dc8e95c5f73ccdb871add8b96ffcc849
Signed-off-by: ahothan <ahothan@cisco.com>
Diffstat (limited to 'nfvbench/config.py')
-rw-r--r-- | nfvbench/config.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/nfvbench/config.py b/nfvbench/config.py new file mode 100644 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('#') |