mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-01-19 01:21:01 +08:00
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
import hashlib
|
||
import textwrap
|
||
|
||
from config import conf
|
||
|
||
MAX_UTF8_LEN = 2048
|
||
|
||
|
||
class WeChatAPIException(Exception):
|
||
pass
|
||
|
||
|
||
def verify_server(data):
|
||
try:
|
||
if len(data) == 0:
|
||
return "None"
|
||
signature = data.signature
|
||
timestamp = data.timestamp
|
||
nonce = data.nonce
|
||
echostr = data.echostr
|
||
token = conf().get("wechatmp_token") # 请按照公众平台官网\基本配置中信息填写
|
||
|
||
data_list = [token, timestamp, nonce]
|
||
data_list.sort()
|
||
sha1 = hashlib.sha1()
|
||
# map(sha1.update, data_list) #python2
|
||
sha1.update("".join(data_list).encode("utf-8"))
|
||
hashcode = sha1.hexdigest()
|
||
print("handle/GET func: hashcode, signature: ", hashcode, signature)
|
||
if hashcode == signature:
|
||
return echostr
|
||
else:
|
||
return ""
|
||
except Exception as Argument:
|
||
return Argument
|
||
|
||
|
||
def subscribe_msg():
|
||
trigger_prefix = conf().get("single_chat_prefix", [""])[0]
|
||
msg = textwrap.dedent(
|
||
f"""\
|
||
感谢您的关注！
|
||
这里是ChatGPT，可以自由对话。
|
||
资源有限，回复较慢，请勿着急。
|
||
支持通用表情输入。
|
||
暂时不支持图片输入。
|
||
支持图片输出，画字开头的问题将回复图片链接。
|
||
支持角色扮演和文字冒险两种定制模式对话。
|
||
输入'{trigger_prefix}#帮助' 查看详细指令。"""
|
||
)
|
||
return msg
|
||
|
||
|
||
def split_string_by_utf8_length(string, max_length, max_split=0):
|
||
encoded = string.encode("utf-8")
|
||
start, end = 0, 0
|
||
result = []
|
||
while end < len(encoded):
|
||
if max_split > 0 and len(result) >= max_split:
|
||
result.append(encoded[start:].decode("utf-8"))
|
||
break
|
||
end = start + max_length
|
||
# 如果当前字节不是 UTF-8 编码的开始字节，则向前查找直到找到开始字节为止
|
||
while end < len(encoded) and (encoded[end] & 0b11000000) == 0b10000000:
|
||
end -= 1
|
||
result.append(encoded[start:end].decode("utf-8"))
|
||
start = end
|
||
return result
|