diff options
-rw-r--r-- | functest/utils/config.py | 30 | ||||
-rw-r--r-- | functest/utils/constants.py | 20 | ||||
-rw-r--r-- | functest/utils/env.py | 18 |
3 files changed, 68 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() diff --git a/functest/utils/constants.py b/functest/utils/constants.py new file mode 100644 index 00000000..2e8eb3f4 --- /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 00000000..b6af767d --- /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() |