functest/utils/env.py

   1 #!/usr/bin/env python
   2
   3 import pkg_resources
   4 import os
   5 import re
   6
   7 import six
   8
   9
  10 default_envs = {
  11     'NODE_NAME': 'unknown_pod',
  12     'CI_DEBUG': 'false',
  13     'DEPLOY_SCENARIO': 'os-nosdn-nofeature-noha',
  14     'DEPLOY_TYPE': 'virt',
  15     'INSTALLER_TYPE': None,
  16     'INSTALLER_IP': None,
  17     'BUILD_TAG': None,
  18     'OS_ENDPOINT_TYPE': None,
  19     'OS_AUTH_URL': None,
  20     'CONFIG_FUNCTEST_YAML': pkg_resources.resource_filename(
  21         'functest', 'ci/config_functest.yaml')
  22 }
  23
  24
  25 class Environment(object):
  26
  27     def __init__(self):
  28         for k, v in six.iteritems(os.environ):
  29             self.__setattr__(k, v)
  30         for k, v in six.iteritems(default_envs):
  31             if k not in os.environ:
  32                 self.__setattr__(k, v)
  33         self._set_ci_run()
  34         self._set_ci_loop()
  35
  36     def _set_ci_run(self):
  37         if self.BUILD_TAG:
  38             self.IS_CI_RUN = True
  39         else:
  40             self.IS_CI_RUN = False
  41
  42     def _set_ci_loop(self):
  43         if self.BUILD_TAG and re.search("daily", self.BUILD_TAG):
  44             self.CI_LOOP = "daily"
  45         else:
  46             self.CI_LOOP = "weekly"
  47
  48
  49 ENV = Environment()