mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2025-04-12 23:29:37 +08:00
459 lines
25 KiB
Python
459 lines
25 KiB
Python
import datetime
|
||
import random
|
||
import re
|
||
import time
|
||
from functools import cmp_to_key
|
||
|
||
try:
|
||
import jieba_fast.analyse as jieba_analyse
|
||
except ImportError:
|
||
import jieba.analyse as jieba_analyse
|
||
from typing import List, Union, Optional, Tuple
|
||
from enum import IntEnum, auto
|
||
from nonebot import get_bot
|
||
from nonebot.adapters.onebot.v11 import GroupMessageEvent, MessageSegment, ActionFailed
|
||
from tortoise.functions import Count
|
||
from LittlePaimon.utils import NICKNAME, SUPERUSERS, logger
|
||
from .models import ChatBlackList, ChatContext, ChatAnswer, ChatMessage
|
||
from .config import config_manager
|
||
|
||
chat_config = config_manager.config
|
||
|
||
NO_PERMISSION_WORDS = [f'{NICKNAME}就喜欢说这个，哼！', f'你管得着{NICKNAME}吗！']
|
||
ENABLE_WORDS = [f'{NICKNAME}会尝试学你们说怪话！', f'好的呢，让{NICKNAME}学学你们的说话方式~']
|
||
DISABLE_WORDS = [f'好好好，{NICKNAME}不学说话就是了！', f'果面呐噻，{NICKNAME}以后不学了...']
|
||
SORRY_WORDS = [f'{NICKNAME}知道错了...达咩!', f'{NICKNAME}不会再这么说了...', f'果面呐噻,{NICKNAME}说错话了...']
|
||
DOUBT_WORDS = [f'{NICKNAME}有说什么奇怪的话吗？']
|
||
BREAK_REPEAT_WORDS = ['打断复读', '打断！']
|
||
ALL_WORDS = NO_PERMISSION_WORDS + SORRY_WORDS + DOUBT_WORDS + ENABLE_WORDS + DISABLE_WORDS + BREAK_REPEAT_WORDS
|
||
|
||
|
||
class Result(IntEnum):
|
||
Learn = auto()
|
||
Pass = auto()
|
||
Repeat = auto()
|
||
Ban = auto()
|
||
SetEnable = auto()
|
||
|
||
|
||
class LearningChat:
|
||
def __init__(self, event: GroupMessageEvent):
|
||
if event.reply:
|
||
self.reply = event.reply
|
||
self.data = ChatMessage(
|
||
group_id=event.group_id,
|
||
user_id=event.user_id,
|
||
message_id=event.message_id,
|
||
message=re.sub(r'(\[CQ:at,qq=.+])|(\[CQ:reply,id=.+])', '',
|
||
re.sub(r'(,subType=\d+,url=.+])', r']', event.raw_message)).strip(),
|
||
raw_message=event.raw_message,
|
||
plain_text=event.get_plaintext(),
|
||
time=event.time
|
||
)
|
||
else:
|
||
self.data = ChatMessage(
|
||
group_id=event.group_id,
|
||
user_id=event.user_id,
|
||
message_id=event.message_id,
|
||
message=re.sub(r'(\[CQ:at,qq=.+])', '',
|
||
re.sub(r'(,subType=\d+,url=.+])', r']', event.raw_message)).strip(),
|
||
raw_message=event.raw_message,
|
||
plain_text=event.get_plaintext(),
|
||
time=event.time
|
||
)
|
||
self.reply = None
|
||
self.bot_id = event.self_id
|
||
self.to_me = event.to_me or NICKNAME in self.data.message
|
||
self.role = 'superuser' if event.user_id in SUPERUSERS else event.sender.role
|
||
self.config = config_manager.get_group_config(self.data.group_id)
|
||
self.ban_users = set(chat_config.ban_users + self.config.ban_users)
|
||
|
||
async def _learn(self) -> Result:
|
||
# logger.debug('群聊学习', f'收到来自群<m>{self.data.group_id}</m>的消息<m>{self.data.message}</m>')
|
||
if not chat_config.total_enable or not self.config.enable or self.data.user_id in self.ban_users:
|
||
# 如果未开启群聊学习或者发言人在屏蔽列表中，跳过
|
||
return Result.Pass
|
||
elif self.to_me and '不可以' in self.data.message:
|
||
# 如果是对某句话进行禁言
|
||
return Result.Ban
|
||
elif self.to_me and any(w in self.data.message for w in {'学说话', '快学', '开启学习', '闭嘴', '别学', '关闭学习'}):
|
||
return Result.SetEnable
|
||
elif not await self._check_allow(self.data):
|
||
# 本消息不合法，跳过
|
||
return Result.Pass
|
||
elif self.reply:
|
||
# 如果是回复消息
|
||
if not (message := await ChatMessage.get_or_none(message_id=self.reply.message_id)):
|
||
# 回复的消息在数据库中有记录
|
||
logger.debug('群聊学习', '➤是否学习：回复的消息不在数据库中，不学习')
|
||
return Result.Pass
|
||
if message.user_id in self.ban_users:
|
||
# 且回复的人不在屏蔽列表中
|
||
return Result.Pass
|
||
if not await self._check_allow(message):
|
||
# 且回复的内容通过校验
|
||
logger.debug('群聊学习', '➤是否学习：回复的消息未通过校验，不学习')
|
||
return Result.Pass
|
||
# 则将该回复作为该消息的答案
|
||
await self._set_answer(message)
|
||
return Result.Learn
|
||
elif messages := await ChatMessage.filter(group_id=self.data.group_id, time__gte=self.data.time - 3600).limit(
|
||
5):
|
||
# 获取本群一个小时内的最后5条消息
|
||
if messages[0].message == self.data.message:
|
||
# 判断是否为复读中
|
||
logger.debug('群聊学习', '➤是否学习：复读中，不学习')
|
||
return Result.Repeat
|
||
for message in messages:
|
||
# 如果5条内有相关信息，就作为该消息的答案
|
||
if message.user_id not in self.ban_users and set(self.data.keyword_list) & set(
|
||
message.keyword_list) and self.data.keyword_list != message.keyword_list and await self._check_allow(
|
||
message):
|
||
await self._set_answer(message)
|
||
return Result.Learn
|
||
# 如果没有相关信息
|
||
if messages[0].user_id in self.ban_users or not await self._check_allow(messages[0]):
|
||
# 且最后一条消息的发送者不在屏蔽列表中并通过校验
|
||
logger.debug('群聊学习', '➤是否学习：最后一条消息未通过校验，不学习')
|
||
return Result.Pass
|
||
# 则作为最后一条消息的答案
|
||
await self._set_answer(messages[0])
|
||
return Result.Learn
|
||
else:
|
||
# 不符合任何情况，跳过
|
||
return Result.Pass
|
||
|
||
async def answer(self) -> Optional[List[Union[MessageSegment, str]]]:
|
||
"""获取这句话的回复"""
|
||
result = await self._learn()
|
||
await self.data.save()
|
||
if result == Result.Ban:
|
||
# 禁用某句话
|
||
if self.role not in {'superuser', 'admin', 'owner'}:
|
||
# 检查权限
|
||
return [random.choice(NO_PERMISSION_WORDS)]
|
||
if self.reply:
|
||
ban_result = await self._ban(message_id=self.reply.message_id)
|
||
else:
|
||
ban_result = await self._ban()
|
||
if ban_result:
|
||
return [random.choice(SORRY_WORDS)]
|
||
else:
|
||
return [random.choice(DOUBT_WORDS)]
|
||
elif result == Result.SetEnable:
|
||
# 开启/关闭学习
|
||
if self.role in {'superuser', 'admin', 'owner'}:
|
||
# 检查权限
|
||
if any(w in self.data.message for w in {'学说话', '快学', '开启学习'}):
|
||
self.config.update(enable=True)
|
||
config_manager.config.group_config[self.data.group_id] = self.config
|
||
config_manager.save()
|
||
return [random.choice(ENABLE_WORDS)]
|
||
else:
|
||
self.config.update(enable=False)
|
||
config_manager.config.group_config[self.data.group_id] = self.config
|
||
config_manager.save()
|
||
return [random.choice(DISABLE_WORDS)]
|
||
else:
|
||
return [random.choice(NO_PERMISSION_WORDS)]
|
||
elif result == Result.Pass:
|
||
# 跳过
|
||
return None
|
||
elif result == Result.Repeat and (messages := await ChatMessage.filter(group_id=self.data.group_id,
|
||
time__gte=self.data.time - 3600).limit(
|
||
self.config.repeat_threshold)):
|
||
# 如果达到阈值，进行复读
|
||
if len(messages) >= self.config.repeat_threshold and all(
|
||
message.message == self.data.message and message.user_id != self.bot_id for message in
|
||
messages):
|
||
if random.random() < self.config.break_probability:
|
||
logger.debug('群聊学习', f'➤➤是否回复：达到复读阈值，打断复读！')
|
||
return [random.choice(BREAK_REPEAT_WORDS)]
|
||
else:
|
||
logger.debug('群聊学习', f'➤➤是否回复：达到复读阈值，复读<m>{messages[0].message}</m>')
|
||
return [self.data.message]
|
||
else:
|
||
# 回复
|
||
if self.data.is_plain_text and len(self.data.plain_text) <= 1:
|
||
logger.debug('群聊学习', '➤➤是否回复：消息过短，不回复')
|
||
return None
|
||
if not (context := await ChatContext.get_or_none(keywords=self.data.keywords)):
|
||
logger.debug('群聊学习', '➤➤是否回复：尚未有已学习的回复，不回复')
|
||
return None
|
||
|
||
# 获取回复阈值
|
||
if not self.to_me:
|
||
answer_choices = list(
|
||
range(self.config.answer_threshold - len(self.config.answer_threshold_weights) + 1,
|
||
self.config.answer_threshold + 1))
|
||
|
||
answer_count_threshold = random.choices(answer_choices, weights=self.config.answer_threshold_weights)[0]
|
||
|
||
if len(self.data.keyword_list) == chat_config.KEYWORDS_SIZE:
|
||
answer_count_threshold -= 1
|
||
cross_group_threshold = chat_config.cross_group_threshold
|
||
else:
|
||
answer_count_threshold = 1
|
||
cross_group_threshold = 1
|
||
|
||
# 获取满足跨群条件的回复
|
||
answers_cross = await ChatAnswer.filter(context=context, count__gte=answer_count_threshold,
|
||
keywords__in=await ChatAnswer.annotate(
|
||
cross=Count('keywords')).group_by('keywords').filter(
|
||
cross__gte=cross_group_threshold).values_list('keywords',
|
||
flat=True))
|
||
|
||
answer_same_group = await ChatAnswer.filter(context=context, count__gte=answer_count_threshold,
|
||
group_id=self.data.group_id)
|
||
|
||
candidate_answers: List[Optional[ChatAnswer]] = []
|
||
# 检查候选回复是否在屏蔽列表中
|
||
for answer in set(answers_cross) | set(answer_same_group):
|
||
if not await self._check_allow(answer):
|
||
continue
|
||
# if answer_count_threshold > 0:
|
||
# answer.count -= answer_count_threshold - 1
|
||
candidate_answers.append(answer)
|
||
if not candidate_answers:
|
||
logger.debug('群聊学习', '➤➤是否回复：没有符合条件的候选回复')
|
||
return None
|
||
|
||
# 从候选回复中进行选择
|
||
sum_count = sum(answer.count for answer in candidate_answers)
|
||
per_list = [answer.count / sum_count * (1 - 1 / answer.count) for answer in candidate_answers]
|
||
|
||
per_list.append(1 - sum(per_list))
|
||
answer_dict = tuple(zip(candidate_answers, per_list))
|
||
logger.debug('群聊学习',
|
||
f'➤➤是否回复：候选回复有<m>{"|".join([f"""{a.keywords}({round(p, 3)})""" for a, p in answer_dict])}|不回复({round(per_list[-1], 3)})</m>')
|
||
|
||
if (result := random.choices(candidate_answers + [None], weights=per_list)[0]) is None:
|
||
logger.debug('群聊学习', '➤➤是否回复：但不进行回复')
|
||
return None
|
||
result_message = random.choice(result.messages)
|
||
logger.debug('群聊学习', f'➤➤是否回复：将回复<m>{result_message}</m>')
|
||
return [result_message]
|
||
|
||
async def _ban(self, message_id: Optional[int] = None) -> bool:
|
||
"""屏蔽消息"""
|
||
bot = get_bot()
|
||
if message_id:
|
||
# 如果有指定消息ID，则屏蔽该消息
|
||
if (message := await ChatMessage.get_or_none(message_id=message_id)) and message.message not in ALL_WORDS:
|
||
keywords = message.keywords
|
||
try:
|
||
await bot.delete_msg(message_id=message_id)
|
||
except ActionFailed:
|
||
logger.info('群聊学习', f'待禁用消息<m>{message_id}</m>尝试撤回<r>失败</r>')
|
||
else:
|
||
return False
|
||
elif (last_reply := await ChatMessage.filter(group_id=self.data.group_id, user_id=self.bot_id).first()) and (
|
||
last_reply.message not in ALL_WORDS):
|
||
# 没有指定消息ID，则屏蔽最后一条回复
|
||
keywords = last_reply.keywords
|
||
try:
|
||
await bot.delete_msg(message_id=last_reply.message_id)
|
||
except ActionFailed:
|
||
logger.info('群聊学习', f'待禁用消息<m>{last_reply.message_id}</m>尝试撤回<r>失败</r>')
|
||
else:
|
||
return False
|
||
if ban_word := await ChatBlackList.get_or_none(keywords=keywords):
|
||
# 如果已有屏蔽记录
|
||
if self.data.group_id not in ban_word.ban_group_id:
|
||
# 如果不在屏蔽群列表中，则添加
|
||
ban_word.ban_group_id.append(self.data.group_id)
|
||
if len(ban_word.ban_group_id) >= 2:
|
||
# 如果有超过2个群都屏蔽了该条消息，则全局屏蔽
|
||
ban_word.global_ban = True
|
||
logger.info('群聊学习', f'学习词<m>{keywords}</m>将被全局禁用')
|
||
await ChatAnswer.filter(keywords=keywords).delete()
|
||
else:
|
||
logger.info('群聊学习', f'群<m>{self.data.group_id}</m>禁用了学习词<m>{keywords}</m>')
|
||
await ChatAnswer.filter(keywords=keywords, group_id=self.data.group_id).delete()
|
||
else:
|
||
# 没有屏蔽记录，则新建
|
||
logger.info('群聊学习', f'群<m>{self.data.group_id}</m>禁用了学习词<m>{keywords}</m>')
|
||
ban_word = ChatBlackList(keywords=keywords, ban_group_id=[self.data.group_id])
|
||
await ChatAnswer.filter(keywords=keywords, group_id=self.data.group_id).delete()
|
||
await ChatContext.filter(keywords=keywords).delete()
|
||
await ban_word.save()
|
||
return True
|
||
|
||
@staticmethod
|
||
async def add_ban(data: Union[ChatMessage, ChatContext, ChatAnswer]):
|
||
if ban_word := await ChatBlackList.get_or_none(keywords=data.keywords):
|
||
# 如果已有屏蔽记录
|
||
if isinstance(data, ChatMessage):
|
||
if data.group_id not in ban_word.ban_group_id:
|
||
# 如果不在屏蔽群列表中，则添加
|
||
ban_word.ban_group_id.append(data.group_id)
|
||
if len(ban_word.ban_group_id) >= 2:
|
||
# 如果有超过2个群都屏蔽了该条消息，则全局屏蔽
|
||
ban_word.global_ban = True
|
||
logger.info('群聊学习', f'学习词<m>{data.keywords}</m>将被全局禁用')
|
||
await ChatAnswer.filter(keywords=data.keywords).delete()
|
||
else:
|
||
logger.info('群聊学习', f'群<m>{data.group_id}</m>禁用了学习词<m>{data.keywords}</m>')
|
||
await ChatAnswer.filter(keywords=data.keywords, group_id=data.group_id).delete()
|
||
else:
|
||
ban_word.global_ban = True
|
||
logger.info('群聊学习', f'学习词<m>{data.keywords}</m>将被全局禁用')
|
||
await ChatAnswer.filter(keywords=data.keywords).delete()
|
||
else:
|
||
# 没有屏蔽记录，则新建
|
||
if isinstance(data, ChatMessage):
|
||
logger.info('群聊学习', f'群<m>{data.group_id}</m>禁用了学习词<m>{data.keywords}</m>')
|
||
ban_word = ChatBlackList(keywords=data.keywords, ban_group_id=[data.group_id])
|
||
await ChatAnswer.filter(keywords=data.keywords, group_id=data.group_id).delete()
|
||
else:
|
||
|
||
logger.info('群聊学习', f'学习词<m>{data.keywords}</m>将被全局禁用')
|
||
ban_word = ChatBlackList(keywords=data.keywords, global_ban=True)
|
||
await ChatAnswer.filter(keywords=data.keywords).delete()
|
||
await ChatContext.filter(keywords=data.keywords).delete()
|
||
await ban_word.save()
|
||
|
||
@staticmethod
|
||
async def speak(self_id: int) -> Optional[Tuple[int, List[Union[str, MessageSegment]]]]:
|
||
# 主动发言
|
||
cur_time = int(time.time())
|
||
today_time = time.mktime(datetime.date.today().timetuple())
|
||
# 获取两小时内消息超过10条的群列表
|
||
groups = await ChatMessage.filter(time__gte=today_time).annotate(count=Count('id')).group_by('group_id'). \
|
||
filter(count__gte=10).values_list('group_id', flat=True)
|
||
if not groups:
|
||
return None
|
||
total_messages = {}
|
||
# 获取这些群的两小时内的所有消息
|
||
for group_id in groups:
|
||
if messages := await ChatMessage.filter(group_id=group_id, time__gte=today_time):
|
||
total_messages[group_id] = messages
|
||
if not total_messages:
|
||
return None
|
||
|
||
# 根据消息平均间隔来对群进行排序
|
||
def group_popularity_cmp(left_group: Tuple[int, List[ChatMessage]],
|
||
right_group: Tuple[int, List[ChatMessage]]):
|
||
def cmp(a, b):
|
||
return (a > b) - (a < b)
|
||
|
||
left_group_id, left_messages = left_group
|
||
right_group_id, right_messages = right_group
|
||
left_duration = left_messages[0].time - left_messages[-1].time
|
||
right_duration = right_messages[0].time - right_messages[-1].time
|
||
return cmp(len(left_messages) / left_duration, len(right_messages) / right_duration)
|
||
|
||
popularity: List[Tuple[int, List[ChatMessage]]] = sorted(total_messages.items(),
|
||
key=cmp_to_key(group_popularity_cmp))
|
||
logger.debug('群聊学习', f'主动发言：群热度排行<m>{">".join([str(g[0]) for g in popularity])}</m>')
|
||
for group_id, messages in popularity:
|
||
config = config_manager.get_group_config(group_id)
|
||
|
||
# 是否开启了主动发言
|
||
if not config.speak_enable:
|
||
continue
|
||
|
||
# 如果最后一条消息是自己发的，则不主动发言
|
||
last_reply = await ChatMessage.filter(group_id=group_id, user_id=self_id).first()
|
||
if last_reply and last_reply.time >= messages[0].time:
|
||
continue
|
||
|
||
# 该群每多少秒发一条消息
|
||
avg_interval = (messages[0].time - messages[-1].time) / len(messages)
|
||
# 如果该群已沉默的时间小于阈值，则不主动发言
|
||
if cur_time - messages[0].time < avg_interval * config.speak_threshold + config.speak_min_interval:
|
||
continue
|
||
|
||
if contexts := await ChatContext.filter(count__gte=config.answer_threshold).all():
|
||
speak_list = []
|
||
# context = random.choices(contexts, weights=[context.count for context in contexts])[0]
|
||
contexts.sort(key=lambda x: x.count)
|
||
for context in contexts:
|
||
if (random.random() < config.speak_continuously_probability or not len(speak_list)) and len(
|
||
speak_list) < config.speak_continuously_max_len and (
|
||
answers := await ChatAnswer.filter(context=context,
|
||
group_id=group_id,
|
||
count__gte=config.answer_threshold)):
|
||
answer = random.choices(answers,
|
||
weights=[answer.count + 1 if answer.time >= today_time else answer.count
|
||
for answer in answers])[0]
|
||
message = random.choice(answer.messages)
|
||
speak_list.append(message)
|
||
while random.random() < config.speak_continuously_probability and len(
|
||
speak_list) < config.speak_continuously_max_len:
|
||
if (follow_context := await ChatContext.get_or_none(keywords=answer.keywords)) and (
|
||
follow_answers := await ChatAnswer.filter(
|
||
group_id=group_id,
|
||
context=follow_context,
|
||
count__gte=config.answer_threshold)):
|
||
answer = random.choices(follow_answers,
|
||
weights=[a.count + 1 if a.time >= today_time else a.count
|
||
for a in follow_answers])[0]
|
||
message = random.choice(answer.messages)
|
||
speak_list.append(message)
|
||
else:
|
||
break
|
||
else:
|
||
break
|
||
if speak_list:
|
||
# logger.debug('群聊学习', f'主动发言：将向群<m>{group_id}</m>主动发言<m>{" ".join(speak_list)}</m>')
|
||
if random.random() < config.speak_poke_probability:
|
||
last_speak_users = set(
|
||
message.user_id for message in messages[:5] if message.user_id != self_id)
|
||
select_user = random.choice(list(last_speak_users))
|
||
speak_list.append(MessageSegment('poke', {'qq': select_user}))
|
||
return group_id, speak_list
|
||
else:
|
||
return None
|
||
|
||
async def _set_answer(self, message: ChatMessage):
|
||
if context := await ChatContext.get_or_none(keywords=message.keywords):
|
||
if context.count < chat_config.learn_max_count:
|
||
context.count += 1
|
||
context.time = self.data.time
|
||
if answer := await ChatAnswer.get_or_none(keywords=self.data.keywords,
|
||
group_id=self.data.group_id,
|
||
context=context):
|
||
if answer.count < chat_config.learn_max_count:
|
||
answer.count += 1
|
||
answer.time = self.data.time
|
||
if self.data.message not in answer.messages:
|
||
answer.messages.append(self.data.message)
|
||
else:
|
||
answer = ChatAnswer(keywords=self.data.keywords,
|
||
group_id=self.data.group_id,
|
||
time=self.data.time,
|
||
context=context,
|
||
messages=[self.data.message])
|
||
await answer.save()
|
||
await context.save()
|
||
else:
|
||
context = await ChatContext.create(keywords=message.keywords,
|
||
time=self.data.time)
|
||
answer = await ChatAnswer.create(keywords=self.data.keywords,
|
||
group_id=self.data.group_id,
|
||
time=self.data.time,
|
||
context=context,
|
||
messages=[self.data.message])
|
||
logger.debug('群聊学习', f'➤将被学习为<m>{message.message}</m>的回答，已学次数为<m>{answer.count}</m>')
|
||
|
||
async def _check_allow(self, message: Union[ChatMessage, ChatAnswer]) -> bool:
|
||
raw_message = message.message if isinstance(message, ChatMessage) else message.messages[0]
|
||
keywords = message.keywords
|
||
if any(i in raw_message for i in
|
||
{'[CQ:xml', '[CQ:json', '[CQ:at', '[CQ:video', '[CQ:record', '[CQ:share'}):
|
||
# logger.debug('群聊学习', f'➤检验<m>{keywords}</m><r>不通过</r>')
|
||
return False
|
||
if any(i in raw_message for i in self.config.ban_words):
|
||
# logger.debug('群聊学习', f'➤检验<m>{keywords}</m><r>不通过</r>')
|
||
return False
|
||
if raw_message.startswith('[') and raw_message.endswith(']'):
|
||
# logger.debug('群聊学习', f'➤检验<m>{keywords}</m><r>不通过</r>')
|
||
return False
|
||
if ban_word := await ChatBlackList.get_or_none(keywords=keywords):
|
||
if ban_word.global_ban or message.group_id in ban_word.ban_group_id:
|
||
# logger.debug('群聊学习', f'➤检验<m>{keywords}</m><r>不通过</r>')
|
||
return False
|
||
# logger.debug('群聊学习', f'➤检验<m>{keywords}</m><g>通过</g>')
|
||
return True
|