itchat 監控微信好友撤回消息

轉載自:https://blog.csdn.net/enweitech/article/details/79585043,修改部分代碼以適應python 3.x版本
# coding:utf-8
import itchat

from itchat.content import TEXT
from itchat.content import *
import sys
import time
import re
import importlib
import os

importlib.reload(sys)

msg_information = {}
face_bug = None  # 針對表情包的內容


@itchat.msg_register([TEXT, PICTURE, FRIENDS, CARD, MAP, SHARING, RECORDING, ATTACHMENT, VIDEO], isFriendChat=True,
                     isGroupChat=True, isMpChat=True)
def handle_receive_msg(msg):
    global face_bug
    msg_time_rec = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())  # 接受消息的時間
    msg_from = itchat.search_friends(userName=msg['FromUserName'])['NickName']  # 在好友列表中查詢發送信息的好友暱稱
    msg_time = msg['CreateTime']  # 信息發送的時間
    msg_id = msg['MsgId']  # 每條信息的id
    msg_content = None  # 儲存信息的內容
    msg_share_url = None  # 儲存分享的連接,好比分享的文章和音樂
    print
    msg['Type']
    print
    msg['MsgId']
    if msg['Type'] == 'Text' or msg['Type'] == 'Friends':  # 若是發送的消息是文本或者好友推薦
        msg_content = msg['Text']
        print
        msg_content

    # 若是發送的消息是附件、視屏、圖片、語音
    elif msg['Type'] == "Attachment" or msg['Type'] == "Video" \
            or msg['Type'] == 'Picture' \
            or msg['Type'] == 'Recording':
        msg_content = msg['FileName']  # 內容就是他們的文件名
        msg['Text'](str(msg_content))  # 下載文件
        # print msg_content
    elif msg['Type'] == 'Card':  # 若是消息是推薦的名片
        msg_content = msg['RecommendInfo']['NickName'] + '的名片'  # 內容就是推薦人的暱稱和性別
        if msg['RecommendInfo']['Sex'] == 1:
            msg_content += '性別爲男'
        else:
            msg_content += '性別爲女'

        print
        msg_content
    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']  # 記錄分享的url
        print
        msg_share_url
    face_bug = msg_content

    ##將信息存儲在字典中,每個msg_id對應一條信息
    msg_information.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
            }
        }
    )


##這個是用於監聽是否有消息撤回
@itchat.msg_register(NOTE, isFriendChat=True, isGroupChat=True, isMpChat=True)
def information(msg):
    # 這裏若是這裏的msg['Content']中包含消息撤回和id,就執行下面的語句
    if '撤回了一條消息' in msg['Content']:
        old_msg_id = re.search("\<msgid\>(.*?)\<\/msgid\>", msg['Content']).group(1)  # 在返回的content查找撤回的消息的id
        old_msg = msg_information.get(old_msg_id)  # 獲得消息
        print
        old_msg
        if len(old_msg_id) < 11:  # 若是發送的是表情包
            itchat.send_file(face_bug, toUserName='filehelper')
        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')
            # 若是是分享的文件被撤回了,那麼就將分享的url加在msg_body中發送給文件助手
            if old_msg['msg_type'] == "Sharing":
                msg_body += "\n就是這個連接➣ " + old_msg.get('msg_share_url')

            # 將撤回消息發送到文件助手
            itchat.send_msg(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' % (old_msg['msg_content'])
                itchat.send(msg=file, toUserName='filehelper')
                os.remove(old_msg['msg_content'])
            # 刪除字典舊消息
            msg_information.pop(old_msg_id)


itchat.auto_login(hotReload=True)
itchat.run()
相關文章
相關標籤/搜索