From 93a64634f99a64009fe762fb69e774e3524cc9be Mon Sep 17 00:00:00 2001
From: helenyao <yaohelan@huawei.com>
Date: Fri, 10 Feb 2017 03:52:58 -0500
Subject: Put the Logger as Object oriented.

There is no need to worry about adding duplicate handler as the logging
has the ability to avoid it.

Change-Id: I7198c1e1c05df347feb0d7192e80592e662314b1
Signed-off-by: helenyao <yaohelan@huawei.com>
---
 functest/utils/functest_logger.py | 58 ++++++++++++++++++---------------------
 1 file changed, 26 insertions(+), 32 deletions(-)

(limited to 'functest/utils')

diff --git a/functest/utils/functest_logger.py b/functest/utils/functest_logger.py
index f09f56be..0cba8c52 100755
--- a/functest/utils/functest_logger.py
+++ b/functest/utils/functest_logger.py
@@ -28,42 +28,36 @@ import json
 
 from functest.utils.constants import CONST
 
-logger = logging.getLogger(__name__)
-
-
-def is_debug():
-    if CONST.CI_DEBUG and CONST.CI_DEBUG.lower() == "true":
-        return True
-    return False
-
-
-def setup_logging(default_path=CONST.dir_functest_logging_cfg,
-                  default_level=logging.INFO,
-                  env_key='LOG_CFG'):
-    path = default_path
-    value = os.getenv(env_key, None)
-    if value:
-        path = value
-    if os.path.exists(path):
-        with open(path, 'rt') as f:
-            config = json.load(f)
-            if (config['handlers'] and
-                    config['handlers']['console']):
-                stream_level = logging.INFO
-                if is_debug():
-                    stream_level = logging.DEBUG
-                config['handlers']['console']['level'] = stream_level
-            logging.config.dictConfig(config)
-    else:
-        logging.basicConfig(level=default_level)
-
-
-setup_logging()
-
 
 class Logger:
     def __init__(self, logger_name):
+        self.setup_logging()
         self.logger = logging.getLogger(logger_name)
 
     def getLogger(self):
         return self.logger
+
+    def is_debug(self):
+        if CONST.CI_DEBUG and CONST.CI_DEBUG.lower() == "true":
+            return True
+        return False
+
+    def setup_logging(self, default_path=CONST.dir_functest_logging_cfg,
+                      default_level=logging.INFO,
+                      env_key='LOG_CFG'):
+        path = default_path
+        value = os.getenv(env_key, None)
+        if value:
+            path = value
+        if os.path.exists(path):
+            with open(path, 'rt') as f:
+                config = json.load(f)
+                if (config['handlers'] and
+                        config['handlers']['console']):
+                    stream_level = logging.INFO
+                    if self.is_debug():
+                        stream_level = logging.DEBUG
+                    config['handlers']['console']['level'] = stream_level
+                logging.config.dictConfig(config)
+        else:
+            logging.basicConfig(level=default_level)
-- 
cgit 1.2.3-korg