From 5f063b6603e87b78d10369882289cc2433a506b2 Mon Sep 17 00:00:00 2001 From: debugtalk Date: Thu, 9 Aug 2018 20:12:01 +0800 Subject: [PATCH] replace root logger with httprunner logger --- httprunner/logger.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/httprunner/logger.py b/httprunner/logger.py index 68d0f7db..9fe841b1 100644 --- a/httprunner/logger.py +++ b/httprunner/logger.py @@ -15,6 +15,8 @@ log_colors_config = { 'ERROR': 'red', 'CRITICAL': 'red', } +logger = logging.getLogger("httprunner") + def setup_logger(log_level, log_file=None): """setup root logger with ColoredFormatter.""" @@ -40,24 +42,26 @@ def setup_logger(log_level, log_file=None): handler = logging.StreamHandler() handler.setFormatter(formatter) - logging.root.addHandler(handler) - logging.root.setLevel(level) + logger.addHandler(handler) + logger.setLevel(level) def coloring(text, color="WHITE"): fore_color = getattr(Fore, color.upper()) return fore_color + text + def color_print(msg, color="WHITE"): fore_color = getattr(Fore, color.upper()) print(fore_color + msg) + def log_with_color(level): """ log with color by different level """ def wrapper(text): color = log_colors_config[level.upper()] - getattr(logging, level.lower())(coloring(text, color)) + getattr(logger, level.lower())(coloring(text, color)) return wrapper