python连接自己的机器人接口在微信聊天

机器人有连接到数据库,所以可以长记忆对话,多线程可以使得机器人同时和几个好友聊天。将消息发送到远程API进行处理,并回复处理后的消息。脚本还允许为每个好友分配一个唯一的随机 ID,以便跟不同的好友聊天并维护聊天记忆。

python 复制代码
# 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()
相关推荐
hi星尘5 小时前
深度解析:基于Python的微信小程序自动化操作实现
python·微信小程序·自动化
Doker 多克6 小时前
Django 缓存框架
python·缓存·django
miracletiger8 小时前
uv 新的包管理工具总结
linux·人工智能·python
我不会编程5558 小时前
Python Cookbook-6.10 保留对被绑定方法的引用且支持垃圾回收
开发语言·python
ʚɞ 短腿欧尼8 小时前
关系数据的可视化
python·pycharm·可视化·数据可视化·图表
PXM的算法星球11 小时前
【软件工程】面向对象编程(OOP)概念详解
java·python·软件工程
Humbunklung11 小时前
PySide6 GUI 学习笔记——常用类及控件使用方法(常用类矩阵QRectF)
笔记·python·学习·pyqt
蹦蹦跳跳真可爱58911 小时前
Python----深度学习(基于DNN的吃鸡预测)
python·深度学习·dnn
JJ1M812 小时前
Git技巧:Git Hook,自动触发,含实战分享
git·python·自动化
拓端研究室TRL12 小时前
PyMC+AI提示词贝叶斯项目反应IRT理论Rasch分析篮球比赛官方数据:球员能力与位置层级结构研究
大数据·人工智能·python·算法·机器学习