chore: ruff的修改

This commit is contained in:
AL76
2025-03-13 03:18:49 +08:00
parent 91fceb0d85
commit 11e8b2fa5f
3 changed files with 11 additions and 22 deletions

20
bot.py
View File

@@ -105,13 +105,11 @@ def load_logger():
current_env = os.getenv("ENV", "dev") current_env = os.getenv("ENV", "dev")
# 公共配置参数 # 公共配置参数
log_level = os.getenv( log_level = os.getenv("LOG_LEVEL", "INFO" if current_env == "prod" else "DEBUG")
"LOG_LEVEL",
"INFO" if current_env == "prod" else "DEBUG"
)
log_filter = lambda record: ( log_filter = lambda record: (
("nonebot" not in record["name"] or record["level"].no >= logger.level("ERROR").no ("nonebot" not in record["name"] or record["level"].no >= logger.level("ERROR").no)
) if current_env == "prod" else True if current_env == "prod"
else True
) )
log_format = ( log_format = (
"<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> " "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> "
@@ -129,17 +127,11 @@ def load_logger():
colorize=False, colorize=False,
level=log_level, level=log_level,
filter=log_filter, filter=log_filter,
encoding="utf-8" encoding="utf-8",
) )
# 终端输出 # 终端输出
logger.add( logger.add(sys.stderr, format=log_format, colorize=True, level=log_level, filter=log_filter)
sys.stderr,
format=log_format,
colorize=True,
level=log_level,
filter=log_filter
)
def scan_provider(env_config: dict): def scan_provider(env_config: dict):

View File

@@ -22,9 +22,7 @@ def __create_database_instance():
if username and password: if username and password:
# 如果有用户名和密码,使用认证连接 # 如果有用户名和密码,使用认证连接
return MongoClient( return MongoClient(host, port, username=username, password=password, authSource=auth_source)
host, port, username=username, password=password, authSource=auth_source
)
# 否则使用无认证连接 # 否则使用无认证连接
return MongoClient(host, port) return MongoClient(host, port)

View File

@@ -4,7 +4,7 @@ import os
from loguru import logger from loguru import logger
from nonebot import get_driver, on_message, require from nonebot import get_driver, on_message, require
from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message, MessageSegment,MessageEvent from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message, MessageSegment, MessageEvent
from nonebot.typing import T_State from nonebot.typing import T_State
from ..moods.moods import MoodManager # 导入情绪管理器 from ..moods.moods import MoodManager # 导入情绪管理器
@@ -99,15 +99,14 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
@scheduler.scheduled_job("interval", seconds=global_config.build_memory_interval, id="build_memory") @scheduler.scheduled_job("interval", seconds=global_config.build_memory_interval, id="build_memory")
async def build_memory_task(): async def build_memory_task():
"""每build_memory_interval秒执行一次记忆构建""" """每build_memory_interval秒执行一次记忆构建"""
logger.debug( logger.debug("[记忆构建]------------------------------------开始构建记忆--------------------------------------")
"[记忆构建]"
"------------------------------------开始构建记忆--------------------------------------")
start_time = time.time() start_time = time.time()
await hippocampus.operation_build_memory(chat_size=20) await hippocampus.operation_build_memory(chat_size=20)
end_time = time.time() end_time = time.time()
logger.success( logger.success(
f"[记忆构建]--------------------------记忆构建完成:耗时: {end_time - start_time:.2f} " f"[记忆构建]--------------------------记忆构建完成:耗时: {end_time - start_time:.2f} "
"秒-------------------------------------------") "秒-------------------------------------------"
)
@scheduler.scheduled_job("interval", seconds=global_config.forget_memory_interval, id="forget_memory") @scheduler.scheduled_job("interval", seconds=global_config.forget_memory_interval, id="forget_memory")