mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-16 16:25:55 +08:00
7c85c6f742
- Add support for message encryption in WeChat MP channel. - Add `wechatmp_aes_key` configuration item to `config.json`.
88 lines
3.8 KiB
Python
88 lines
3.8 KiB
Python
import time
|
|
|
|
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 wechatpy import parse_message
|
|
from common.log import logger
|
|
from config import conf
|
|
from wechatpy.replies import create_reply
|
|
|
|
# This class is instantiated once per query
|
|
class Query:
|
|
def GET(self):
|
|
return verify_server(web.input())
|
|
|
|
def POST(self):
|
|
# Make sure to return the instance that first created, @singleton will do that.
|
|
try:
|
|
args = web.input()
|
|
verify_server(args)
|
|
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
|
|
|
|
logger.info(
|
|
"[wechatmp] {}:{} Receive post query {} {}: {}".format(
|
|
web.ctx.env.get("REMOTE_ADDR"),
|
|
web.ctx.env.get("REMOTE_PORT"),
|
|
from_user,
|
|
message_id,
|
|
content,
|
|
)
|
|
)
|
|
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
|
|
)
|
|
if 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"
|
|
) # None or user openai_api_key
|
|
channel.produce(context)
|
|
# The reply will be sent by channel.send() in another thread
|
|
return "success"
|
|
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
|