arthurcai/Joint-Teapot
2
0
Fork 0
forked from JOJ/Joint-Teapot
Code Pull Requests Activity

feat: remove unused log part

Browse Source
This commit is contained in:
张泊明518370910136 2024-11-30 07:24:17 -05:00
parent 064802496c
commit 7efd5259c9
Signed by untrusted user: 张泊明518370910136
GPG Key ID: CA088E6D9284F870
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
joint_teapot/utils/logger.py
View File

@ -11,10 +11,8 @@ from joint_teapot.config import settings
# recipe from https://loguru.readthedocs.io/en/stable/overview.html#entirely-compatible-with-standard-logging # recipe from https://loguru.readthedocs.io/en/stable/overview.html#entirely-compatible-with-standard-logging
class InterceptHandler(logging.Handler): class InterceptHandler(logging.Handler):
def __init__(self, diagnose: bool = True, backtrace: bool = True): def __init__(self) -> None:
super().__init__() super().__init__()
self.diagnose = diagnose
self.backtrace = backtrace
def emit(self, record: logging.LogRecord) -> None: def emit(self, record: logging.LogRecord) -> None:
try: try:
@ -30,25 +28,19 @@ class InterceptHandler(logging.Handler):
depth += 1 depth += 1
logger.opt(depth=depth, exception=record.exc_info).log( logger.opt(depth=depth, exception=record.exc_info).log(
level, record.getMessage(), diagnose=self.diagnose, backtrace=self.backtrace level,
record.getMessage(),
) )
def set_logger( def set_logger(
stderr_log_level: str = settings.stderr_log_level, stderr_log_level: str = settings.stderr_log_level,
*,
diagnose: bool = True,
backtrace: bool = True,
) -> None: ) -> None:
logging.basicConfig( logging.basicConfig(handlers=[InterceptHandler()], level=0, force=True)
handlers=[InterceptHandler(diagnose, backtrace)], level=0, force=True
)
logger.remove() logger.remove()
logger.add( logger.add(
stderr, stderr,
level=stderr_log_level, level=stderr_log_level,
diagnose=diagnose,
backtrace=backtrace,
) )
logger.add(settings.log_file_path, level="DEBUG") logger.add(settings.log_file_path, level="DEBUG")