fix: set log level for pytest

This commit is contained in:
debugtalk
2022-06-17 21:01:13 +08:00
parent 39fdff343d
commit e72a9d8d4c
5 changed files with 23 additions and 9 deletions

View File

@@ -323,7 +323,11 @@ def gen_cartesian_product(*args: List[Dict]) -> List[Dict]:
LOGGER_FORMAT = "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level}</level> | <level>{message}</level>"
def init_logger():
def init_logger(level: str):
level = level.upper()
if level not in ["TRACE", "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]:
level = "INFO" # default
# set log level to INFO
logger.remove()
logger.add(sys.stderr, format=LOGGER_FORMAT, level="INFO")
logger.add(sys.stdout, format=LOGGER_FORMAT, level=level)