functest/utils/env.py

   1 import os
   2 import re
   3
   4 default_envs = {
   5     'NODE_NAME': 'unknown_pod',
   6     'CI_DEBUG': 'true',
   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 }
  14
  15
  16 class Environment(object):
  17
  18     def __init__(self):
  19         for k, v in os.environ.iteritems():
  20             self.__setattr__(k, v)
  21         for k, v in default_envs.iteritems():
  22             if k not in os.environ:
  23                 self.__setattr__(k, v)
  24         self._set_ci_run()
  25         self._set_ci_loop()
  26
  27     def _set_ci_run(self):
  28         if self.BUILD_TAG:
  29             self.IS_CI_RUN = True
  30         else:
  31             self.IS_CI_RUN = False
  32
  33     def _set_ci_loop(self):
  34         if self.BUILD_TAG and re.search("daily", self.BUILD_TAG):
  35             self.CI_LOOP = "daily"
  36         else:
  37             self.CI_LOOP = "weekly"
  38
  39
  40 ENV = Environment()