還在好奇好友撤回了什麼消息嗎?羣裏撤回了什麼消息?下面的代碼實現了:即使羣、好友撤回了文本消息、表情、圖片等消息,自己也能知道撤回的什麼。
#coding=utf-8
import itchat
from itchat.content import TEXT
from itchat.content import *
import sys
import time
import re
import os
msg_information = {}
face_bug=None #針對表情包的內容
@itchat.msg_register([TEXT,PICTURE,FRIENDS,CARD,MAP,SHARING,RECORDING,ATTACHMENT,VIDEO],isFriendChat=True,isGroupChat=True)
def receive_msg(msg):
global face_bug
msg_time_rec = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) #接收消息的時間
if 'ActualNickName' in msg:
from_user = msg['ActualUserName'] #羣消息的發送者,用戶的唯一標識
msg_from = msg['ActualNickName']#發送者羣內的暱稱
friends = itchat.get_friends(update=True)#獲取所有好友
for f in friends:
if from_user == f['UserName']: #如果羣消息是好友發的
if f['RemarkName']: # 優先使用好友的備註名稱,沒有則使用暱稱
msg_from = f['RemarkName']
else:
msg_from = f['NickName']
break
groups = itchat.get_chatrooms(update=True)#獲取所有的羣
for g in groups:
if msg['FromUserName'] == g['UserName']:#根據羣消息的FromUserName匹配是哪個羣
group_name = g['NickName']
group_menbers = g['MemberCount']
break
group_name = group_name + "(" + str(group_menbers) +")"
else:
if itchat.search_friends(userName=msg['FromUserName'])['RemarkName']:#優先使用備註名稱
msg_from = itchat.search_friends(userName=msg['FromUserName'])['RemarkName']
else:
msg_from = itchat.search_friends(userName=msg['FromUserName'])['NickName'] #在好友列表中查詢發送信息的好友暱稱
group_name = ""
msg_time = msg['CreateTime'] #信息發送的時間
msg_id = msg['MsgId'] #每條信息的id
msg_content = None #儲存信息的內容
msg_share_url = None #儲存分享的鏈接,比如分享的文章和音樂
if msg['Type'] == 'Text' or msg['Type'] == 'Friends': #如果發送的消息是文本或者好友推薦
msg_content = msg['Text']
#如果發送的消息是附件、視頻、圖片、語音
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)) #下載文件
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
face_bug = msg_content
#將信息存儲在字典中,每一個msg_id對應一條信息
time.sleep(2)
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,
"group_name":group_name
}
}
)
del_info = []
for k in msg_information:
m_time = msg_information[k]['msg_time'] #取得消息時間
if int(time.time()) - m_time > 130: #如果消息時間是130秒甚至更久之前的,則刪除。
del_info.append(k)
if del_info:
for i in del_info:
msg_information.pop(i)
#監聽是否有消息撤回
@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 = old_msg['group_name'] + old_msg['msg_from'] +"\n" + old_msg['msg_time_rec'] \
+ "撤回了:" + "\n" + r"" + old_msg['msg_content']
#如果是分享的文件被撤回了,那麼就將分享的url加在msg_body中發送給文件助手
if old_msg['msg_type'] == "Sharing":
msg_body += "\n鏈接是:" + old_msg.get('msg_share_url')
#print(msg_body)
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,enableCmdQR=1)
itchat.run()