使用itchat監控微信消息,今後再也不爲撤回煩惱

##強大的Itchat itchat是一個開源的微信我的號接口,使用python封裝接入微信網頁版接口,經過調用itchat來登陸微信網頁版收發消息。 項目簡介 - itchat 掌握itchat以後,只要你有興趣能夠隨意開發本身的專屬微信網頁版客戶端。目前比較流行的就是利用它來找回他人撤回的微信消息,和設置自動聊天機器人python

##撤回的微信消息 找回邏輯很是簡單,對方撤回消息的時候,咱們會收到一個[note]類型的推送消息,裏面包含了撤回的消息idmsg_id。咱們能夠對全部收到的消息進行監控並暫時保存,當收到撤回的通知時從存儲中找到這條消息記錄經過某種方式告知本身。簡單概括以下linux

  1. 暫存微信消息,以消息ID做爲主鍵,建議直接存內存裏。若是是圖片等其餘資源,須要下載到一個臨時目錄
  2. 監控撤回通知。經過撤回消息關鍵字找到撤回的消息id,到暫存的歷史消息裏找到對應記錄,經過文件轉發助手直接發送到手機客戶端。若是是圖片等資源,則從臨時目錄找到對應文件發送
  3. 由於微信消息撤回時限爲2分鐘,所以能夠把2分鐘以前的全部歷史記錄刪除,以節省內存(或者保存到其餘地方)

前提是2分鐘內不會收到太多消息以致內存耗盡

代碼示例

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/wechat"
rev_tmp_dir = "F:\private files\wechat\\"
qr_dir = "F:\\qrcode.png"

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):
    # 獲取的是本地時間戳並格式化本地時間戳 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']
    #print(msg_id)
    # 消息時間
    msg_time = msg['CreateTime']
    # 消息發送人暱稱 | 這裏也可使用RemarkName備註 可是本身或者沒有備註的人爲None
    msg_from = (itchat.search_friends(userName=msg['FromUserName']))["NickName"]
    #print(msg_from)
    # 消息內容
    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" 的名片"
        # itchat.send(msg_content, toUserName='filehelper')
    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']
    # 更新字典
    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
        }
    })
    # 刪除過時記錄
    for mid in list(msg_dict.keys()):
        if (msg_dict[mid].get('msg_time') < time.time()-120):
            # 刪除記錄,以及對應的文件(略)
            del msg_dict[mid]
        else:
            break

 # 收到note通知類消息,判斷是否是撤回並進行相應操做
@ itchat.msg_register([NOTE])
def send_msg_helper(msg):
    # 匹配撤回消息
    if re.search(r"\<\!\[CDATA\[.*撤回了一條消息\]\]\>", msg['Content']) is not None:
        # 獲取消息的id
        old_msg_id = re.search("<msgid>(.*?)</msgid>", msg['Content']).group(1)
        print("發現撤回消息 " + old_msg_id)
        old_msg = msg_dict.get(old_msg_id, {})
        # print(old_msg)
        msg_body = "[撤回]" + " " + old_msg.get('msg_from') + " " + old_msg.get('msg_time_rec') + "\n" + old_msg.get('msg_content')
        # 若是是分享
        if old_msg['msg_type'] == "Sharing":
            msg_body += 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 = '@%s@%s' % ('img' if msg['Type'] == 'Picture' else 'fil', rev_tmp_dir + old_msg['msg_content'])
            itchat.send(msg=file, toUserName='filehelper')

if __name__ == '__main__':
    # 打開 hotReload 短期內可自動從新登陸
    # 命令行查看二維碼使用 enableCmdQR=True,關閉則會調用系統的文件查看命令(win 使用 open, linux 使用 xdg-open,須要yum install xdg-utils)
    itchat.auto_login(hotReload=True, enableCmdQR=1)
    itchat.run()

其餘應用

  • 聊天機器人自動回覆
  • 發送定時消息
  • 鬥圖(調用圖像識別api)
  • 聊天記錄備份,分析
  • 微信好友數據分析
相關文章
相關標籤/搜索