成为海王的日子——我做了一个微信自动聊天的工具

一直幻想着能够成为一个海王，于是做了一个微信自动聊天的工具。

测试微信版本：wechat 3.9.12.17

采用技术：

• Bmob后端云（AI对话和存储需要自动聊天的人和prompt）
• uiautomation
• pyautogui

开发语言：

• python，conda环境下运行

最终的效果大家可以看B站：www.bilibili.com/video/BV1yK…

一、获取微信对话内容

这里采用了网上的一些开源项目进行修改，写了一个自己的WeChat控制类，采用的是uiautomation和pyautogui组件进行UI的控制，模拟人的操作。

WeChat控制类的内容比较多，为了方便阅读，这里只呈现一部分，有需要的朋友可以联系我获取。

class WeChat:
    def __init__(self, path, locale="zh-CN"):
        # 微信打开路径
        self.path = path

        # 用于复制内容到剪切板
        self.app = QApplication([])

        self.lc = WeChatLocale(locale)

    # 鼠标移动到控件上
    def move(self,element):
        x, y = element.GetPosition()
        auto.SetCursorPos(x, y)

    # 鼠标快速点击控件
    def click(self,element):
        x, y = element.GetPosition()
        auto.Click(x, y)

    # 鼠标右键点击控件
    def right_click(self,element):
        x, y = element.GetPosition()
        auto.RightClick(x, y)


    # 鼠标快速点击两下控件
    def double_click(self,element):
        x, y = element.GetPosition()
        auto.SetCursorPos(x, y)
        element.DoubleClick()


    # 打开微信客户端
    def open_wechat(self):
        subprocess.Popen(self.path)

    # 搜寻微信客户端控件
    def get_wechat(self):
        return auto.WindowControl(Depth=1, Name=self.lc.weixin)

    # 防止微信长时间挂机导致掉线
    def prevent_offline(self):
        self.open_wechat()
        self.get_wechat()

        search_box = auto.EditControl(Depth=8, Name=self.lc.search)
        self.click(search_box)

    # 搜索指定用户
    def get_contact(self, name):
        self.open_wechat()
        self.get_wechat()

        search_box = auto.EditControl(Depth=8, Name=self.lc.search)
        self.click(search_box)

        pyperclip.copy(name)
        auto.SendKeys("{Ctrl}v")

        # 等待客户端搜索联系人
        time.sleep(0.3)
        search_box.SendKeys("{enter}")

    # 鼠标移动到发送按钮处点击发送消息
    def press_enter(self):
        # 获取发送按钮
        send_button = auto.ButtonControl(Depth=15, Name=self.lc.send)
        self.click(send_button)
        
    # 检测微信发新消息的用户
    def check_new_user(self):
        self.open_wechat()
        self.get_wechat()

        users = []

        # 获取左侧聊天按钮
        chat_btn = auto.ButtonControl(Name=self.lc.chats)
        self.double_click(chat_btn)

        # 持续点击聊天按钮，直到获取完全部新消息
        item = auto.ListItemControl(Depth=10)
        prev_name = item.ButtonControl().Name
        while True:
            # 判断该联系人是否有新消息
            pane_control = item.PaneControl()
            if len(pane_control.GetChildren()) == 3:
                users.append(item.ButtonControl().Name)

            self.click(item)

            # 跳转到下一个新消息
            self.double_click(chat_btn)
            item = auto.ListItemControl(Depth=10)

            # 已经完成遍历，退出循环
            if prev_name == item.ButtonControl().Name:
                break

            prev_name = item.ButtonControl().Name

        return users

二、获取需要自动聊天的人和对应的prompt

这部分信息我存储在Bmob后端云上面，对应的表结构（表名为：autochat，创建的字段为：name、prompt）和测试的内容如下：

获取信息的时候，我采用了子线程的方式，每隔300秒获取一次新的需要自动对话的微信和对应的prompt，代码如下：

# Bmob对象
bmob = Bmob(config['bmob_appid'], config['bmob_secret'])

# 存储从Bmob后端云获取到的自动对话的微信名和对应的prompt
name_prompts = {}

# 从Bmob后端云获取自动聊天的微信名和prompt
def get_user_prompt():
    users = bmob.findObjects('autochat')
    name_prompts.clear()
    for user in users:
        name_prompts[user.name] = user.prompt

# 每隔5分钟获取一次要自动聊天微信名称和对应的prompt
def run_with_interval():
    while True:
        get_user_prompt()
        time.sleep(300)

if __name__ == "__main__":
    t = threading.Thread(target=run_with_interval)
    t.start()

三、组装对话上下文和自动对话

这里主要用到了WeChat类的获取新对话人名字、获取某个人历史聊天记录的方法，和Bmob后端云的AI功能，具体代码如下：

# 执行自动聊天功能
def main():
    wechat = WeChat(path=config['wechat_path'])
    # 创建一个锁
    lock = threading.Lock()

    while True:
        try:
            comtypes.CoInitialize()
            # 确保微信操作的线程安全
            with lock:
                wechat.click(auto.ButtonControl(Name=wechat.lc.facorites))
                users = wechat.check_new_user()
                if len(users) <= 0:
                    time.sleep(5)
                    print("暂时没有新消息")
                    continue

                for user in users:
                    if user not in name_prompts.keys():
                        time.sleep(5)
                        print(f"{user}不在需要自动问答的名单上")
                        continue
                    else:
                        # 获取聊天记录，30这个数字可以调整，越大的话，AI会越了解你，但消耗的token也越多
                        msgList = wechat.get_dialogs(user, 30)
                        if len(msgList) <= 0:
                            continue
                        
                        # 组装上下文对话记录
                        ai = []
                        ai.append({"content": name_prompts[user], "role": "system"})
                        for msg in msgList:
                            chatmsg = msg[2].replace('[动画表情]','')
                            if chatmsg=='':
                                continue

                            if msg[1] == user:
                                ai.append({"content": chatmsg, "role": "user"})
                            else:
                                ai.append({"content": chatmsg, "role": "assistant"})

                        bmob.connectAI()
                        to = bmob.chat2(ai)
                        bmob.closeAI()
                        print('ai:'+to)

                        if to != "":
                            wechat.send_msg(user, to)
        except Exception as e:
            print(e)