40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
from app.core.config import settings
|
|
|
|
# logger
|
|
logger = logging.getLogger()
|
|
if settings.DEBUG:
|
|
logger.setLevel(logging.DEBUG)
|
|
else:
|
|
logger.setLevel(logging.INFO)
|
|
|
|
# 创建终端输出Handler
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(logging.DEBUG)
|
|
|
|
# 创建文件输出Handler
|
|
file_handler = RotatingFileHandler(filename=settings.LOG_PATH / 'moviepilot.log',
|
|
mode='w',
|
|
maxBytes=5 * 1024 * 1024,
|
|
backupCount=3,
|
|
encoding='utf-8')
|
|
file_handler.setLevel(logging.INFO)
|
|
|
|
|
|
# 定义日志输出格式
|
|
class CustomFormatter(logging.Formatter):
|
|
def format(self, record):
|
|
record.levelname = record.levelname + ':'
|
|
return super().format(record)
|
|
|
|
|
|
formatter = CustomFormatter("%(levelname)-10s%(asctime)s - %(filename)s - %(message)s")
|
|
console_handler.setFormatter(formatter)
|
|
file_handler.setFormatter(formatter)
|
|
|
|
# 将Handler添加到Logger
|
|
logger.addHandler(console_handler)
|
|
logger.addHandler(file_handler)
|