gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: 格式化代码

Browse Source
This commit is contained in:
John Richard
2025-10-02 19:38:39 +08:00
committed by Windpicker-owo
parent e7aaafde2f
commit 00ba07e0e1
111 changed files with 2343 additions and 2316 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
src/chat/message_receive/storage.py
View File

@@ -53,7 +53,11 @@ class MessageStorage:
filtered_display_message = re.sub(pattern, "", display_message, flags=re.DOTALL)
else:
# 如果没有设置display_message，使用processed_plain_text作为显示消息
filtered_display_message = re.sub(pattern, "", message.processed_plain_text, flags=re.DOTALL) if message.processed_plain_text else ""
filtered_display_message = (
re.sub(pattern, "", message.processed_plain_text, flags=re.DOTALL)
if message.processed_plain_text
else ""
)
interest_value = 0
is_mentioned = False
reply_to = message.reply_to
@@ -168,9 +172,11 @@ class MessageStorage:
from src.common.database.sqlalchemy_models import get_db_session
async with get_db_session() as session:
matched_message = (await session.execute(
select(Messages).where(Messages.message_id == mmc_message_id).order_by(desc(Messages.time))
)).scalar()
matched_message = (
await session.execute(
select(Messages).where(Messages.message_id == mmc_message_id).order_by(desc(Messages.time))
)
).scalar()
if matched_message:
await session.execute(
@@ -204,9 +210,11 @@ class MessageStorage:
from src.common.database.sqlalchemy_models import get_db_session
async with get_db_session() as session:
image_record = (await session.execute(
select(Images).where(Images.description == description).order_by(desc(Images.timestamp))
)).scalar()
image_record = (
await session.execute(
select(Images).where(Images.description == description).order_by(desc(Images.timestamp))
)
).scalar()
return f"[picid:{image_record.image_id}]" if image_record else match.group(0)
except Exception:
return match.group(0)
@@ -287,15 +295,19 @@ class MessageStorage:
from src.common.database.sqlalchemy_models import Messages
# 查找需要修复的记录：interest_value为0、null或很小的值
query = select(Messages).where(
(Messages.chat_id == chat_id) &
(Messages.time >= since_time) &
(
(Messages.interest_value == 0) |
(Messages.interest_value.is_(None)) |
(Messages.interest_value < 0.1)
query = (
select(Messages)
.where(
(Messages.chat_id == chat_id)
& (Messages.time >= since_time)
& (
(Messages.interest_value == 0)
| (Messages.interest_value.is_(None))
| (Messages.interest_value < 0.1)
)
)
).limit(50) # 限制每次修复的数量，避免性能问题
.limit(50)
) # 限制每次修复的数量，避免性能问题
result = await session.execute(query)
messages_to_fix = result.scalars().all()
@@ -307,7 +319,7 @@ class MessageStorage:
default_interest = 0.3 # 默认中等兴趣度
# 如果消息内容较长，可能是重要消息，兴趣度稍高
if hasattr(msg, 'processed_plain_text') and msg.processed_plain_text:
if hasattr(msg, "processed_plain_text") and msg.processed_plain_text:
text_length = len(msg.processed_plain_text)
if text_length > 50: # 长消息
default_interest = 0.4
@@ -315,13 +327,15 @@ class MessageStorage:
default_interest = 0.35
# 如果是被@的消息，兴趣度更高
if getattr(msg, 'is_mentioned', False):
if getattr(msg, "is_mentioned", False):
default_interest = min(default_interest + 0.2, 0.8)
# 执行更新
update_stmt = update(Messages).where(
Messages.message_id == msg.message_id
).values(interest_value=default_interest)
update_stmt = (
update(Messages)
.where(Messages.message_id == msg.message_id)
.values(interest_value=default_interest)
)
result = await session.execute(update_stmt)
if result.rowcount > 0: