|
- import time
- import asyncio
-
- import web
-
- from channel.wechatmp.wechatmp_message import WeChatMPMessage
- from bridge.context import *
- from bridge.reply import *
- from channel.wechatmp.common import *
- from channel.wechatmp.wechatmp_channel import WechatMPChannel
- from common.log import logger
- from config import conf
- from wechatpy import parse_message
- from wechatpy.replies import create_reply, ImageReply, VoiceReply
-
- # 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()
- 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
|
