mirror of
https://github.com/httprunner/httprunner.git
synced 2026-07-10 23:12:41 +08:00
Merge branch 'master' into loader
This commit is contained in:
@@ -15,6 +15,8 @@ log_colors_config = {
|
|||||||
'ERROR': 'red',
|
'ERROR': 'red',
|
||||||
'CRITICAL': 'red',
|
'CRITICAL': 'red',
|
||||||
}
|
}
|
||||||
|
logger = logging.getLogger("httprunner")
|
||||||
|
|
||||||
|
|
||||||
def setup_logger(log_level, log_file=None):
|
def setup_logger(log_level, log_file=None):
|
||||||
"""setup root logger with ColoredFormatter."""
|
"""setup root logger with ColoredFormatter."""
|
||||||
@@ -40,24 +42,26 @@ def setup_logger(log_level, log_file=None):
|
|||||||
handler = logging.StreamHandler()
|
handler = logging.StreamHandler()
|
||||||
|
|
||||||
handler.setFormatter(formatter)
|
handler.setFormatter(formatter)
|
||||||
logging.root.addHandler(handler)
|
logger.addHandler(handler)
|
||||||
logging.root.setLevel(level)
|
logger.setLevel(level)
|
||||||
|
|
||||||
|
|
||||||
def coloring(text, color="WHITE"):
|
def coloring(text, color="WHITE"):
|
||||||
fore_color = getattr(Fore, color.upper())
|
fore_color = getattr(Fore, color.upper())
|
||||||
return fore_color + text
|
return fore_color + text
|
||||||
|
|
||||||
|
|
||||||
def color_print(msg, color="WHITE"):
|
def color_print(msg, color="WHITE"):
|
||||||
fore_color = getattr(Fore, color.upper())
|
fore_color = getattr(Fore, color.upper())
|
||||||
print(fore_color + msg)
|
print(fore_color + msg)
|
||||||
|
|
||||||
|
|
||||||
def log_with_color(level):
|
def log_with_color(level):
|
||||||
""" log with color by different level
|
""" log with color by different level
|
||||||
"""
|
"""
|
||||||
def wrapper(text):
|
def wrapper(text):
|
||||||
color = log_colors_config[level.upper()]
|
color = log_colors_config[level.upper()]
|
||||||
getattr(logging, level.lower())(coloring(text, color))
|
getattr(logger, level.lower())(coloring(text, color))
|
||||||
|
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user