51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
from typing import Optional
|
|
|
|
from ...common.database import Database
|
|
from .message import Message
|
|
from loguru import logger
|
|
|
|
|
|
class MessageStorage:
|
|
def __init__(self):
|
|
self.db = Database.get_instance()
|
|
|
|
async def store_message(self, message: Message, topic: Optional[str] = None) -> None:
|
|
"""存储消息到数据库"""
|
|
try:
|
|
if not message.is_emoji:
|
|
message_data = {
|
|
"group_id": message.group_id,
|
|
"user_id": message.user_id,
|
|
"message_id": message.message_id,
|
|
"raw_message": message.raw_message,
|
|
"plain_text": message.plain_text,
|
|
"processed_plain_text": message.processed_plain_text,
|
|
"time": message.time,
|
|
"user_nickname": message.user_nickname,
|
|
"user_cardname": message.user_cardname,
|
|
"group_name": message.group_name,
|
|
"topic": topic,
|
|
"detailed_plain_text": message.detailed_plain_text,
|
|
}
|
|
else:
|
|
message_data = {
|
|
"group_id": message.group_id,
|
|
"user_id": message.user_id,
|
|
"message_id": message.message_id,
|
|
"raw_message": message.raw_message,
|
|
"plain_text": message.plain_text,
|
|
"processed_plain_text": '[表情包]',
|
|
"time": message.time,
|
|
"user_nickname": message.user_nickname,
|
|
"user_cardname": message.user_cardname,
|
|
"group_name": message.group_name,
|
|
"topic": topic,
|
|
"detailed_plain_text": message.detailed_plain_text,
|
|
}
|
|
|
|
self.db.db.messages.insert_one(message_data)
|
|
except Exception:
|
|
logger.exception(f"存储消息失败")
|
|
|
|
# 如果需要其他存储相关的函数，可以在这里添加
|