import asyncio import time import web from wechatpy import parse_message from wechatpy.replies import ImageReply, VoiceReply, create_reply from bridge.context import * from bridge.reply import * from channel.wechatmp.common import * from channel.wechatmp.wechatmp_channel import WechatMPChannel from channel.wechatmp.wechatmp_message import WeChatMPMessage from common.log import logger from common.utils import split_string_by_utf8_length from config import conf, subscribe_msg # This class is instantiated once per query class Query: def GET(self): return verify_server(web.input()) def POST(self): try: args = web.input() verify_server(args) request_time = time.time() channel = WechatMPChannel() message = web.data() encrypt_func = lambda x: x if args.get("encrypt_type") == "aes": logger.debug("[wechatmp] Receive encrypted post data:\n" + message.decode("utf-8")) if not channel.crypto: raise Exception("Crypto not initialized, Please set wechatmp_aes_key in config.json") message = channel.crypto.decrypt_message(message, args.msg_signature, args.timestamp, args.nonce) encrypt_func = lambda x: channel.crypto.encrypt_message(x, args.nonce, args.timestamp) else: logger.debug("[wechatmp] Receive post data:\n" + message.decode("utf-8")) msg = parse_message(message) if msg.type in ["text", "voice", "image"]: wechatmp_msg = WeChatMPMessage(msg, client=channel.client) from_user = wechatmp_msg.from_user_id content = wechatmp_msg.content message_id = wechatmp_msg.msg_id supported = True if "【收到不支持的消息类型，暂无法显示】" in content: supported = False # not supported, used to refresh # New request if ( from_user not in channel.cache_dict and from_user not in channel.running or content.startswith("#") and message_id not in channel.request_cnt # insert the godcmd ): # The first query begin if msg.type == "voice" and wechatmp_msg.ctype == ContextType.TEXT and conf().get("voice_reply_voice", False): context = channel._compose_context(wechatmp_msg.ctype, content, isgroup=False, desire_rtype=ReplyType.VOICE, msg=wechatmp_msg) else: context = channel._compose_context(wechatmp_msg.ctype, content, isgroup=False, msg=wechatmp_msg) logger.debug("[wechatmp] context: {} {} {}".format(context, wechatmp_msg, supported)) if supported and context: # set private openai_api_key # if from_user is not changed in itchat, this can be placed at chat_channel user_data = conf().get_user_data(from_user) context["openai_api_key"] = user_data.get("openai_api_key") channel.running.add(from_user) channel.produce(context) else: trigger_prefix = conf().get("single_chat_prefix", [""])[0] if trigger_prefix or not supported: if trigger_prefix: reply_text = textwrap.dedent( f"""\ 请输入'{trigger_prefix}'接你想说的话跟我说话。 例如: {trigger_prefix}你好，很高兴见到你。""" ) else: reply_text = textwrap.dedent( """\ 你好，很高兴见到你。 请跟我说话吧。""" ) else: logger.error(f"[wechatmp] unknown error") reply_text = textwrap.dedent( """\ 未知错误，请稍后再试""" ) replyPost = create_reply(reply_text, msg) return encrypt_func(replyPost.render()) # Wechat official server will request 3 times (5 seconds each), with the same message_id. # Because the interval is 5 seconds, here assumed that do not have multithreading problems. request_cnt = channel.request_cnt.get(message_id, 0) + 1 channel.request_cnt[message_id] = request_cnt logger.info( "[wechatmp] Request {} from {} {} {}:{}\n{}".format( request_cnt, from_user, message_id, web.ctx.env.get("REMOTE_ADDR"), web.ctx.env.get("REMOTE_PORT"), content ) ) task_running = True waiting_until = request_time + 4 while time.time() < waiting_until: if from_user in channel.running: time.sleep(0.1) else: task_running = False break reply_text = "" if task_running: if request_cnt < 3: # waiting for timeout (the POST request will be closed by Wechat official server) time.sleep(2) # and do nothing, waiting for the next request return "success" else: # request_cnt == 3: # return timeout message reply_text = "【正在思考中，回复任意文字尝试获取回复】" replyPost = create_reply(reply_text, msg) return encrypt_func(replyPost.render()) # reply is ready channel.request_cnt.pop(message_id) # no return because of bandwords or other reasons if from_user not in channel.cache_dict and from_user not in channel.running: return "success" # Only one request can access to the cached data try: (reply_type, reply_content) = channel.cache_dict.pop(from_user) except KeyError: return "success" if reply_type == "text": if len(reply_content.encode("utf8")) <= MAX_UTF8_LEN: reply_text = reply_content else: continue_text = "\n【未完待续，回复任意文字以继续】" splits = split_string_by_utf8_length( reply_content, MAX_UTF8_LEN - len(continue_text.encode("utf-8")), max_split=1, ) reply_text = splits[0] + continue_text channel.cache_dict[from_user] = ("text", splits[1]) logger.info( "[wechatmp] Request {} do send to {} {}: {}\n{}".format( request_cnt, from_user, message_id, content, reply_text, ) ) replyPost = create_reply(reply_text, msg) return encrypt_func(replyPost.render()) elif reply_type == "voice": media_id = reply_content asyncio.run_coroutine_threadsafe(channel.delete_media(media_id), channel.delete_media_loop) logger.info( "[wechatmp] Request {} do send to {} {}: {} voice media_id {}".format( request_cnt, from_user, message_id, content, media_id, ) ) replyPost = VoiceReply(message=msg) replyPost.media_id = media_id return encrypt_func(replyPost.render()) elif reply_type == "image": media_id = reply_content asyncio.run_coroutine_threadsafe(channel.delete_media(media_id), channel.delete_media_loop) logger.info( "[wechatmp] Request {} do send to {} {}: {} image media_id {}".format( request_cnt, from_user, message_id, content, media_id, ) ) replyPost = ImageReply(message=msg) replyPost.media_id = media_id return encrypt_func(replyPost.render()) elif msg.type == "event": logger.info("[wechatmp] Event {} from {}".format(msg.event, msg.source)) if msg.event in ["subscribe", "subscribe_scan"]: reply_text = subscribe_msg() if reply_text: replyPost = create_reply(reply_text, msg) return encrypt_func(replyPost.render()) else: return "success" else: logger.info("暂且不处理") return "success" except Exception as exc: logger.exception(exc) return exc