#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 #針對表情包的內(nèi)容
@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'] #群消息的發(fā)送者,用戶的唯一標(biāo)識
    msg_from = msg['ActualNickName']#發(fā)送者群內(nèi)的昵稱
    friends = itchat.get_friends(update=True)#獲取所有好友
    for f in friends:
      if from_user == f['UserName']: #如果群消息是好友發(fā)的
        if f['RemarkName']: # 優(yōu)先使用好友的備注名稱，沒有則使用昵稱
          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']:#根據(jù)群消息的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']:#優(yōu)先使用備注名稱
      msg_from = itchat.search_friends(userName=msg['FromUserName'])['RemarkName']
    else:
      msg_from = itchat.search_friends(userName=msg['FromUserName'])['NickName'] #在好友列表中查詢發(fā)送信息的好友昵稱
    group_name = ""
  msg_time = msg['CreateTime'] #信息發(fā)送的時間
  msg_id = msg['MsgId']  #每條信息的id
  msg_content = None   #儲存信息的內(nèi)容
  msg_share_url = None  #儲存分享的鏈接，比如分享的文章和音樂
  if msg['Type'] == 'Text' or msg['Type'] == 'Friends':   #如果發(fā)送的消息是文本或者好友推薦
    msg_content = msg['Text']
  #如果發(fā)送的消息是附件、視頻、圖片、語音
  elif msg['Type'] == "Attachment" or msg['Type'] == "Video" \
      or msg['Type'] == 'Picture' \
      or msg['Type'] == 'Recording':
    msg_content = msg['FileName']  #內(nèi)容就是他們的文件名
    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__() + " 經(jīng)度->" + y.__str__()   #內(nèi)容為詳細(xì)的地址
    else:
      msg_content = r"" + location
  elif msg['Type'] == 'Sharing':   #如果消息為分享的音樂或者文章，詳細(xì)的內(nèi)容為文章的標(biāo)題或者是分享的名字
    msg_content = msg['Text']
    msg_share_url = msg['Url']    #記錄分享的url
  face_bug = msg_content
  #將信息存儲在字典中，每一個msg_id對應(yīng)一條信息
  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)
#監(jiān)聽是否有消息撤回
@itchat.msg_register(NOTE,isFriendChat=True,isGroupChat=True,isMpChat=True)
def information(msg):
  #如果這里的msg['Content']中包含消息撤回和id，就執(zhí)行下面的語句
  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: #如果發(fā)送的是表情包
      itchat.send_file(face_bug,toUserName='filehelper')
    else: #發(fā)送撤回的提示給文件助手
      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中發(fā)送給文件助手
      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')#將撤回消息發(fā)給文件助手
      #有文件的話也要將文件發(fā)送回去
      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()