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

View File

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

View File

@@ -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")
async def build_memory_task():
"""每build_memory_interval秒执行一次记忆构建"""
logger.debug(
"[记忆构建]"
"------------------------------------开始构建记忆--------------------------------------")
logger.debug("[记忆构建]------------------------------------开始构建记忆--------------------------------------")
start_time = time.time()
await hippocampus.operation_build_memory(chat_size=20)
end_time = time.time()
logger.success(
f"[记忆构建]--------------------------记忆构建完成:耗时: {end_time - start_time:.2f} "
"秒-------------------------------------------")
"秒-------------------------------------------"
)
@scheduler.scheduled_job("interval", seconds=global_config.forget_memory_interval, id="forget_memory")