aboutsummaryrefslogtreecommitdiffstats
path: root/functest/utils/config.py
diff options
context:
space:
mode:
Diffstat (limited to 'functest/utils/config.py')
-rw-r--r--functest/utils/config.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/functest/utils/config.py b/functest/utils/config.py
new file mode 100644
index 00000000..4cee6349
--- /dev/null
+++ b/functest/utils/config.py
@@ -0,0 +1,30 @@
+import os
+
+import yaml
+
+
+class Config(object):
+ def __init__(self):
+ if 'CONFIG_FUNCTEST_YAML' not in os.environ:
+ raise Exception('CONFIG_FUNCTEST_YAML not configed')
+ self.config_functest = os.environ['CONFIG_FUNCTEST_YAML']
+ try:
+ with open(self.config_functest) as f:
+ self.functest_yaml = yaml.safe_load(f)
+ self.parse(None, self.functest_yaml)
+ except:
+ raise Exception('Parse {} failed'.format(self.config_functest))
+
+ def parse(self, attr_now, left_parametes):
+ for param_n, param_v in left_parametes.iteritems():
+ attr_further = self.get_attr_further(attr_now, param_n)
+ if not isinstance(param_v, dict):
+ self.__setattr__(attr_further, param_v)
+ else:
+ self.parse(attr_further, param_v)
+
+ def get_attr_further(self, attr_now, next):
+ return attr_now if next == 'general' else (
+ '{}_{}'.format(attr_now, next) if attr_now else next)
+
+CONF = Config()