Spaces:
Paused
Paused
| # encoding:utf-8 | |
| """ | |
| wechat channel | |
| """ | |
| import itchat | |
| import json | |
| from itchat.content import * | |
| from channel.channel import Channel | |
| from concurrent.futures import ThreadPoolExecutor | |
| from common.log import logger | |
| from config import conf | |
| import requests | |
| import io | |
| thread_pool = ThreadPoolExecutor(max_workers=8) | |
| class WechatChannel(Channel): | |
| qrcode = b'' | |
| newInstance=None | |
| def __init__(self): | |
| pass | |
| def startup(self): | |
| # login by scan QRCode | |
| newInstance = itchat.load_sync_itchat() | |
| self.newInstance = newInstance | |
| def handler_single_msg(msg): | |
| self.handle(msg) | |
| return None | |
| def handler_group_msg(msg): | |
| self.handle_group(msg) | |
| return None | |
| newInstance.auto_login(qrCallback=self.qrCallback) | |
| # start message listener | |
| newInstance.run() | |
| def qrCallback(self, uuid, status, qrcode): | |
| self.qrcode = qrcode | |
| def getQrCode(self): | |
| return self.qrcode | |
| def handle(self, msg): | |
| logger.debug("[WX]receive msg: " + json.dumps(msg, ensure_ascii=False)) | |
| from_user_id = msg['FromUserName'] | |
| to_user_id = msg['ToUserName'] # 接收人id | |
| other_user_id = msg['User']['UserName'] # 对手方id | |
| content = msg['Text'] | |
| match_prefix = self.check_prefix(content, conf().get('single_chat_prefix')) | |
| if from_user_id == other_user_id and match_prefix is not None: | |
| # 好友向自己发送消息 | |
| if match_prefix != '': | |
| str_list = content.split(match_prefix, 1) | |
| if len(str_list) == 2: | |
| content = str_list[1].strip() | |
| img_match_prefix = self.check_prefix(content, conf().get('image_create_prefix')) | |
| if img_match_prefix: | |
| content = content.split(img_match_prefix, 1)[1].strip() | |
| thread_pool.submit(self._do_send_img, content, from_user_id) | |
| else: | |
| thread_pool.submit(self._do_send, content, from_user_id) | |
| elif to_user_id == other_user_id and match_prefix: | |
| # 自己给好友发送消息 | |
| str_list = content.split(match_prefix, 1) | |
| if len(str_list) == 2: | |
| content = str_list[1].strip() | |
| img_match_prefix = self.check_prefix(content, conf().get('image_create_prefix')) | |
| if img_match_prefix: | |
| content = content.split(img_match_prefix, 1)[1].strip() | |
| thread_pool.submit(self._do_send_img, content, to_user_id) | |
| else: | |
| thread_pool.submit(self._do_send, content, to_user_id) | |
| def handle_group(self, msg): | |
| logger.debug("[WX]receive group msg: " + json.dumps(msg, ensure_ascii=False)) | |
| group_name = msg['User'].get('NickName', None) | |
| group_id = msg['User'].get('UserName', None) | |
| if not group_name: | |
| return "" | |
| origin_content = msg['Content'] | |
| content = msg['Content'] | |
| content_list = content.split(' ', 1) | |
| context_special_list = content.split('\u2005', 1) | |
| if len(context_special_list) == 2: | |
| content = context_special_list[1] | |
| elif len(content_list) == 2: | |
| content = content_list[1] | |
| config = conf() | |
| match_prefix = (msg['IsAt'] and not config.get("group_at_off", False)) or self.check_prefix(origin_content, config.get('group_chat_prefix')) \ | |
| or self.check_contain(origin_content, config.get('group_chat_keyword')) | |
| if ('ALL_GROUP' in config.get('group_name_white_list') or group_name in config.get('group_name_white_list') or self.check_contain(group_name, config.get('group_name_keyword_white_list'))) and match_prefix: | |
| img_match_prefix = self.check_prefix(content, conf().get('image_create_prefix')) | |
| if img_match_prefix: | |
| content = content.split(img_match_prefix, 1)[1].strip() | |
| thread_pool.submit(self._do_send_img, content, group_id) | |
| else: | |
| thread_pool.submit(self._do_send_group, content, msg) | |
| def send(self, msg, receiver): | |
| logger.info('[WX] sendMsg={}, receiver={}'.format(msg, receiver)) | |
| self.newInstance.send(msg, toUserName=receiver) | |
| def _do_send(self, query, reply_user_id): | |
| try: | |
| if not query: | |
| return | |
| context = dict() | |
| context['from_user_id'] = reply_user_id | |
| reply_text = super().build_reply_content(query, context) | |
| if reply_text: | |
| self.send(conf().get("single_chat_reply_prefix") + reply_text, reply_user_id) | |
| except Exception as e: | |
| logger.exception(e) | |
| def _do_send_img(self, query, reply_user_id): | |
| try: | |
| if not query: | |
| return | |
| context = dict() | |
| context['type'] = 'IMAGE_CREATE' | |
| img_url = super().build_reply_content(query, context) | |
| if not img_url: | |
| return | |
| # 图片下载 | |
| 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) | |
| # 图片发送 | |
| logger.info('[WX] sendImage, receiver={}'.format(reply_user_id)) | |
| self.newInstance.send_image(image_storage, reply_user_id) | |
| except Exception as e: | |
| logger.exception(e) | |
| def _do_send_group(self, query, msg): | |
| if not query: | |
| return | |
| context = dict() | |
| context['from_user_id'] = msg['ActualUserName'] | |
| reply_text = super().build_reply_content(query, context) | |
| if reply_text: | |
| reply_text = '@' + msg['ActualNickName'] + ' ' + reply_text.strip() | |
| self.send(conf().get("group_chat_reply_prefix", "") + reply_text, msg['User']['UserName']) | |
| def check_prefix(self, content, prefix_list): | |
| for prefix in prefix_list: | |
| if content.startswith(prefix): | |
| return prefix | |
| return None | |
| def check_contain(self, content, keyword_list): | |
| if not keyword_list: | |
| return None | |
| for ky in keyword_list: | |
| if content.find(ky) != -1: | |
| return True | |
| return None | |