|
|
@@ -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
|
ZQ7