mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-01-19 01:21:01 +08:00
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
import time
|
|
|
|
import web
|
|
|
|
import channel.wechatmp.receive as receive
|
|
import channel.wechatmp.reply as reply
|
|
from bridge.context import *
|
|
from channel.wechatmp.common import *
|
|
from channel.wechatmp.wechatmp_channel import WechatMPChannel
|
|
from common.log import logger
|
|
from config import conf
|
|
|
|
|
|
# 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.
|
|
channel = WechatMPChannel()
|
|
try:
|
|
webData = web.data()
|
|
# logger.debug("[wechatmp] Receive request:\n" + webData.decode("utf-8"))
|
|
wechatmp_msg = receive.parse_xml(webData)
|
|
if wechatmp_msg.msg_type == "text" or wechatmp_msg.msg_type == "voice":
|
|
from_user = wechatmp_msg.from_user_id
|
|
message = wechatmp_msg.content.decode("utf-8")
|
|
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,
|
|
message,
|
|
)
|
|
)
|
|
context = channel._compose_context(
|
|
ContextType.TEXT, message, 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 wechatmp_msg.msg_type == "event":
|
|
logger.info(
|
|
"[wechatmp] Event {} from {}".format(
|
|
wechatmp_msg.Event, wechatmp_msg.from_user_id
|
|
)
|
|
)
|
|
content = subscribe_msg()
|
|
replyMsg = reply.TextMsg(
|
|
wechatmp_msg.from_user_id, wechatmp_msg.to_user_id, content
|
|
)
|
|
return replyMsg.send()
|
|
else:
|
|
logger.info("暂且不处理")
|
|
return "success"
|
|
except Exception as exc:
|
|
logger.exception(exc)
|
|
return exc
|