78行Python代码实现现微信撤回消息功能

python曾经对我说："时日不多，赶紧用python"。于是看到了一个基于python的微信开源库：itchat，玩了一天，做了一个程序，把私聊撤回的信息可以收集起来并发送到个人微信的文件传输助手，包括：

• who ：谁发送的
• when ：什么时候发送的消息
• what：什么信息
• which：哪一类信息，包括：文本、图片、语音、视频、分享、位置、附件...

01 代码实现

# -*-encoding:utf-8-*- 
import os 
import re 
import shutil 
import time 
import itchat 
from itchat.content import * 
# 说明：可以撤回的有文本文字、语音、视频、图片、位置、名片、分享、附件 
# {msg_id:(msg_from,msg_to,msg_time,msg_time_rec,msg_type,msg_content,msg_share_url)} 
msg_dict = {} 
# 文件存储临时目录 
rev_tmp_dir = "/home/alic/revdir/" 
if not os.path.exists(rev_tmp_dir): os.mkdir(rev_tmp_dir) 
# 表情有一个问题 | 接受信息和接受note的msg_id不一致 巧合解决方案 
face_bug = none 
# 将接收到的消息存放在字典中，当接收到新消息时对字典中超时的消息进行清理 | 不接受不具有撤回功能的信息 
# [text, picture, map, card, sharing, recording, attachment, video, friends, note] 
@itchat.msg_register([text, picture, map, card, sharing, recording, attachment, video]) 
def handler_receive_msg(msg): 
  global face_bug 
  # 获取的是本地时间戳并格式化本地时间戳 e: 2017-04-21 21:30:08 
  msg_time_rec = time.strftime("%y-%m-%d %h:%m:%s", time.localtime()) 
  # 消息id 
  msg_id = msg['msgid'] 
  # 消息时间 
  msg_time = msg['createtime'] 
  # 消息发送人昵称 | 这里也可以使用remarkname备注　但是自己或者没有备注的人为none 
  msg_from = (itchat.search_friends(username=msg['fromusername']))["nickname"] 
  # 消息内容 
  msg_content = none 
  # 分享的链接 
  msg_share_url = none 
  if msg['type'] == 'text' \ 
      or msg['type'] == 'friends': 
    msg_content = msg['text'] 
  elif msg['type'] == 'recording' \ 
      or msg['type'] == 'attachment' \ 
      or msg['type'] == 'video' \ 
      or msg['type'] == 'picture': 
    msg_content = r"" + msg['filename'] 
    # 保存文件 
    msg['text'](rev_tmp_dir + msg['filename']) 
  elif msg['type'] == 'card': 
    msg_content = msg['recommendinfo']['nickname'] + r" 的名片" 
  elif msg['type'] == 'map': 
    x, y, location = re.search( 
      "<location x=\"(.*?)\" y=\"(.*?)\".*label=\"(.*?)\".*", msg['oricontent']).group(1, 2, 3) 
    if location is none: 
      msg_content = r"纬度->" + x.__str__() + " 经度->" + y.__str__() 
    else: 
      msg_content = r"" + location 
  elif msg['type'] == 'sharing': 
    msg_content = msg['text'] 
    msg_share_url = msg['url'] 
  face_bug = msg_content 
  # 更新字典 
  msg_dict.update( 
    { 
      msg_id: { 
        "msg_from": msg_from, "msg_time": msg_time, "msg_time_rec": msg_time_rec, 
        "msg_type": msg["type"], 
        "msg_content": msg_content, "msg_share_url": msg_share_url 
      } 
    } 
  ) 
# 收到note通知类消息，判断是不是撤回并进行相应操作 
@itchat.msg_register([note]) 
def send_msg_helper(msg): 
  global face_bug 
  if re.search(r"\<\!\[cdata\[.*撤回了一条消息\]\]\>", msg['content']) is not none: 
    # 获取消息的id 
    old_msg_id = re.search("\<msgid\>(.*?)\<\/msgid\>", msg['content']).group(1) 
    old_msg = msg_dict.get(old_msg_id, {}) 
    if len(old_msg_id) < 11: 
      itchat.send_file(rev_tmp_dir + face_bug, tousername='filehelper') 
      os.remove(rev_tmp_dir + face_bug) 
    else: 
      msg_body = "告诉你一个秘密~" + "\n" \ 
            + old_msg.get('msg_from') + " 撤回了 " + old_msg.get("msg_type") + " 消息" + "\n" \ 
            + old_msg.get('msg_time_rec') + "\n" \ 
            + "撤回了什么 ⇣" + "\n" \ 
            + r"" + old_msg.get('msg_content') 
      # 如果是分享存在链接 
      if old_msg['msg_type'] == "sharing": msg_body += "\n就是这个链接➣ " + old_msg.get('msg_share_url') 
      # 将撤回消息发送到文件助手 
      itchat.send(msg_body, tousername='filehelper') 
      # 有文件的话也要将文件发送回去 
      if old_msg["msg_type"] == "picture" \ 
          or old_msg["msg_type"] == "recording" \ 
          or old_msg["msg_type"] == "video" \ 
          or old_msg["msg_type"] == "attachment": 
        file = '@fil@%s' % (rev_tmp_dir + old_msg['msg_content']) 
        itchat.send(msg=file, tousername='filehelper') 
        os.remove(rev_tmp_dir + old_msg['msg_content']) 
      # 删除字典旧消息 
      msg_dict.pop(old_msg_id) 
if __name__ == '__main__': 
  itchat.auto_login(hotreload=true,enablecmdqr=2) 
  itchat.run()

该程序可以直接在终端运行，在终端扫码成功够即可登录成功，同时也可以打包在window系统运行(注意修改一下路径，推荐使用相对路径)。

 ~ python wx.py 
getting uuid of qr code. 
downloading qr code. 
please scan the qr code to log in. 
please press confirm on your phone. 
loading the contact, this may take a little while. 
�[3;j 
login successfully as alicfeng 
start auto replying.

02 效果图

 

03 itchat

上面都是编程逻辑的小事，我还是记录一下itchat微信这个开源库。

1. 简介

itchat是一个开源的微信个人号接口，使用python调用微信变得非常简单。简单是用itchat代码即可构建一个基于微信的即时通讯，更不错的体现在于方便扩展个人微信的在其他平台的更多通讯功能。

2. 安装

pip3 install itchat

3. itchat - helloworld

仅仅三行代码发送一条信息给文件传输助手。

import itchat 
itchat.auto_login(hotreload=true) 
itchat.send('hello alicfeng', tousername='filehelper')

4. 查看客户端

 

学习最重要的还是api说明手册：

github for itchat：

https://github.com/liduanwei/itchat

中文api：