#coding=utf-8
importitchatfrom itchat.content importTEXTfrom itchat.content import *
importsysimporttimeimportreimportos
msg_information={}
face_bug=None #针对表情包的内容#这里的TEXT表示如果有人发送文本消息()#TEXT 文本 文本内容(文字消息)#MAP 地图 位置文本(位置分享)#CARD 名片 推荐人字典(推荐人的名片)#SHARING 分享 分享名称(分享的音乐或者文章等)#PICTURE 下载方法 图片/表情#RECORDING 语音 下载方法#ATTACHMENT 附件 下载方法#VIDEO 小视频 下载方法#FRIENDS 好友邀请 添加好友所需参数#SYSTEM 系统消息 更新内容的用户或群聊的UserName组成的列表#NOTE 通知 通知文本(消息撤回等),那么就会调用下面的方法#其中isFriendChat表示好友之间,isGroupChat表示群聊,isMapChat表示公众号
@itchat.msg_register([TEXT,PICTURE,FRIENDS,CARD,MAP,SHARING,RECORDING,ATTACHMENT,VIDEO],isFriendChat=True,isGroupChat=True)defreceive_msg(msg):globalface_bug#print("消息是:"+str(msg))
msg_time_rec = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) #接收消息的时间
#ActualNickName : 实际 NickName(昵称) 群消息里(msg)才有这个值
if 'ActualNickName' inmsg:
from_user= msg['ActualUserName'] #群消息的发送者,用户的唯一标识
msg_from = msg['ActualNickName']#发送者群内的昵称
friends = itchat.get_friends(update=True)#获取所有好友
for f infriends:if from_user == f['UserName']: #如果群消息是好友发的
if f['RemarkName']: #优先使用好友的备注名称,没有则使用昵称
msg_from = f['RemarkName']else:
msg_from= f['NickName']breakgroups= itchat.get_chatrooms(update=True)#获取所有的群
for g ingroups:if msg['FromUserName'] == g['UserName']:#根据群消息的FromUserName匹配是哪个群
group_name = g['NickName']
group_menbers= g['MemberCount']breakgroup_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_content = "F:\\weixininfo\\"+msg['FileName']
msg['Text'](str(msg_content)) #下载文件
elif msg['Type'] == 'Map': #如果消息为分享的位置信息
x, y, location =re.search("
msg_content= r"纬度->" + x.__str__() + "经度->" + y.__str__() #内容为详细的地址
else:
msg_content= r"" +locationelif 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
}
}
)#自动删除130秒之前的消息,避免数据量太大后引起内存不足
del_info =[]for k inmsg_information:
m_time= msg_information[k]['msg_time'] #取得消息时间
if int(time.time()) - m_time > 130:
del_info.append(k)ifdel_info:for i indel_info:
msg_information.pop(i)#监听是否有消息撤回
@itchat.msg_register(NOTE,isFriendChat=True,isGroupChat=True,isMpChat=True)definformation(msg):#如果这里的msg['Content']中包含消息撤回和id,就执行下面的语句
if '撤回了一条消息' in msg['Content']:
old_msg_id= re.search("\(.*?)\", 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()