|
|
@@ -12,9 +12,12 @@ from concurrent.futures import ThreadPoolExecutor
|
|
|
from common.log import logger
|
|
|
from common.tmp_dir import TmpDir
|
|
|
from config import conf
|
|
|
+from plugins import *
|
|
|
+
|
|
|
import requests
|
|
|
import io
|
|
|
|
|
|
+
|
|
|
thread_pool = ThreadPoolExecutor(max_workers=8)
|
|
|
|
|
|
|
|
|
@@ -49,8 +52,8 @@ class WechatChannel(Channel):
|
|
|
|
|
|
# handle_* 系列函数处理收到的消息后构造context,然后调用handle函数处理context
|
|
|
# context是一个字典,包含了消息的所有信息,包括以下key
|
|
|
- # type: 消息类型,包括TEXT、VOICE、CMD_IMAGE_CREATE
|
|
|
- # content: 消息内容,如果是TEXT类型,content就是文本内容,如果是VOICE类型,content就是语音文件名,如果是CMD_IMAGE_CREATE类型,content就是图片生成命令
|
|
|
+ # type: 消息类型,包括TEXT、VOICE、IMAGE_CREATE
|
|
|
+ # content: 消息内容,如果是TEXT类型,content就是文本内容,如果是VOICE类型,content就是语音文件名,如果是IMAGE_CREATE类型,content就是图片生成命令
|
|
|
# session_id: 会话id
|
|
|
# isgroup: 是否是群聊
|
|
|
# msg: 原始消息对象
|
|
|
@@ -88,7 +91,7 @@ class WechatChannel(Channel):
|
|
|
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'] = 'CMD_IMAGE_CREATE'
|
|
|
+ context['type'] = 'IMAGE_CREATE'
|
|
|
else:
|
|
|
context['type'] = 'TEXT'
|
|
|
|
|
|
@@ -121,7 +124,7 @@ class WechatChannel(Channel):
|
|
|
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'] = 'CMD_IMAGE_CREATE'
|
|
|
+ context['type'] = 'IMAGE_CREATE'
|
|
|
else:
|
|
|
context['type'] = 'TEXT'
|
|
|
context['content'] = content
|
|
|
@@ -136,8 +139,7 @@ class WechatChannel(Channel):
|
|
|
|
|
|
thread_pool.submit(self.handle, context)
|
|
|
|
|
|
- # 统一的发送函数,根据reply的type字段发送不同类型的消息
|
|
|
-
|
|
|
+ # 统一的发送函数,每个Channel自行实现,根据reply的type字段发送不同类型的消息
|
|
|
def send(self, reply, receiver):
|
|
|
if reply['type'] == 'TEXT':
|
|
|
itchat.send(reply['content'], toUserName=receiver)
|
|
|
@@ -163,54 +165,63 @@ class WechatChannel(Channel):
|
|
|
itchat.send_image(image_storage, toUserName=receiver)
|
|
|
logger.info('[WX] sendImage, receiver={}'.format(receiver))
|
|
|
|
|
|
- # 处理消息
|
|
|
+ # 处理消息 TODO: 如果wechaty解耦,此处逻辑可以放置到父类
|
|
|
def handle(self, context):
|
|
|
- content = context['content']
|
|
|
- reply = None
|
|
|
+ reply = {}
|
|
|
|
|
|
logger.debug('[WX] ready to handle context: {}'.format(context))
|
|
|
+
|
|
|
# reply的构建步骤
|
|
|
- if context['type'] == 'TEXT' or context['type'] == 'CMD_IMAGE_CREATE':
|
|
|
- reply = super().build_reply_content(content, context)
|
|
|
- elif context['type'] == 'VOICE':
|
|
|
- msg = context['msg']
|
|
|
- file_name = TmpDir().path() + msg['FileName']
|
|
|
- msg.download(file_name)
|
|
|
- reply = super().build_voice_to_text(file_name)
|
|
|
- if reply['type'] != 'ERROR' and reply['type'] != 'INFO':
|
|
|
- reply = super().build_reply_content(reply['content'], context)
|
|
|
- if reply['type'] == 'TEXT':
|
|
|
- if conf().get('voice_reply_voice'):
|
|
|
- reply = super().build_text_to_voice(reply['content'])
|
|
|
- else:
|
|
|
- logger.error('[WX] unknown context type: {}'.format(context['type']))
|
|
|
- return
|
|
|
+ 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'] == 'TEXT' or context['type'] == 'IMAGE_CREATE':
|
|
|
+ reply = super().build_reply_content(context['content'], context)
|
|
|
+ elif context['type'] == 'VOICE':
|
|
|
+ msg = context['msg']
|
|
|
+ file_name = TmpDir().path() + msg['FileName']
|
|
|
+ msg.download(file_name)
|
|
|
+ reply = super().build_voice_to_text(file_name)
|
|
|
+ if reply['type'] != 'ERROR' and reply['type'] != 'INFO':
|
|
|
+ reply = super().build_reply_content(reply['content'], context)
|
|
|
+ if reply['type'] == 'TEXT':
|
|
|
+ if conf().get('voice_reply_voice'):
|
|
|
+ reply = super().build_text_to_voice(reply['content'])
|
|
|
+ else:
|
|
|
+ logger.error('[WX] unknown context type: {}'.format(context['type']))
|
|
|
+ return
|
|
|
|
|
|
logger.debug('[WX] ready to decorate reply: {}'.format(reply))
|
|
|
+
|
|
|
# reply的包装步骤
|
|
|
- if reply:
|
|
|
- if reply['type'] == 'TEXT':
|
|
|
- reply_text = reply['content']
|
|
|
- if context['isgroup']:
|
|
|
- reply_text = '@' + \
|
|
|
- context['msg']['ActualNickName'] + \
|
|
|
- ' ' + reply_text.strip()
|
|
|
- reply_text = conf().get("group_chat_reply_prefix", "")+reply_text
|
|
|
+ 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']
|
|
|
+ if not e_context.is_pass() and reply and reply['type']:
|
|
|
+ if reply['type'] == 'TEXT':
|
|
|
+ reply_text = reply['content']
|
|
|
+ if context['isgroup']:
|
|
|
+ reply_text = '@' + context['msg']['ActualNickName'] + ' ' + 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'] == 'ERROR' or reply['type'] == 'INFO':
|
|
|
+ reply['content'] = reply['type']+": " + reply['content']
|
|
|
+ elif reply['type'] == 'IMAGE_URL' or reply['type'] == 'VOICE' or reply['type'] == 'IMAGE':
|
|
|
+ pass
|
|
|
else:
|
|
|
- reply_text = conf().get("single_chat_reply_prefix", "")+reply_text
|
|
|
- reply['content'] = reply_text
|
|
|
- elif reply['type'] == 'ERROR' or reply['type'] == 'INFO':
|
|
|
- reply['content'] = reply['type']+": " + reply['content']
|
|
|
- elif reply['type'] == 'IMAGE_URL' or reply['type'] == 'VOICE':
|
|
|
- pass
|
|
|
- else:
|
|
|
- logger.error(
|
|
|
- '[WX] unknown reply type: {}'.format(reply['type']))
|
|
|
- return
|
|
|
- if reply:
|
|
|
- logger.debug('[WX] ready to send reply: {} to {}'.format(
|
|
|
- reply, context['receiver']))
|
|
|
- self.send(reply, context['receiver'])
|
|
|
+ logger.error('[WX] unknown reply type: {}'.format(reply['type']))
|
|
|
+ return
|
|
|
+
|
|
|
+ # 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):
|