From 87df588c8065749fc00ab569bffa7907d07a4ff4 Mon Sep 17 00:00:00 2001 From: lanvent Date: Fri, 31 Mar 2023 22:31:10 +0800 Subject: [PATCH] refactor: stripp processing unrelated to channel --- channel/channel.py | 2 +- channel/chat_channel.py | 223 ++++++++++++++++++++++++ channel/chat_message.py | 3 +- channel/wechat/wechat_channel.py | 284 +++++-------------------------- 4 files changed, 267 insertions(+), 245 deletions(-) create mode 100644 channel/chat_channel.py diff --git a/channel/channel.py b/channel/channel.py index 62e1ad3..f65a031 100644 --- a/channel/channel.py +++ b/channel/channel.py @@ -20,7 +20,7 @@ class Channel(object): """ raise NotImplementedError - def send(self, msg, receiver): + def send(self, reply: Reply, context: Context): """ send message to user :param msg: message content diff --git a/channel/chat_channel.py b/channel/chat_channel.py new file mode 100644 index 0000000..42470f9 --- /dev/null +++ b/channel/chat_channel.py @@ -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 diff --git a/channel/chat_message.py b/channel/chat_message.py index 9fc2075..4fdb1dd 100644 --- a/channel/chat_message.py +++ b/channel/chat_message.py @@ -1,6 +1,7 @@ """ 本类表示聊天消息，用于对itchat和wechaty的消息进行统一的封装 + ChatMessage msg_id: 消息id create_time: 消息创建时间 @@ -13,7 +14,7 @@ from_user_nickname: 发送者昵称 to_user_id: 接收者id to_user_nickname: 接收者昵称 -other_user_id: 对方的id，如果你是发送者，那这个就是接收者id，如果你是接收者，那这个就是发送者id +other_user_id: 对方的id，如果你是发送者，那这个就是接收者id，如果你是接收者，那这个就是发送者id，如果是群消息，那这一直是群id other_user_nickname: 同上 is_group: 是否是群消息 diff --git a/channel/wechat/wechat_channel.py b/channel/wechat/wechat_channel.py index cdbb333..6e9abaa 100644 --- a/channel/wechat/wechat_channel.py +++ b/channel/wechat/wechat_channel.py @@ -5,10 +5,10 @@ wechat channel """ import os -import re import requests import io import time +from channel.chat_channel import ChatChannel from channel.wechat.wechat_message import * from common.singleton import singleton from lib import itchat @@ -19,7 +19,6 @@ from bridge.context import * from channel.channel import Channel from concurrent.futures import ThreadPoolExecutor from common.log import logger -from common.tmp_dir import TmpDir from config import conf from common.time_check import time_checker from common.expired_dict import ExpiredDict @@ -72,10 +71,9 @@ def _check(func): return wrapper @singleton -class WechatChannel(Channel): +class WechatChannel(ChatChannel): def __init__(self): - self.user_id = None - self.name = None + super().__init__() self.receivedMsgs = ExpiredDict(60*60*24) def startup(self): @@ -99,7 +97,7 @@ class WechatChannel(Channel): # start message listener itchat.run() - # handle_* 系列函数处理收到的消息后构造Context，然后传入handle函数中处理Context和发送回复 + # handle_* 系列函数处理收到的消息后构造Context，然后传入_handle函数中处理Context和发送回复 # Context包含了消息的所有信息，包括以下属性 # type 消息类型, 包括TEXT、VOICE、IMAGE_CREATE # content 消息内容，如果是TEXT类型，content就是文本内容，如果是VOICE类型，content就是语音文件名，如果是IMAGE_CREATE类型，content就是图片生成命令 @@ -107,7 +105,7 @@ class WechatChannel(Channel): # session_id: 会话id # isgroup: 是否是群聊 # receiver: 需要回复的对象 - # msg: itchat的原始消息对象 + # msg: ChatMessage消息对象 # origin_ctype: 原始消息类型，语音转文字后，私聊时如果匹配前缀失败，会根据初始消息是否是语音来放宽触发规则 # desire_rtype: 希望回复类型，默认是文本回复，设置为ReplyType.VOICE是语音回复 @@ -117,55 +115,25 @@ class WechatChannel(Channel): if conf().get('speech_recognition') != True: return logger.debug("[WX]receive voice msg: {}".format(cmsg.content)) - from_user_id = cmsg.from_user_id - other_user_id = cmsg.other_user_id - if from_user_id == other_user_id: - 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) + context = self._compose_context(ContextType.VOICE, cmsg.content, isgroup=False, msg=cmsg) + if context: + thread_pool.submit(self._handle, context).add_done_callback(thread_pool_callback) @time_checker @_check def handle_text(self, cmsg : ChatMessage): logger.debug("[WX]receive text msg: " + json.dumps(cmsg._rawmsg, ensure_ascii=False)) - content = cmsg.content - 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) + context = self._compose_context(ContextType.TEXT, cmsg.content, isgroup=False, msg=cmsg) 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 @_check def handle_group(self, cmsg : ChatMessage): logger.debug("[WX]receive group msg: " + json.dumps(cmsg._rawmsg, ensure_ascii=False)) - group_name = cmsg.other_user_nickname - group_id = cmsg.other_user_id - if not group_name: - 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) + context = self._compose_context(ContextType.TEXT, cmsg.content, isgroup=True, msg=cmsg) + if context: + thread_pool.submit(self._handle, context).add_done_callback(thread_pool_callback) @time_checker @_check @@ -173,203 +141,33 @@ class WechatChannel(Channel): if conf().get('group_speech_recognition', False) != True: return logger.debug("[WX]receive voice for group msg: {}".format(cmsg.content)) - group_name = cmsg.other_user_nickname - group_id = cmsg.other_user_id - # 验证群名 - 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 + context = self._compose_context(ContextType.VOICE, cmsg.content, isgroup=True, msg=cmsg) + if context: + thread_pool.submit(self._handle, context).add_done_callback(thread_pool_callback) # 统一的发送函数，每个Channel自行实现，根据reply的type字段发送不同类型的消息 - def send(self, reply: Reply, receiver, retry_cnt = 0): - try: - if reply.type == ReplyType.TEXT: - itchat.send(reply.content, toUserName=receiver) - logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver)) - elif reply.type == ReplyType.ERROR or reply.type == ReplyType.INFO: - itchat.send(reply.content, toUserName=receiver) - logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver)) - elif reply.type == ReplyType.VOICE: - itchat.send_file(reply.content, toUserName=receiver) - logger.info('[WX] sendFile={}, receiver={}'.format(reply.content, receiver)) - elif reply.type == ReplyType.IMAGE_URL: # 从网络下载图片 - img_url = reply.content - pic_res = requests.get(img_url, stream=True) - image_storage = io.BytesIO() - for block in pic_res.iter_content(1024): - image_storage.write(block) - image_storage.seek(0) - itchat.send_image(image_storage, toUserName=receiver) - logger.info('[WX] sendImage url={}, receiver={}'.format(img_url,receiver)) - elif reply.type == ReplyType.IMAGE: # 从文件读取图片 - image_storage = reply.content - image_storage.seek(0) - itchat.send_image(image_storage, toUserName=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 + def send(self, reply: Reply, context: Context): + receiver = context["receiver"] + if reply.type == ReplyType.TEXT: + itchat.send(reply.content, toUserName=receiver) + logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver)) + elif reply.type == ReplyType.ERROR or reply.type == ReplyType.INFO: + itchat.send(reply.content, toUserName=receiver) + logger.info('[WX] sendMsg={}, receiver={}'.format(reply, receiver)) + elif reply.type == ReplyType.VOICE: + itchat.send_file(reply.content, toUserName=receiver) + logger.info('[WX] sendFile={}, receiver={}'.format(reply.content, receiver)) + elif reply.type == ReplyType.IMAGE_URL: # 从网络下载图片 + img_url = reply.content + pic_res = requests.get(img_url, stream=True) + image_storage = io.BytesIO() + for block in pic_res.iter_content(1024): + image_storage.write(block) + image_storage.seek(0) + itchat.send_image(image_storage, toUserName=receiver) + logger.info('[WX] sendImage url={}, receiver={}'.format(img_url,receiver)) + elif reply.type == ReplyType.IMAGE: # 从文件读取图片 + image_storage = reply.content + image_storage.seek(0) + itchat.send_image(image_storage, toUserName=receiver) + logger.info('[WX] sendImage, receiver={}'.format(receiver))