From 3e664565353ac0a09636157c2dc38895ac8898a9 Mon Sep 17 00:00:00 2001 From: BoYanZh Date: Fri, 27 May 2022 12:42:32 +0800 Subject: [PATCH] feat: stderr log level in config (#12) --- joint_teapot/config.py | 1 + joint_teapot/utils/logger.py | 11 ++++++++--- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/joint_teapot/config.py b/joint_teapot/config.py index 912d7c6..5d829ed 100644 --- a/joint_teapot/config.py +++ b/joint_teapot/config.py @@ -34,6 +34,7 @@ class Settings(BaseSettings): # log file log_file_path: str = "joint-teapot.log" + stderr_log_level: str = "INFO" class Config: env_file = ".env" diff --git a/joint_teapot/utils/logger.py b/joint_teapot/utils/logger.py index db8d3d2..d489f3c 100644 --- a/joint_teapot/utils/logger.py +++ b/joint_teapot/utils/logger.py @@ -4,6 +4,11 @@ from loguru import logger as logger from joint_teapot.config import settings -logger.remove() -logger.add(stderr, level="INFO") -logger.add(settings.log_file_path, level="INFO") + +def set_logger(stderr_log_level: str = settings.stderr_log_level) -> None: + logger.remove() + logger.add(stderr, level=stderr_log_level) + logger.add(settings.log_file_path, level="DEBUG") + + +set_logger()