github/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-03-18 04:25:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: user session cache

Browse Source
This commit is contained in:
zhayujie
2022-12-10 16:46:12 +08:00
parent 5581afaa0d
commit b17f49f96b
6 changed files with 39 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
channel/wechat/wechat_channel.py
View File

@@ -5,7 +5,9 @@ import itchat
import json
from itchat.content import *
from channel.channel import Channel
from concurrent.futures import ThreadPoolExecutor
thead_pool = ThreadPoolExecutor(max_workers=8)
@itchat.msg_register([TEXT])
def handler_receive_msg(msg):
@@ -18,20 +20,25 @@ class WechatChannel(Channel):
def startup(self):
# login by scan QRCode
itchat.auto_login()
itchat.auto_login(enableCmdQR=2)
# start message listener
itchat.run()
def handle(self, msg):
print("handle: ", msg)
print(json.dumps(msg, ensure_ascii=False))
# print("handle: ", msg)
print("[WX]receive msg: " + json.dumps(msg, ensure_ascii=False))
from_user_id = msg['FromUserName']
other_user_id = msg['User']['UserName']
if from_user_id == other_user_id:
self.send(super().build_reply_content(msg['Text']), from_user_id)
thead_pool.submit(self._do_send, msg['Text'], from_user_id)
def send(self, msg, receiver):
# time.sleep(random.randint(1, 3))
print(msg, receiver)
itchat.send(msg + " [bot]", toUserName=receiver)
def _do_send(self, send_msg, reply_user_id):
context = dict()
context['from_user_id'] = reply_user_id
self.send(super().build_reply_content(send_msg, context), reply_user_id)