Merge "dovetail tool: add debug option to control the log level showing in screen"

2 files changed, 21 insertions, 16 deletions

diff --git a/dovetail/conf/cmd_config.yml b/dovetail/conf/cmd_config.yml
index 4e3d0110..35c000ef 100644
--- a/dovetail/conf/cmd_config.yml
+++ b/dovetail/conf/cmd_config.yml
@@ -31,11 +31,11 @@ cli:
           - '--DEPLOY_TYPE'
           - '-T'
         help: 'DEPLOY_TYPE of the system under test (SUT).'
-      DEBUG:
+      CON_DEBUG:
         flags:
-          - '--DEBUG'
-          - '-d'
-        help: 'DEBUG for showing debug log.'
+          - '--CON_DEBUG'
+          - '-c'
+        help: 'True for showing debug log in functest/yardstick container.'
     non-envs:
       testsuite:
         flags:
@@ -52,3 +52,8 @@ cli:
           - '--tag'
           - '-o'
         help: 'Overwrite tags for each docker container (e.g. "functest:stable,yardstick:latest")'
+      debug:
+        flags:
+          - '--debug'
+          - '-d'
+        help: 'True for showing debug log in screen.'
diff --git a/dovetail/conf/dovetail_config.py b/dovetail/conf/dovetail_config.py
index a5669bda..d812c5c9 100644
--- a/dovetail/conf/dovetail_config.py
+++ b/dovetail/conf/dovetail_config.py
@@ -16,6 +16,12 @@ class DovetailConfig:
 
     dovetail_config = {}
 
+    CMD_NAME_TRANS = {
+        'SUT_TYPE': 'INSTALLER_TYPE',
+        'SUT_IP': 'INSTALLER_IP',
+        'CON_DEBUG': 'CI_DEBUG',
+    }
+
     @classmethod
     def load_config_files(cls):
         curr_path = os.path.dirname(os.path.abspath(__file__))
@@ -35,11 +41,7 @@ class DovetailConfig:
     @classmethod
     def cmd_name_trans(cls, cmd_name):
         key = cmd_name.upper()
-        if key == 'SUT_TYPE':
-            key = 'INSTALLER_TYPE'
-        if key == 'SUT_IP':
-            key = 'INSTALLER_IP'
-        return key
+        return cls.CMD_NAME_TRANS.get(key, key)
 
     @classmethod
     def update_envs(cls, options):
@@ -48,18 +50,16 @@ class DovetailConfig:
             if not options[item] and key in os.environ:
                 options[item] = os.environ[key]
             if options[item]:
-                cls.update_config_envs('functest', key)
-                cls.update_config_envs('yardstick', key)
+                cls.update_config_envs('functest', key, options[item])
+                cls.update_config_envs('yardstick', key, options[item])
 
     @classmethod
-    def update_config_envs(cls, script_type, key):
-        if key == 'DEBUG':
-            os.environ['CI_DEBUG'] = os.environ[key]
+    def update_config_envs(cls, script_type, key, value):
         envs = cls.dovetail_config[script_type]['envs']
         old_value = re.findall(r'\s+%s=(.*?)(\s+|$)' % key, envs)
         if old_value == []:
-            envs += ' -e ' + key + '=' + os.environ[key]
+            envs += ' -e ' + key + '=' + value
         else:
-            envs = envs.replace(old_value[0][0], os.environ[key])
+            envs = envs.replace(old_value[0][0], value)
         cls.dovetail_config[script_type]['envs'] = envs
         return envs