55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
 
|
"""语录 —— 回复提取工具(utils 层)。"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from nonebot import logger
|
||
|
|
from nonebot.adapters.onebot.v11 import Bot, Message
|
||
|
|
|
||
|
|
|
||
|
|
class Reply:
|
||
|
|
def __init__(self):
|
||
|
|
self.message: str = ""
|
||
|
|
self.user_id: int = -1
|
||
|
|
self.user_card: str = ""
|
||
|
|
self.time: int = -1
|
||
|
|
|
||
|
|
|
||
|
|
async def extract_reply(event, bot: Bot) -> Reply:
|
||
|
|
"""
|
||
|
|
提取回复内容
|
||
|
|
"""
|
||
|
|
logger.info(event["group_id"])
|
||
|
|
reply_data = Reply()
|
||
|
|
if event["reply"] and event["reply"]["message_id"]: # 待优化
|
||
|
|
reply = event["reply"]
|
||
|
|
sender = reply["sender"]
|
||
|
|
mes: Message = reply["message"]
|
||
|
|
logger.info(mes)
|
||
|
|
msg = mes[0]["data"]["text"]
|
||
|
|
if msg:
|
||
|
|
reply_data.message = msg
|
||
|
|
meminfo = await get_member_info(bot=bot, user_id=sender["user_id"], group_id=event["group_id"])
|
||
|
|
reply_data.user_card = meminfo["card"] or meminfo["nickname"]
|
||
|
|
reply_data.time = reply["time"]
|
||
|
|
reply_data.user_id = reply["sender"]["user_id"]
|
||
|
|
else:
|
||
|
|
reply_data.message = -1
|
||
|
|
else:
|
||
|
|
mes: Message = event["message"]
|
||
|
|
logger.info(mes)
|
||
|
|
msg = mes[0]["data"]["text"].replace("maq", "")
|
||
|
|
if msg:
|
||
|
|
reply_data.message = msg
|
||
|
|
meminfo = await get_member_info(bot=bot, user_id=event["user_id"], group_id=event["group_id"])
|
||
|
|
reply_data.user_card = meminfo["card"] or meminfo["nickname"]
|
||
|
|
reply_data.time = event["time"]
|
||
|
|
reply_data.user_id = event["user_id"]
|
||
|
|
else:
|
||
|
|
reply_data.message = -1
|
||
|
|
return reply_data
|
||
|
|
|
||
|
|
|
||
|
|
async def get_member_info(bot: Bot, group_id: int, user_id: int):
|
||
|
|
member_info = await bot.get_group_member_info(group_id=group_id, user_id=user_id)
|
||
|
|
return member_info
|