117 lines
3.6 KiB
Python
117 lines
3.6 KiB
Python
 
|
"""群聊学习 —— 触发器(Trigger)层。"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
import random
|
||
|
|
import time
|
||
|
|
|
||
|
|
from nonebot import get_adapter, logger, on_message, require
|
||
|
|
from nonebot.adapters.onebot.v11 import (
|
||
|
|
GROUP,
|
||
|
|
ActionFailed,
|
||
|
|
Adapter,
|
||
|
|
GroupMessageEvent,
|
||
|
|
Message,
|
||
|
|
)
|
||
|
|
from nonebot.params import Arg
|
||
|
|
from nonebot.rule import Rule
|
||
|
|
from nonebot.typing import T_State
|
||
|
|
|
||
|
|
require("nonebot_plugin_orm")
|
||
|
|
require("nonebot_plugin_apscheduler")
|
||
|
|
from nonebot_plugin_apscheduler import scheduler
|
||
|
|
|
||
|
|
from nonebot_plugin_orm import get_session
|
||
|
|
|
||
|
|
from ..config import NICKNAME, config_manager
|
||
|
|
from ..models import ChatMessage
|
||
|
|
from ..services.learn import LearningChat
|
||
|
|
|
||
|
|
|
||
|
|
async def ChatRule(event: GroupMessageEvent, state: T_State) -> bool:
|
||
|
|
if answers := await LearningChat(event).answer():
|
||
|
|
state["answers"] = answers
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
learning_chat = on_message(
|
||
|
|
priority=99,
|
||
|
|
block=False,
|
||
|
|
rule=Rule(ChatRule),
|
||
|
|
permission=GROUP,
|
||
|
|
state={
|
||
|
|
"pm_name": "群聊学习",
|
||
|
|
"pm_description": "(被动技能)bot会学习群友们的发言",
|
||
|
|
"pm_usage": "群聊学习",
|
||
|
|
"pm_priority": 1,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@learning_chat.handle()
|
||
|
|
async def _(event: GroupMessageEvent, answers=Arg("answers")):
|
||
|
|
for answer in answers:
|
||
|
|
try:
|
||
|
|
logger.info(
|
||
|
|
"群聊学习", f'{NICKNAME}将向群<m>{event.group_id}</m>回复<m>"{answer}"</m>'
|
||
|
|
)
|
||
|
|
msg = await learning_chat.send(Message(answer))
|
||
|
|
async with get_session() as session:
|
||
|
|
session.add(ChatMessage(
|
||
|
|
group_id=event.group_id,
|
||
|
|
user_id=event.self_id,
|
||
|
|
message_id=msg["message_id"],
|
||
|
|
message=answer,
|
||
|
|
raw_message=answer,
|
||
|
|
time=int(time.time()),
|
||
|
|
plain_text=Message(answer).extract_plain_text(),
|
||
|
|
))
|
||
|
|
await session.commit()
|
||
|
|
await asyncio.sleep(random.random() + 0.5)
|
||
|
|
except ActionFailed:
|
||
|
|
logger.info(
|
||
|
|
"群聊学习",
|
||
|
|
f'{NICKNAME}向群<m>{event.group_id}</m>的回复<m>"{answer}"</m>发送<r>失败，可能处于风控中</r>',
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@scheduler.scheduled_job("interval", minutes=3, misfire_grace_time=5)
|
||
|
|
async def speak_up():
|
||
|
|
if not config_manager.config.total_enable:
|
||
|
|
return
|
||
|
|
try:
|
||
|
|
bots = get_adapter(Adapter).bots
|
||
|
|
if len(bots) == 0:
|
||
|
|
return
|
||
|
|
bot = list(bots.values())[0]
|
||
|
|
except ValueError:
|
||
|
|
return
|
||
|
|
if not (speak := await LearningChat.speak(int(bot.self_id))):
|
||
|
|
return
|
||
|
|
group_id, messages = speak
|
||
|
|
for msg in messages:
|
||
|
|
try:
|
||
|
|
logger.info("群聊学习", f'{NICKNAME}向群<m>{group_id}</m>主动发言<m>"{msg}"</m>')
|
||
|
|
send_result = await bot.send_group_msg(
|
||
|
|
group_id=group_id, message=Message(msg)
|
||
|
|
)
|
||
|
|
async with get_session() as session:
|
||
|
|
session.add(ChatMessage(
|
||
|
|
group_id=group_id,
|
||
|
|
user_id=int(bot.self_id),
|
||
|
|
message_id=send_result["message_id"],
|
||
|
|
message=msg,
|
||
|
|
raw_message=msg,
|
||
|
|
time=int(time.time()),
|
||
|
|
plain_text=Message(msg).extract_plain_text(),
|
||
|
|
))
|
||
|
|
await session.commit()
|
||
|
|
await asyncio.sleep(random.randint(2, 4))
|
||
|
|
except ActionFailed:
|
||
|
|
logger.info(
|
||
|
|
"群聊学习",
|
||
|
|
f'{NICKNAME}向群<m>{group_id}</m>主动发言<m>"{msg}"</m><r>发送失败，可能处于风控中</r>',
|
||
|
|
)
|