Merge "Unified way to provide configurations and env variables(proposal 1)"

3 files changed, 68 insertions, 0 deletions

diff --git a/functest/utils/config.py b/functest/utils/config.py
new file mode 100644
index 000000000..4cee63494
--- /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()
diff --git a/functest/utils/constants.py b/functest/utils/constants.py
new file mode 100644
index 000000000..2e8eb3f44
--- /dev/null
+++ b/functest/utils/constants.py
@@ -0,0 +1,20 @@
+import config
+import env
+
+
+class Constants(object):
+    def __init__(self):
+        for attr_n, attr_v in config.CONF.__dict__.iteritems():
+            self.__setattr__(attr_n, attr_v)
+        for env_n, env_v in env.ENV.__dict__.iteritems():
+            self.__setattr__(env_n, env_v)
+
+
+CONST = Constants()
+
+if __name__ == '__main__':
+    print CONST.__dict__
+    print CONST.NODE_NAME
+    print CONST.vIMS_clearwater_blueprint_url
+    print CONST.vIMS_clearwater_blueprint_file_name
+    print CONST.vIMS_clearwater_blueprint_name
diff --git a/functest/utils/env.py b/functest/utils/env.py
new file mode 100644
index 000000000..b6af767da
--- /dev/null
+++ b/functest/utils/env.py
@@ -0,0 +1,18 @@
+import os
+
+default_envs = {
+    'NODE_NAME': 'unknown_pod',
+    'CI_DEBUG': 'true'
+}
+
+
+class Environment(object):
+    def __init__(self):
+        for k, v in os.environ.iteritems():
+            self.__setattr__(k, v)
+        for k, v in default_envs.iteritems():
+            if k not in os.environ:
+                self.__setattr__(k, v)
+
+
+ENV = Environment()