functest/utils/env.py

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