forked from HsuHeinrich/we-chatgpt
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
100 lines (85 loc) · 3.14 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from flask import Flask, request, jsonify
import hashlib
import time
import xmltodict
import openai
import os
from chatbotv3 import Chatbot
os.environ['GPT_ENGINE'] = 'gpt-3.5-turbo'
api_key = os.environ.get('API_KEY')
app = Flask(__name__)
@app.route('/')
def index():
return 'hello world!'
@app.route('/wechat',methods=['GET','POST'])
def wechat():
print(api_key)
# if request.method =='GET':
# # 设置token,开发者配置中心使用
# token = 'hsuheinrich003'
# # 获取微信服务器发送过来的参数
# data = request.args
# signature = data.get('signature')
# timestamp = data.get('timestamp')
# nonce = data.get('nonce')
# echostr = data.get('echostr')
# # 对参数进行字典排序,拼接字符串
# temp = [timestamp, nonce, token]
# temp.sort()
# temp = ''.join(temp)
# # 加密
# if (hashlib.sha1(temp.encode('utf8')).hexdigest() == signature):
# return echostr
# else:
# return 'error', 403
# 根据请求方式进行判断
if request.method == 'POST':
# 获取微信服务器post过来的xml数据
xml = request.data
# 把xml格式的数据进行处理,转换成字典进行取值
print('xml:',xml)
try:
print('is ok')
req = xmltodict.parse(xml)['xml'] # 云端调试空内容报错解决:no element found
print('try:',req)
except:
return 'no element', 200
print('chat访问前:',req)
# 判断post过来的数据中数据类型是不是文本
if 'text' == req.get('MsgType'):
# 获取用户的信息,开始构造返回数据
try:
resp = {
'ToUserName':req.get('FromUserName'),
'FromUserName':req.get('ToUserName'),
'CreateTime':int(time.time()),
'MsgType':'text',
'Content':chatbot.ask(req.get('Content'))
}
print('chat访问后:',req)
# 把构造的字典转换成xml格式
xml = xmltodict.unparse({'xml':resp})
return xml
except:
resp = {
'ToUserName':req.get('FromUserName'),
'FromUserName':req.get('ToUserName'),
'CreateTime':int(time.time()),
'MsgType':'text',
'Content':'好像发生了点问题,请稍后再重新提问~'
}
xml = xmltodict.unparse({'xml':resp})
return xml
else:
resp = {
'ToUserName': req.get('FromUserName', ''),
'FromUserName': req.get('ToUserName', ''),
'CreateTime': int(time.time()),
'MsgType': 'text',
'Content': '目前仅支持文本消息~'
}
xml = xmltodict.unparse({'xml':resp})
return xml
if __name__ == '__main__':
chatbot = Chatbot(api_key=api_key)
app.run(host='0.0.0.0', port=80, debug=True)