From 18e9d6a9b95accd0f5cf05e49213f675a49fc6d0 Mon Sep 17 00:00:00 2001 From: ZQ7 Date: Mon, 20 Feb 2023 12:03:28 +0800 Subject: [PATCH] add wechaty --- README.md | 2 + channel/channel_factory.py | 6 +- channel/wechat/wechaty_channel.py | 201 ++++++++++++++++++++++++++++++ config-template.json | 1 + requirement.txt | 1 + 5 files changed, 210 insertions(+), 1 deletion(-) create mode 100644 channel/wechat/wechaty_channel.py diff --git a/README.md b/README.md index 4d11260..a377c77 100644 --- a/README.md +++ b/README.md @@ -13,6 +13,8 @@ # 更新日志 +>**2023.02.20：** 支持 [python-wechaty](https://github.com/wechaty/python-wechaty) Pad协议相对稳定，不易封号，但Token收费，可申请七天体验Token + >**2023.02.09：** 扫码登录存在封号风险，请谨慎使用，参考[#58](https://github.com/AutumnWhj/ChatGPT-wechat-bot/issues/158) >**2023.02.05：** 在openai官方接口方案中 (GPT-3模型) 实现上下文对话 diff --git a/channel/channel_factory.py b/channel/channel_factory.py index 5ecdbeb..95d3c67 100644 --- a/channel/channel_factory.py +++ b/channel/channel_factory.py @@ -3,6 +3,8 @@ channel factory """ from channel.wechat.wechat_channel import WechatChannel +from channel.wechat.wechaty_channel import WechatyChannel + def create_channel(channel_type): """ @@ -12,4 +14,6 @@ def create_channel(channel_type): """ if channel_type == 'wx': return WechatChannel() - raise RuntimeError \ No newline at end of file + elif channel_type == 'wxy': + return WechatyChannel() + raise RuntimeError diff --git a/channel/wechat/wechaty_channel.py b/channel/wechat/wechaty_channel.py new file mode 100644 index 0000000..8f27f6d --- /dev/null +++ b/channel/wechat/wechaty_channel.py @@ -0,0 +1,201 @@ +# encoding:utf-8 + +""" +wechaty channel +Python Wechaty - https://github.com/wechaty/python-wechaty +""" +import io +import os +import json +import time +import asyncio +import requests +from typing import Optional, Union +from wechaty_puppet import MessageType, FileBox, ScanStatus # type: ignore +from wechaty import Wechaty, Contact +from wechaty.user import Message, Room, MiniProgram, UrlLink +from channel.channel import Channel +from common.log import logger +from config import conf + + +class WechatyChannel(Channel): + + def __init__(self): + pass + + def startup(self): + asyncio.run(self.main()) + + async def main(self): + config = conf() + # 使用PadLocal协议 比较稳定(免费web协议 os.environ['WECHATY_PUPPET_SERVICE_ENDPOINT'] = '127.0.0.1:8080') + token = config.get('wechaty_puppet_service_token') + os.environ['WECHATY_PUPPET_SERVICE_TOKEN'] = token + global bot + bot = Wechaty() + + bot.on('scan', self.on_scan) + bot.on('login', self.on_login) + bot.on('message', self.on_message) + await bot.start() + + async def on_login(self, contact: Contact): + logger.info('[WX] login user={}'.format(contact)) + + async def on_scan(self, status: ScanStatus, qr_code: Optional[str] = None, + data: Optional[str] = None): + contact = self.Contact.load(self.contact_id) + logger.info('[WX] scan user={}, scan status={}, scan qr_code={}'.format(contact, status.name, qr_code)) + # print(f'user <{contact}> scan status: {status.name} , 'f'qr_code: {qr_code}') + + async def on_message(self, msg: Message): + """ + listen for message event + """ + from_contact = msg.talker() # 获取消息的发送者 + to_contact = msg.to() # 接收人 + room = msg.room() # 获取消息来自的群聊. 如果消息不是来自群聊, 则返回None + from_user_id = from_contact.contact_id + to_user_id = to_contact.contact_id # 接收人id + # other_user_id = msg['User']['UserName'] # 对手方id + content = msg.text() + mention_content = await msg.mention_text() # 返回过滤掉@name后的消息 + match_prefix = self.check_prefix(content, conf().get('single_chat_prefix')) + conversation: Union[Room, Contact] = from_contact if room is None else room + + if room is None and msg.type() == MessageType.MESSAGE_TYPE_TEXT: + if not msg.is_self() 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() + await self._do_send_img(content, from_user_id) + else: + await self._do_send(content, from_user_id) + elif msg.is_self() 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() + await self._do_send_img(content, to_user_id) + else: + await self._do_send(content, to_user_id) + elif room and msg.type() == MessageType.MESSAGE_TYPE_TEXT: + # 群组&文本消息 + room_id = room.room_id + room_name = await room.topic() + from_user_id = from_contact.contact_id + from_user_name = from_contact.name + is_at = await msg.mention_self() + content = mention_content + config = conf() + match_prefix = (is_at and not config.get("group_at_off", False)) \ + or self.check_prefix(content, config.get('group_chat_prefix')) \ + or self.check_contain(content, config.get('group_chat_keyword')) + if ('ALL_GROUP' in config.get('group_name_white_list') or room_name in config.get( + 'group_name_white_list') or self.check_contain(room_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() + await self._do_send_group_img(content, room_id) + else: + await self._do_send_group(content, room_id, from_user_id, from_user_name) + + async def send(self, message: Union[str, Message, FileBox, Contact, UrlLink, MiniProgram], receiver): + logger.info('[WX] sendMsg={}, receiver={}'.format(message, receiver)) + if receiver: + contact = await bot.Contact.find(receiver) + await contact.say(message) + + async def send_group(self, message: Union[str, Message, FileBox, Contact, UrlLink, MiniProgram], receiver): + logger.info('[WX] sendMsg={}, receiver={}'.format(message, receiver)) + if receiver: + room = await bot.Room.find(receiver) + await room.say(message) + + async 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: + await self.send(conf().get("single_chat_reply_prefix") + reply_text, reply_user_id) + except Exception as e: + logger.exception(e) + + async 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)) + t = int(time.time()) + file_box = FileBox.from_url(url=img_url, name=str(t) + '.png') + await self.send(file_box, reply_user_id) + except Exception as e: + logger.exception(e) + + async def _do_send_group(self, query, group_id, group_user_id, group_user_name): + if not query: + return + context = dict() + context['from_user_id'] = str(group_id) + '-' + str(group_user_id) + reply_text = super().build_reply_content(query, context) + if reply_text: + reply_text = '@' + group_user_name + ' ' + reply_text.strip() + await self.send_group(conf().get("group_chat_reply_prefix", "") + reply_text, group_id) + + async def _do_send_group_img(self, query, reply_room_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 + # 图片发送 + logger.info('[WX] sendImage, receiver={}'.format(reply_room_id)) + t = int(time.time()) + file_box = FileBox.from_url(url=img_url, name=str(t) + '.png') + await self.send_group(file_box, reply_room_id) + except Exception as e: + logger.exception(e) + + 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 diff --git a/config-template.json b/config-template.json index 6e99c48..bafb557 100644 --- a/config-template.json +++ b/config-template.json @@ -1,5 +1,6 @@ { "open_ai_api_key": "YOUR API KEY", + "wechaty_puppet_service_token": "WECHATY PUPPET SERVICE TOKEN", "single_chat_prefix": ["bot", "@bot"], "single_chat_reply_prefix": "[bot] ", "group_chat_prefix": ["@bot"], diff --git a/requirement.txt b/requirement.txt index 18dce49..98dd5af 100644 --- a/requirement.txt +++ b/requirement.txt @@ -1,2 +1,3 @@ itchat-uos==1.5.0.dev0 openai +wechaty