Refactor code structure for improved readability and maintainability
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
from pkg.log import log
|
||||
from loguru import logger as log
|
||||
|
||||
load_dotenv()
|
||||
|
||||
|
||||
@@ -1,2 +0,0 @@
|
||||
from .log_handle import logger as log
|
||||
from .log_handle import set_log_level
|
||||
@@ -1,34 +0,0 @@
|
||||
import logging
|
||||
from rich.logging import RichHandler
|
||||
|
||||
FOTMAT = "%(message)s"
|
||||
logging.basicConfig(
|
||||
level="NOTSET",
|
||||
format=FOTMAT,
|
||||
datefmt="[%X]",
|
||||
handlers=[RichHandler(rich_tracebacks=True)],
|
||||
)
|
||||
|
||||
logger = logging.getLogger("rich")
|
||||
|
||||
def set_log_level(level: str):
|
||||
"""
|
||||
设置日志等级。
|
||||
|
||||
:param level: 日志等级 (DEBUG, INFO, WARNING, ERROR, CRITICAL)
|
||||
:type level: str
|
||||
"""
|
||||
level = level.upper()
|
||||
if level == "DEBUG":
|
||||
logger.setLevel(logging.DEBUG)
|
||||
elif level == "INFO":
|
||||
logger.setLevel(logging.INFO)
|
||||
elif level == "WARNING":
|
||||
logger.setLevel(logging.WARNING)
|
||||
elif level == "ERROR":
|
||||
logger.setLevel(logging.ERROR)
|
||||
elif level == "CRITICAL":
|
||||
logger.setLevel(logging.CRITICAL)
|
||||
else:
|
||||
logger.setLevel(logging.INFO)
|
||||
logger.warning(f"未知的日志等级 '{level}',已设置为默认等级 'INFO'。")
|
||||
Reference in New Issue
Block a user