机器人有连接到数据库,所以可以长记忆对话,多线程可以使得机器人同时和几个好友聊天。将消息发送到远程API进行处理,并回复处理后的消息。脚本还允许为每个好友分配一个唯一的随机 ID,以便跟不同的好友聊天并维护聊天记忆。文章来源地址https://www.toymoban.com/news/detail-696246.html
# encoding:utf-8
import io
from itchat.content import *
import itchat
import threading
import requests
import random
import string
import configparser
# 存储每个好友的记忆和随机 ID 的映射
friend_memories = {}
friend_ids = {}
# 初始化配置文件
config = configparser.ConfigParser()
config.read('config.ini', encoding='utf-8')
role_id = "001" # 机器人id
def qrCallback(uuid, status, qrcode):
if status == "0":
try:
from PIL import Image
img = Image.open(io.BytesIO(qrcode))
_thread = threading.Thread(target=img.show, args=("QRCode",))
_thread.setDaemon(True)
_thread.start()
except Exception as e:
pass
import qrcode
url = f"https://login.weixin.qq.com/l/{uuid}"
qr_api1 = "https://api.isoyu.com/qr/?m=1&e=L&p=20&url={}".format(url)
qr_api2 = "https://api.qrserver.com/v1/create-qr-code/?size=400×400&data={}".format(url)
qr_api3 = "https://api.pwmqr.com/qrcode/create/?url={}".format(url)
qr_api4 = "https://my.tv.sohu.com/user/a/wvideo/getQRCode.do?text={}".format(url)
print("You can also scan QRCode in any website below:")
print(qr_api3)
print(qr_api4)
print(qr_api2)
print(qr_api1)
qr = qrcode.QRCode(border=1)
qr.add_data(url)
qr.make(fit=True)
qr.print_ascii(invert=True)
# 生成随机 ID
def generate_random_id():
return ''.join(random.choices(string.ascii_letters + string.digits, k=10))
# 读取用户ID
def get_user_id(username):
try:
user_id = config.get('UserIDs', username)
return user_id
except configparser.NoOptionError:
return None
# 更新用户ID
def update_user_id(username, user_id):
config.set('UserIDs', username, user_id)
with open('config.ini', 'w') as configfile:
config.write(configfile)
# 处理消息的函数
def process_message(msg):
user_text = msg.text
friend_name = msg.user.nickName # 使用好友的昵称作为标识
print(friend_name)
# 初始化好友的记忆和随机 ID,如果没有的话
if friend_name not in friend_memories:
friend_memories[friend_name] = ""
friend_id = get_user_id(friend_name)
if friend_id is None:
friend_id = generate_random_id() # 生成新的ID
update_user_id(friend_name, friend_id) # 保存到配置文件
friend_ids[friend_name] = friend_id # 将好友昵称和对应的ID关联起来
print(f"{friend_name} 的用户ID是:{friend_ids[friend_name]}")
print(f"Processing message from {friend_name} (ID: {friend_ids[friend_name]}): {user_text}")
api_url = "http://xxxxxx:8080/ask"
data = {'content': user_text, 'role_id': role_id, 'uid': friend_ids[friend_name]} # 使用好友的随机 ID 作为 uid
response = requests.post(api_url, data=data)
print(data)
if response.status_code == 200:
# 从API获取聊天响应
chat_response = response.json().get('data', [])
send_message = " ".join(chat_response) # 将聊天响应连接成一个字符串
# 更新好友的记忆
friend_memories[friend_name] = send_message
print("friend_memories:", friend_memories)
# 自动回复给发送消息的用户
msg.user.send(send_message)
print(f"Reply to {friend_name} (ID: {friend_ids[friend_name]}): {send_message}")
else:
print(f"Error processing message from {friend_name} (ID: {friend_ids[friend_name]})")
# 处理接收消息的函数
@itchat.msg_register(itchat.content.TEXT)
def text_reply(msg):
# 使用线程处理消息
thread = threading.Thread(target=process_message, args=(msg,))
thread.start()
if __name__ == "__main__":
itchat.auto_login(hotReload=True)
itchat.run()
文章来源:https://www.toymoban.com/news/detail-696246.html
到了这里,关于python连接自己的机器人接口在微信聊天的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!