原理很简单,每一条消息都是有id的,我们把所有消息都存储在dict字典里,就存储id和对应的消息内容。
itchat.content.TEXT
:消息
itchat.content.NOTE
:通知
撤回消息就属于通知了,每次有通知,我们都检测一下,如果是撤回消息,就把撤回的消息发送出去。return就是在哪个聊天框里检测到,发送到那里。
# -*- coding: UTF8 -*-
import itchat
import requests
import re
data = {} # 用于存储所有消息的dict字典
# 当接收到消息会触发
@itchat.msg_register(itchat.content.TEXT,isFriendChat=True,isGroupChat=True)
def record(msg):
content = msg['Content']
id = msg['MsgId']
data[id] = content
# 当收到通知会触发
@itchat.msg_register(itchat.content.NOTE,isFriendChat=True,isGroupChat=True)
def reply(msg):
if '撤回了一条消息' in msg['Content']:
old_msg_id = re.search("\<msgid\>(.*?)\<\/msgid\>", msg['Content']).group(1) # 用正则表达式匹配出最近一次撤回消息的id
old_msg = data.get(old_msg_id) # 读取存储在字典中对应id的消息
if old_msg != None:
return "你撤回了消息:" + str(old_msg)
当然我们有点不足之处就是如果消息存储过多的话,会占用我们的很多的内存。
当然我们有改进方法: 方法1: 就是把消息存储到本地用的时候读取,当然这样存储的文件也会越来越大。 方法2: 每个消息都有对应的发送时间,我们获取到并且和当前时间对比,当时间超过2分钟就把消息删除,正好消息本身超过2分钟也不能撤回了。
程序运行效果图: