|
-
-
- """
- wechat channel
- """
-
- import os
- import requests
- import io
- import time
- import json
- from channel.chat_channel import ChatChannel
- from channel.wechat.wechat_message import *
- from common.singleton import singleton
- from common.log import logger
- from lib import itchat
- from lib.itchat.content import *
- from bridge.reply import *
- from bridge.context import *
- from concurrent.futures import ThreadPoolExecutor
- from config import conf
- from common.time_check import time_checker
- from common.expired_dict import ExpiredDict
- from plugins import *
- thread_pool = ThreadPoolExecutor(max_workers=8)
-
- def thread_pool_callback(worker):
- worker_exception = worker.exception()
- if worker_exception:
- logger.exception("Worker return exception: {}".format(worker_exception))
-
-
- @itchat.msg_register(TEXT)
- def handler_single_msg(msg):
- WechatChannel().handle_text(WeChatMessage(msg))
- return None
-
- @itchat.msg_register(TEXT, isGroupChat=True)
- def handler_group_msg(msg):
- WechatChannel().handle_group(WeChatMessage(msg,True))
- return None
-
- @itchat.msg_register(VOICE)
- def handler_single_voice(msg):
- WechatChannel().handle_voice(WeChatMessage(msg))
- return None
-
- @itchat.msg_register(VOICE, isGroupChat=True)
- def handler_group_voice(msg):
- WechatChannel().handle_group_voice(WeChatMessage(msg,True))
- return None
-
- def _check(func):
- def wrapper(self, cmsg: ChatMessage):
- msgId = cmsg.msg_id
- if msgId in self.receivedMsgs:
- logger.info("Wechat message {} already received, ignore".format(msgId))
- return
- self.receivedMsgs[msgId] = cmsg
- create_time = cmsg.create_time
- if conf().get('hot_reload') == True and int(create_time) < int(time.time()) - 60:
- logger.debug("[WX]history message {} skipped".format(msgId))
- return
- return func(self, cmsg)
- return wrapper
-
-
-
-
- def qrCallback(uuid,status,qrcode):
-
- if status == '0':
- import qrcode
- url = f"https://login.weixin.qq.com/l/{uuid}"
- qr = qrcode.QRCode(border=1)
- qr.add_data(url)
- qr.make(fit=True)
- qr.print_ascii(invert=True)
-
- qr_api="https://api.isoyu.com/qr/?m=1&e=L&p=20&url={}".format(url)
- print("You can also scan QRCode in website below:\n{}".format(qr_api))
- @singleton
- class WechatChannel(ChatChannel):
- def __init__(self):
- super().__init__()
- self.receivedMsgs = ExpiredDict(60*60*24)
-
- def startup(self):
-
- itchat.instance.receivingRetryCount = 600
-
- hotReload = conf().get('hot_reload', False)
- try:
- itchat.auto_login(enableCmdQR=2, hotReload=hotReload, qrCallback=qrCallback)
- except Exception as e:
- if hotReload:
- logger.error("Hot reload failed, try to login without hot reload")
- itchat.logout()
- os.remove("itchat.pkl")
- itchat.auto_login(enableCmdQR=2, hotReload=hotReload, qrCallback=qrCallback)
- else:
- raise e
- self.user_id = itchat.instance.storageClass.userName
- self.name = itchat.instance.storageClass.nickName
- logger.info("Wechat login success, user_id: {}, nickname: {}".format(self.user_id, self.name))
-
- itchat.run()
-
-
-
-
-
-
-
-
-
-
-
-
-
- @time_checker
- @_check
- def handle_voice(self, cmsg : ChatMessage):
- if conf().get('speech_recognition') != True:
- return
- logger.debug("[WX]receive voice msg: {}".format(cmsg.content))
- 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: {}, cmsg={}".format(json.dumps(cmsg._rawmsg, ensure_ascii=False), cmsg))
- 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)
-
- @time_checker
- @_check
- def handle_group(self, cmsg : ChatMessage):
- logger.debug("[WX]receive group msg: {}, cmsg={}".format(json.dumps(cmsg._rawmsg, ensure_ascii=False), cmsg))
- 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
- def handle_group_voice(self, cmsg : ChatMessage):
- if conf().get('group_speech_recognition', False) != True:
- return
- logger.debug("[WX]receive voice for group msg: {}".format(cmsg.content))
- 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)
-
-
- 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))
|