mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-07-20 13:47:15 +08:00
refactor: stripp processing unrelated to channel
This commit is contained in:
@@ -20,7 +20,7 @@ class Channel(object):
|
|||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
def send(self, msg, receiver):
|
def send(self, reply: Reply, context: Context):
|
||||||
"""
|
"""
|
||||||
send message to user
|
send message to user
|
||||||
:param msg: message content
|
:param msg: message content
|
||||||
|
|||||||
223
channel/chat_channel.py
Normal file
223
channel/chat_channel.py
Normal file
@@ -0,0 +1,223 @@
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
import re
|
||||||
|
import time
|
||||||
|
from common.expired_dict import ExpiredDict
|
||||||
|
from channel.channel import Channel
|
||||||
|
from bridge.reply import *
|
||||||
|
from bridge.context import *
|
||||||
|
from config import conf
|
||||||
|
from common.log import logger
|
||||||
|
from plugins import *
|
||||||
|
|
||||||
|
|
||||||
|
# 抽象类, 它包含了与消息通道无关的通用处理逻辑
|
||||||
|
class ChatChannel(Channel):
|
||||||
|
name = None # 登录的用户名
|
||||||
|
user_id = None # 登录的用户id
|
||||||
|
def __init__(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# 根据消息构造context,消息内容相关的触发项写在这里
|
||||||
|
def _compose_context(self, ctype: ContextType, content, **kwargs):
|
||||||
|
context = Context(ctype, content)
|
||||||
|
context.kwargs = kwargs
|
||||||
|
# context首次传入时,origin_ctype是None,
|
||||||
|
# 引入的起因是:当输入语音时,会嵌套生成两个context,第一步语音转文本,第二步通过文本生成文字回复。
|
||||||
|
# origin_ctype用于第二步文本回复时,判断是否需要匹配前缀,如果是私聊的语音,就不需要匹配前缀
|
||||||
|
if 'origin_ctype' not in context:
|
||||||
|
context['origin_ctype'] = ctype
|
||||||
|
# context首次传入时,receiver是None,根据类型设置receiver
|
||||||
|
first_in = 'receiver' not in context
|
||||||
|
|
||||||
|
# 群名匹配过程,设置session_id和receiver
|
||||||
|
if first_in: # context首次传入时,receiver是None,根据类型设置receiver
|
||||||
|
config = conf()
|
||||||
|
cmsg = context['msg']
|
||||||
|
if context["isgroup"]:
|
||||||
|
group_name = cmsg.other_user_nickname
|
||||||
|
group_id = cmsg.other_user_id
|
||||||
|
|
||||||
|
group_name_white_list = config.get('group_name_white_list', [])
|
||||||
|
group_name_keyword_white_list = config.get('group_name_keyword_white_list', [])
|
||||||
|
if any([group_name in group_name_white_list, 'ALL_GROUP' in group_name_white_list, check_contain(group_name, group_name_keyword_white_list)]):
|
||||||
|
group_chat_in_one_session = conf().get('group_chat_in_one_session', [])
|
||||||
|
session_id = cmsg.actual_user_id
|
||||||
|
if any([group_name in group_chat_in_one_session, 'ALL_GROUP' in group_chat_in_one_session]):
|
||||||
|
session_id = group_id
|
||||||
|
else:
|
||||||
|
return
|
||||||
|
context['session_id'] = session_id
|
||||||
|
context['receiver'] = group_id
|
||||||
|
else:
|
||||||
|
context['session_id'] = cmsg.other_user_id
|
||||||
|
context['receiver'] = cmsg.other_user_id
|
||||||
|
|
||||||
|
|
||||||
|
# 消息内容匹配过程,并处理content
|
||||||
|
if ctype == ContextType.TEXT:
|
||||||
|
if first_in and "」\n- - - - - - -" in content: # 初次匹配 过滤引用消息
|
||||||
|
logger.debug("[WX]reference query skipped")
|
||||||
|
return None
|
||||||
|
|
||||||
|
if context["isgroup"]: # 群聊
|
||||||
|
# 校验关键字
|
||||||
|
match_prefix = check_prefix(content, conf().get('group_chat_prefix'))
|
||||||
|
match_contain = check_contain(content, conf().get('group_chat_keyword'))
|
||||||
|
if match_prefix is not None or match_contain is not None:
|
||||||
|
if match_prefix:
|
||||||
|
content = content.replace(match_prefix, '', 1).strip()
|
||||||
|
elif context['msg'].is_at and not conf().get("group_at_off", False):
|
||||||
|
logger.info("[WX]receive group at, continue")
|
||||||
|
pattern = f'@{self.name}(\u2005|\u0020)'
|
||||||
|
content = re.sub(pattern, r'', content)
|
||||||
|
elif context["origin_ctype"] == ContextType.VOICE:
|
||||||
|
logger.info("[WX]receive group voice, checkprefix didn't match")
|
||||||
|
return None
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
else: # 单聊
|
||||||
|
match_prefix = check_prefix(content, conf().get('single_chat_prefix'))
|
||||||
|
if match_prefix is not None: # 判断如果匹配到自定义前缀,则返回过滤掉前缀+空格后的内容
|
||||||
|
content = content.replace(match_prefix, '', 1).strip()
|
||||||
|
elif context["origin_ctype"] == ContextType.VOICE: # 如果源消息是私聊的语音消息,允许不匹配前缀,放宽条件
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
img_match_prefix = check_prefix(content, conf().get('image_create_prefix'))
|
||||||
|
if img_match_prefix:
|
||||||
|
content = content.replace(img_match_prefix, '', 1).strip()
|
||||||
|
context.type = ContextType.IMAGE_CREATE
|
||||||
|
else:
|
||||||
|
context.type = ContextType.TEXT
|
||||||
|
context.content = content
|
||||||
|
elif context.type == ContextType.VOICE:
|
||||||
|
if 'desire_rtype' not in context and conf().get('voice_reply_voice'):
|
||||||
|
context['desire_rtype'] = ReplyType.VOICE
|
||||||
|
|
||||||
|
|
||||||
|
return context
|
||||||
|
|
||||||
|
# 统一的发送函数,每个Channel自行实现,根据reply的type字段发送不同类型的消息
|
||||||
|
def send(self, reply: Reply, context: Context, retry_cnt = 0):
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
|
# 处理消息 TODO: 如果wechaty解耦,此处逻辑可以放置到父类
|
||||||
|
def _handle(self, context: Context):
|
||||||
|
if context is None or not context.content:
|
||||||
|
return
|
||||||
|
logger.debug('[WX] ready to handle context: {}'.format(context))
|
||||||
|
# reply的构建步骤
|
||||||
|
reply = self._generate_reply(context)
|
||||||
|
|
||||||
|
logger.debug('[WX] ready to decorate reply: {}'.format(reply))
|
||||||
|
# reply的包装步骤
|
||||||
|
reply = self._decorate_reply(context, reply)
|
||||||
|
|
||||||
|
# reply的发送步骤
|
||||||
|
self._send_reply(context, reply)
|
||||||
|
|
||||||
|
def _generate_reply(self, context: Context, reply: Reply = Reply()) -> Reply:
|
||||||
|
e_context = PluginManager().emit_event(EventContext(Event.ON_HANDLE_CONTEXT, {
|
||||||
|
'channel': self, 'context': context, 'reply': reply}))
|
||||||
|
reply = e_context['reply']
|
||||||
|
if not e_context.is_pass():
|
||||||
|
logger.debug('[WX] ready to handle context: type={}, content={}'.format(context.type, context.content))
|
||||||
|
if context.type == ContextType.TEXT or context.type == ContextType.IMAGE_CREATE: # 文字和图片消息
|
||||||
|
reply = super().build_reply_content(context.content, context)
|
||||||
|
elif context.type == ContextType.VOICE: # 语音消息
|
||||||
|
msg = context['msg']
|
||||||
|
msg.prepare()
|
||||||
|
mp3_path = context.content
|
||||||
|
# mp3转wav
|
||||||
|
wav_path = os.path.splitext(mp3_path)[0] + '.wav'
|
||||||
|
try:
|
||||||
|
mp3_to_wav(mp3_path=mp3_path, wav_path=wav_path)
|
||||||
|
except Exception as e: # 转换失败,直接使用mp3,对于某些api,mp3也可以识别
|
||||||
|
logger.warning("[WX]mp3 to wav error, use mp3 path. " + str(e))
|
||||||
|
wav_path = mp3_path
|
||||||
|
# 语音识别
|
||||||
|
reply = super().build_voice_to_text(wav_path)
|
||||||
|
# 删除临时文件
|
||||||
|
try:
|
||||||
|
os.remove(wav_path)
|
||||||
|
os.remove(mp3_path)
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning("[WX]delete temp file error: " + str(e))
|
||||||
|
|
||||||
|
if reply.type == ReplyType.TEXT:
|
||||||
|
new_context = self._compose_context(
|
||||||
|
ContextType.TEXT, reply.content, **context.kwargs)
|
||||||
|
if new_context:
|
||||||
|
reply = self._generate_reply(new_context)
|
||||||
|
else:
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
logger.error('[WX] unknown context type: {}'.format(context.type))
|
||||||
|
return
|
||||||
|
return reply
|
||||||
|
|
||||||
|
def _decorate_reply(self, context: Context, reply: Reply) -> Reply:
|
||||||
|
if reply and reply.type:
|
||||||
|
e_context = PluginManager().emit_event(EventContext(Event.ON_DECORATE_REPLY, {
|
||||||
|
'channel': self, 'context': context, 'reply': reply}))
|
||||||
|
reply = e_context['reply']
|
||||||
|
desire_rtype = context.get('desire_rtype')
|
||||||
|
if not e_context.is_pass() and reply and reply.type:
|
||||||
|
if reply.type == ReplyType.TEXT:
|
||||||
|
reply_text = reply.content
|
||||||
|
if desire_rtype == ReplyType.VOICE:
|
||||||
|
reply = super().build_text_to_voice(reply.content)
|
||||||
|
return self._decorate_reply(context, reply)
|
||||||
|
if context['isgroup']:
|
||||||
|
reply_text = '@' + context['msg'].actual_user_nickname + ' ' + reply_text.strip()
|
||||||
|
reply_text = conf().get("group_chat_reply_prefix", "")+reply_text
|
||||||
|
else:
|
||||||
|
reply_text = conf().get("single_chat_reply_prefix", "")+reply_text
|
||||||
|
reply.content = reply_text
|
||||||
|
elif reply.type == ReplyType.ERROR or reply.type == ReplyType.INFO:
|
||||||
|
reply.content = str(reply.type)+":\n" + reply.content
|
||||||
|
elif reply.type == ReplyType.IMAGE_URL or reply.type == ReplyType.VOICE or reply.type == ReplyType.IMAGE:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
logger.error('[WX] unknown reply type: {}'.format(reply.type))
|
||||||
|
return
|
||||||
|
if desire_rtype and desire_rtype != reply.type and reply.type not in [ReplyType.ERROR, ReplyType.INFO]:
|
||||||
|
logger.warning('[WX] desire_rtype: {}, but reply type: {}'.format(context.get('desire_rtype'), reply.type))
|
||||||
|
return reply
|
||||||
|
|
||||||
|
def _send_reply(self, context: Context, reply: Reply):
|
||||||
|
if reply and reply.type:
|
||||||
|
e_context = PluginManager().emit_event(EventContext(Event.ON_SEND_REPLY, {
|
||||||
|
'channel': self, 'context': context, 'reply': reply}))
|
||||||
|
reply = e_context['reply']
|
||||||
|
if not e_context.is_pass() and reply and reply.type:
|
||||||
|
logger.debug('[WX] ready to send reply: {} to {}'.format(reply, context))
|
||||||
|
self._send(reply, context)
|
||||||
|
|
||||||
|
def _send(self, reply: Reply, context: Context, retry_cnt = 0):
|
||||||
|
try:
|
||||||
|
self.send(reply, context)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error('[WX] sendMsg error: {}'.format(e))
|
||||||
|
if retry_cnt < 2:
|
||||||
|
time.sleep(3+3*retry_cnt)
|
||||||
|
self._send(reply, context, retry_cnt+1)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def check_prefix(content, prefix_list):
|
||||||
|
for prefix in prefix_list:
|
||||||
|
if content.startswith(prefix):
|
||||||
|
return prefix
|
||||||
|
return None
|
||||||
|
|
||||||
|
def check_contain(content, keyword_list):
|
||||||
|
if not keyword_list:
|
||||||
|
return None
|
||||||
|
for ky in keyword_list:
|
||||||
|
if content.find(ky) != -1:
|
||||||
|
return True
|
||||||
|
return None
|
||||||
@@ -1,6 +1,7 @@
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
本类表示聊天消息,用于对itchat和wechaty的消息进行统一的封装
|
本类表示聊天消息,用于对itchat和wechaty的消息进行统一的封装
|
||||||
|
|
||||||
ChatMessage
|
ChatMessage
|
||||||
msg_id: 消息id
|
msg_id: 消息id
|
||||||
create_time: 消息创建时间
|
create_time: 消息创建时间
|
||||||
@@ -13,7 +14,7 @@ from_user_nickname: 发送者昵称
|
|||||||
to_user_id: 接收者id
|
to_user_id: 接收者id
|
||||||
to_user_nickname: 接收者昵称
|
to_user_nickname: 接收者昵称
|
||||||
|
|
||||||
other_user_id: 对方的id,如果你是发送者,那这个就是接收者id,如果你是接收者,那这个就是发送者id
|
other_user_id: 对方的id,如果你是发送者,那这个就是接收者id,如果你是接收者,那这个就是发送者id,如果是群消息,那这一直是群id
|
||||||
other_user_nickname: 同上
|
other_user_nickname: 同上
|
||||||
|
|
||||||
is_group: 是否是群消息
|
is_group: 是否是群消息
|
||||||
|
|||||||
@@ -5,10 +5,10 @@ wechat channel
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import re
|
|
||||||
import requests
|
import requests
|
||||||
import io
|
import io
|
||||||
import time
|
import time
|
||||||
|
from channel.chat_channel import ChatChannel
|
||||||
from channel.wechat.wechat_message import *
|
from channel.wechat.wechat_message import *
|
||||||
from common.singleton import singleton
|
from common.singleton import singleton
|
||||||
from lib import itchat
|
from lib import itchat
|
||||||
@@ -19,7 +19,6 @@ from bridge.context import *
|
|||||||
from channel.channel import Channel
|
from channel.channel import Channel
|
||||||
from concurrent.futures import ThreadPoolExecutor
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
from common.log import logger
|
from common.log import logger
|
||||||
from common.tmp_dir import TmpDir
|
|
||||||
from config import conf
|
from config import conf
|
||||||
from common.time_check import time_checker
|
from common.time_check import time_checker
|
||||||
from common.expired_dict import ExpiredDict
|
from common.expired_dict import ExpiredDict
|
||||||
@@ -72,10 +71,9 @@ def _check(func):
|
|||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
@singleton
|
@singleton
|
||||||
class WechatChannel(Channel):
|
class WechatChannel(ChatChannel):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.user_id = None
|
super().__init__()
|
||||||
self.name = None
|
|
||||||
self.receivedMsgs = ExpiredDict(60*60*24)
|
self.receivedMsgs = ExpiredDict(60*60*24)
|
||||||
|
|
||||||
def startup(self):
|
def startup(self):
|
||||||
@@ -99,7 +97,7 @@ class WechatChannel(Channel):
|
|||||||
# start message listener
|
# start message listener
|
||||||
itchat.run()
|
itchat.run()
|
||||||
|
|
||||||
# handle_* 系列函数处理收到的消息后构造Context,然后传入handle函数中处理Context和发送回复
|
# handle_* 系列函数处理收到的消息后构造Context,然后传入_handle函数中处理Context和发送回复
|
||||||
# Context包含了消息的所有信息,包括以下属性
|
# Context包含了消息的所有信息,包括以下属性
|
||||||
# type 消息类型, 包括TEXT、VOICE、IMAGE_CREATE
|
# type 消息类型, 包括TEXT、VOICE、IMAGE_CREATE
|
||||||
# content 消息内容,如果是TEXT类型,content就是文本内容,如果是VOICE类型,content就是语音文件名,如果是IMAGE_CREATE类型,content就是图片生成命令
|
# content 消息内容,如果是TEXT类型,content就是文本内容,如果是VOICE类型,content就是语音文件名,如果是IMAGE_CREATE类型,content就是图片生成命令
|
||||||
@@ -107,7 +105,7 @@ class WechatChannel(Channel):
|
|||||||
# session_id: 会话id
|
# session_id: 会话id
|
||||||
# isgroup: 是否是群聊
|
# isgroup: 是否是群聊
|
||||||
# receiver: 需要回复的对象
|
# receiver: 需要回复的对象
|
||||||
# msg: itchat的原始消息对象
|
# msg: ChatMessage消息对象
|
||||||
# origin_ctype: 原始消息类型,语音转文字后,私聊时如果匹配前缀失败,会根据初始消息是否是语音来放宽触发规则
|
# origin_ctype: 原始消息类型,语音转文字后,私聊时如果匹配前缀失败,会根据初始消息是否是语音来放宽触发规则
|
||||||
# desire_rtype: 希望回复类型,默认是文本回复,设置为ReplyType.VOICE是语音回复
|
# desire_rtype: 希望回复类型,默认是文本回复,设置为ReplyType.VOICE是语音回复
|
||||||
|
|
||||||
@@ -117,55 +115,25 @@ class WechatChannel(Channel):
|
|||||||
if conf().get('speech_recognition') != True:
|
if conf().get('speech_recognition') != True:
|
||||||
return
|
return
|
||||||
logger.debug("[WX]receive voice msg: {}".format(cmsg.content))
|
logger.debug("[WX]receive voice msg: {}".format(cmsg.content))
|
||||||
from_user_id = cmsg.from_user_id
|
context = self._compose_context(ContextType.VOICE, cmsg.content, isgroup=False, msg=cmsg)
|
||||||
other_user_id = cmsg.other_user_id
|
if context:
|
||||||
if from_user_id == other_user_id:
|
thread_pool.submit(self._handle, context).add_done_callback(thread_pool_callback)
|
||||||
context = self._compose_context(ContextType.VOICE, cmsg.content, isgroup=False, msg=cmsg, receiver=other_user_id, session_id=other_user_id)
|
|
||||||
if context:
|
|
||||||
thread_pool.submit(self.handle, context).add_done_callback(thread_pool_callback)
|
|
||||||
|
|
||||||
@time_checker
|
@time_checker
|
||||||
@_check
|
@_check
|
||||||
def handle_text(self, cmsg : ChatMessage):
|
def handle_text(self, cmsg : ChatMessage):
|
||||||
logger.debug("[WX]receive text msg: " + json.dumps(cmsg._rawmsg, ensure_ascii=False))
|
logger.debug("[WX]receive text msg: " + json.dumps(cmsg._rawmsg, ensure_ascii=False))
|
||||||
content = cmsg.content
|
context = self._compose_context(ContextType.TEXT, cmsg.content, isgroup=False, msg=cmsg)
|
||||||
other_user_id = cmsg.other_user_id
|
|
||||||
if "」\n- - - - - - - - - - - - - - -" in cmsg.content:
|
|
||||||
logger.debug("[WX]reference query skipped")
|
|
||||||
return
|
|
||||||
|
|
||||||
context = self._compose_context(ContextType.TEXT, content, isgroup=False, msg=cmsg, receiver=other_user_id, session_id=other_user_id)
|
|
||||||
if context:
|
if context:
|
||||||
thread_pool.submit(self.handle, context).add_done_callback(thread_pool_callback)
|
thread_pool.submit(self._handle, context).add_done_callback(thread_pool_callback)
|
||||||
|
|
||||||
@time_checker
|
@time_checker
|
||||||
@_check
|
@_check
|
||||||
def handle_group(self, cmsg : ChatMessage):
|
def handle_group(self, cmsg : ChatMessage):
|
||||||
logger.debug("[WX]receive group msg: " + json.dumps(cmsg._rawmsg, ensure_ascii=False))
|
logger.debug("[WX]receive group msg: " + json.dumps(cmsg._rawmsg, ensure_ascii=False))
|
||||||
group_name = cmsg.other_user_nickname
|
context = self._compose_context(ContextType.TEXT, cmsg.content, isgroup=True, msg=cmsg)
|
||||||
group_id = cmsg.other_user_id
|
if context:
|
||||||
if not group_name:
|
thread_pool.submit(self._handle, context).add_done_callback(thread_pool_callback)
|
||||||
return ""
|
|
||||||
content = cmsg.content
|
|
||||||
if "」\n- - - - - - - - - - - - - - -" in content:
|
|
||||||
logger.debug("[WX]reference query skipped")
|
|
||||||
return ""
|
|
||||||
pattern = f'@{self.name}(\u2005|\u0020)'
|
|
||||||
content = re.sub(pattern, r'', content)
|
|
||||||
|
|
||||||
config = conf()
|
|
||||||
group_name_white_list = config.get('group_name_white_list', [])
|
|
||||||
group_name_keyword_white_list = config.get('group_name_keyword_white_list', [])
|
|
||||||
|
|
||||||
if any([group_name in group_name_white_list, 'ALL_GROUP' in group_name_white_list, check_contain(group_name, group_name_keyword_white_list)]):
|
|
||||||
group_chat_in_one_session = conf().get('group_chat_in_one_session', [])
|
|
||||||
session_id = cmsg.actual_user_id
|
|
||||||
if any([group_name in group_chat_in_one_session, 'ALL_GROUP' in group_chat_in_one_session]):
|
|
||||||
session_id = group_id
|
|
||||||
|
|
||||||
context = self._compose_context(ContextType.TEXT, content, isgroup=True, msg=cmsg, receiver=group_id, session_id=session_id)
|
|
||||||
if context:
|
|
||||||
thread_pool.submit(self.handle, context).add_done_callback(thread_pool_callback)
|
|
||||||
|
|
||||||
@time_checker
|
@time_checker
|
||||||
@_check
|
@_check
|
||||||
@@ -173,203 +141,33 @@ class WechatChannel(Channel):
|
|||||||
if conf().get('group_speech_recognition', False) != True:
|
if conf().get('group_speech_recognition', False) != True:
|
||||||
return
|
return
|
||||||
logger.debug("[WX]receive voice for group msg: {}".format(cmsg.content))
|
logger.debug("[WX]receive voice for group msg: {}".format(cmsg.content))
|
||||||
group_name = cmsg.other_user_nickname
|
context = self._compose_context(ContextType.VOICE, cmsg.content, isgroup=True, msg=cmsg)
|
||||||
group_id = cmsg.other_user_id
|
if context:
|
||||||
# 验证群名
|
thread_pool.submit(self._handle, context).add_done_callback(thread_pool_callback)
|
||||||
if not group_name:
|
|
||||||
return ""
|
|
||||||
|
|
||||||
config = conf()
|
|
||||||
group_name_white_list = config.get('group_name_white_list', [])
|
|
||||||
group_name_keyword_white_list = config.get('group_name_keyword_white_list', [])
|
|
||||||
if any([group_name in group_name_white_list, 'ALL_GROUP' in group_name_white_list, check_contain(group_name, group_name_keyword_white_list)]):
|
|
||||||
group_chat_in_one_session = conf().get('group_chat_in_one_session', [])
|
|
||||||
session_id = cmsg.actual_user_id
|
|
||||||
if any([group_name in group_chat_in_one_session, 'ALL_GROUP' in group_chat_in_one_session]):
|
|
||||||
session_id = group_id
|
|
||||||
context = self._compose_context(ContextType.VOICE, cmsg.content, isgroup=True, msg=cmsg, receiver=group_id, session_id=session_id)
|
|
||||||
if context:
|
|
||||||
thread_pool.submit(self.handle, context).add_done_callback(thread_pool_callback)
|
|
||||||
|
|
||||||
# 根据消息构造context,消息内容相关的触发项写在这里
|
|
||||||
def _compose_context(self, ctype: ContextType, content, **kwargs):
|
|
||||||
context = Context(ctype, content)
|
|
||||||
context.kwargs = kwargs
|
|
||||||
if 'origin_ctype' not in context:
|
|
||||||
context['origin_ctype'] = ctype
|
|
||||||
|
|
||||||
if ctype == ContextType.TEXT:
|
|
||||||
if context["isgroup"]: # 群聊
|
|
||||||
# 校验关键字
|
|
||||||
match_prefix = check_prefix(content, conf().get('group_chat_prefix'))
|
|
||||||
match_contain = check_contain(content, conf().get('group_chat_keyword'))
|
|
||||||
if match_prefix is not None or match_contain is not None:
|
|
||||||
# 判断如果匹配到自定义前缀,则返回过滤掉前缀+空格后的内容,用于实现类似自定义+前缀触发生成AI图片的功能
|
|
||||||
if match_prefix:
|
|
||||||
content = content.replace(match_prefix, '', 1).strip()
|
|
||||||
elif context['msg'].is_at and not conf().get("group_at_off", False):
|
|
||||||
logger.info("[WX]receive group at, continue")
|
|
||||||
elif context["origin_ctype"] == ContextType.VOICE:
|
|
||||||
logger.info("[WX]receive group voice, checkprefix didn't match")
|
|
||||||
return None
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
else: # 单聊
|
|
||||||
match_prefix = check_prefix(content, conf().get('single_chat_prefix'))
|
|
||||||
if match_prefix is not None: # 判断如果匹配到自定义前缀,则返回过滤掉前缀+空格后的内容
|
|
||||||
content = content.replace(match_prefix, '', 1).strip()
|
|
||||||
elif context["origin_ctype"] == ContextType.VOICE: # 如果源消息是私聊的语音消息,允许不匹配前缀,放宽条件
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
img_match_prefix = check_prefix(content, conf().get('image_create_prefix'))
|
|
||||||
if img_match_prefix:
|
|
||||||
content = content.replace(img_match_prefix, '', 1).strip()
|
|
||||||
context.type = ContextType.IMAGE_CREATE
|
|
||||||
else:
|
|
||||||
context.type = ContextType.TEXT
|
|
||||||
context.content = content
|
|
||||||
elif context.type == ContextType.VOICE:
|
|
||||||
if 'desire_rtype' not in context and conf().get('voice_reply_voice'):
|
|
||||||
context['desire_rtype'] = ReplyType.VOICE
|
|
||||||
return context
|
|
||||||
|
|
||||||
# 统一的发送函数,每个Channel自行实现,根据reply的type字段发送不同类型的消息
|
# 统一的发送函数,每个Channel自行实现,根据reply的type字段发送不同类型的消息
|
||||||
def send(self, reply: Reply, receiver, retry_cnt = 0):
|
def send(self, reply: Reply, context: Context):
|
||||||
try:
|
receiver = context["receiver"]
|
||||||
if reply.type == ReplyType.TEXT:
|
if reply.type == ReplyType.TEXT:
|
||||||
itchat.send(reply.content, toUserName=receiver)
|
itchat.send(reply.content, toUserName=receiver)
|
||||||
logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver))
|
logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver))
|
||||||
elif reply.type == ReplyType.ERROR or reply.type == ReplyType.INFO:
|
elif reply.type == ReplyType.ERROR or reply.type == ReplyType.INFO:
|
||||||
itchat.send(reply.content, toUserName=receiver)
|
itchat.send(reply.content, toUserName=receiver)
|
||||||
logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver))
|
logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver))
|
||||||
elif reply.type == ReplyType.VOICE:
|
elif reply.type == ReplyType.VOICE:
|
||||||
itchat.send_file(reply.content, toUserName=receiver)
|
itchat.send_file(reply.content, toUserName=receiver)
|
||||||
logger.info('[WX] sendFile={}, receiver={}'.format(reply.content, receiver))
|
logger.info('[WX] sendFile={}, receiver={}'.format(reply.content, receiver))
|
||||||
elif reply.type == ReplyType.IMAGE_URL: # 从网络下载图片
|
elif reply.type == ReplyType.IMAGE_URL: # 从网络下载图片
|
||||||
img_url = reply.content
|
img_url = reply.content
|
||||||
pic_res = requests.get(img_url, stream=True)
|
pic_res = requests.get(img_url, stream=True)
|
||||||
image_storage = io.BytesIO()
|
image_storage = io.BytesIO()
|
||||||
for block in pic_res.iter_content(1024):
|
for block in pic_res.iter_content(1024):
|
||||||
image_storage.write(block)
|
image_storage.write(block)
|
||||||
image_storage.seek(0)
|
image_storage.seek(0)
|
||||||
itchat.send_image(image_storage, toUserName=receiver)
|
itchat.send_image(image_storage, toUserName=receiver)
|
||||||
logger.info('[WX] sendImage url={}, receiver={}'.format(img_url,receiver))
|
logger.info('[WX] sendImage url={}, receiver={}'.format(img_url,receiver))
|
||||||
elif reply.type == ReplyType.IMAGE: # 从文件读取图片
|
elif reply.type == ReplyType.IMAGE: # 从文件读取图片
|
||||||
image_storage = reply.content
|
image_storage = reply.content
|
||||||
image_storage.seek(0)
|
image_storage.seek(0)
|
||||||
itchat.send_image(image_storage, toUserName=receiver)
|
itchat.send_image(image_storage, toUserName=receiver)
|
||||||
logger.info('[WX] sendImage, receiver={}'.format(receiver))
|
logger.info('[WX] sendImage, receiver={}'.format(receiver))
|
||||||
except Exception as e:
|
|
||||||
logger.error('[WX] sendMsg error: {}, receiver={}'.format(e, receiver))
|
|
||||||
if retry_cnt < 2:
|
|
||||||
time.sleep(3+3*retry_cnt)
|
|
||||||
self.send(reply, receiver, retry_cnt + 1)
|
|
||||||
|
|
||||||
# 处理消息 TODO: 如果wechaty解耦,此处逻辑可以放置到父类
|
|
||||||
def handle(self, context: Context):
|
|
||||||
if context is None or not context.content:
|
|
||||||
return
|
|
||||||
logger.debug('[WX] ready to handle context: {}'.format(context))
|
|
||||||
# reply的构建步骤
|
|
||||||
reply = self._generate_reply(context)
|
|
||||||
|
|
||||||
logger.debug('[WX] ready to decorate reply: {}'.format(reply))
|
|
||||||
# reply的包装步骤
|
|
||||||
reply = self._decorate_reply(context, reply)
|
|
||||||
|
|
||||||
# reply的发送步骤
|
|
||||||
self._send_reply(context, reply)
|
|
||||||
|
|
||||||
def _generate_reply(self, context: Context, reply: Reply = Reply()) -> Reply:
|
|
||||||
e_context = PluginManager().emit_event(EventContext(Event.ON_HANDLE_CONTEXT, {
|
|
||||||
'channel': self, 'context': context, 'reply': reply}))
|
|
||||||
reply = e_context['reply']
|
|
||||||
if not e_context.is_pass():
|
|
||||||
logger.debug('[WX] ready to handle context: type={}, content={}'.format(context.type, context.content))
|
|
||||||
if context.type == ContextType.TEXT or context.type == ContextType.IMAGE_CREATE: # 文字和图片消息
|
|
||||||
reply = super().build_reply_content(context.content, context)
|
|
||||||
elif context.type == ContextType.VOICE: # 语音消息
|
|
||||||
msg = context['msg']
|
|
||||||
msg.prepare()
|
|
||||||
mp3_path = context.content
|
|
||||||
# mp3转wav
|
|
||||||
wav_path = os.path.splitext(mp3_path)[0] + '.wav'
|
|
||||||
try:
|
|
||||||
mp3_to_wav(mp3_path=mp3_path, wav_path=wav_path)
|
|
||||||
except Exception as e: # 转换失败,直接使用mp3,对于某些api,mp3也可以识别
|
|
||||||
logger.warning("[WX]mp3 to wav error, use mp3 path. " + str(e))
|
|
||||||
wav_path = mp3_path
|
|
||||||
# 语音识别
|
|
||||||
reply = super().build_voice_to_text(wav_path)
|
|
||||||
# 删除临时文件
|
|
||||||
try:
|
|
||||||
os.remove(wav_path)
|
|
||||||
os.remove(mp3_path)
|
|
||||||
except Exception as e:
|
|
||||||
logger.warning("[WX]delete temp file error: " + str(e))
|
|
||||||
|
|
||||||
if reply.type == ReplyType.TEXT:
|
|
||||||
new_context = self._compose_context(
|
|
||||||
ContextType.TEXT, reply.content, **context.kwargs)
|
|
||||||
if new_context:
|
|
||||||
reply = self._generate_reply(new_context)
|
|
||||||
else:
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
logger.error('[WX] unknown context type: {}'.format(context.type))
|
|
||||||
return
|
|
||||||
return reply
|
|
||||||
|
|
||||||
def _decorate_reply(self, context: Context, reply: Reply) -> Reply:
|
|
||||||
if reply and reply.type:
|
|
||||||
e_context = PluginManager().emit_event(EventContext(Event.ON_DECORATE_REPLY, {
|
|
||||||
'channel': self, 'context': context, 'reply': reply}))
|
|
||||||
reply = e_context['reply']
|
|
||||||
desire_rtype = context.get('desire_rtype')
|
|
||||||
if not e_context.is_pass() and reply and reply.type:
|
|
||||||
if reply.type == ReplyType.TEXT:
|
|
||||||
reply_text = reply.content
|
|
||||||
if desire_rtype == ReplyType.VOICE:
|
|
||||||
reply = super().build_text_to_voice(reply.content)
|
|
||||||
return self._decorate_reply(context, reply)
|
|
||||||
if context['isgroup']:
|
|
||||||
reply_text = '@' + context['msg'].actual_user_nickname + ' ' + reply_text.strip()
|
|
||||||
reply_text = conf().get("group_chat_reply_prefix", "")+reply_text
|
|
||||||
else:
|
|
||||||
reply_text = conf().get("single_chat_reply_prefix", "")+reply_text
|
|
||||||
reply.content = reply_text
|
|
||||||
elif reply.type == ReplyType.ERROR or reply.type == ReplyType.INFO:
|
|
||||||
reply.content = str(reply.type)+":\n" + reply.content
|
|
||||||
elif reply.type == ReplyType.IMAGE_URL or reply.type == ReplyType.VOICE or reply.type == ReplyType.IMAGE:
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
logger.error('[WX] unknown reply type: {}'.format(reply.type))
|
|
||||||
return
|
|
||||||
if desire_rtype and desire_rtype != reply.type and reply.type not in [ReplyType.ERROR, ReplyType.INFO]:
|
|
||||||
logger.warning('[WX] desire_rtype: {}, but reply type: {}'.format(context.get('desire_rtype'), reply.type))
|
|
||||||
return reply
|
|
||||||
|
|
||||||
def _send_reply(self, context: Context, reply: Reply):
|
|
||||||
if reply and reply.type:
|
|
||||||
e_context = PluginManager().emit_event(EventContext(Event.ON_SEND_REPLY, {
|
|
||||||
'channel': self, 'context': context, 'reply': reply}))
|
|
||||||
reply = e_context['reply']
|
|
||||||
if not e_context.is_pass() and reply and reply.type:
|
|
||||||
logger.debug('[WX] ready to send reply: {} to {}'.format(reply, context['receiver']))
|
|
||||||
self.send(reply, context['receiver'])
|
|
||||||
|
|
||||||
|
|
||||||
def check_prefix(content, prefix_list):
|
|
||||||
for prefix in prefix_list:
|
|
||||||
if content.startswith(prefix):
|
|
||||||
return prefix
|
|
||||||
return None
|
|
||||||
|
|
||||||
def check_contain(content, keyword_list):
|
|
||||||
if not keyword_list:
|
|
||||||
return None
|
|
||||||
for ky in keyword_list:
|
|
||||||
if content.find(ky) != -1:
|
|
||||||
return True
|
|
||||||
return None
|
|
||||||
|
|||||||
Reference in New Issue
Block a user